@glodon-aiot/dataset-annotation 3.19.1 → 3.19.2-alpha.1

package/dist/es/components/VLMAnnotationDetail/VlmModal/index.mjs

@@ -1,294 +1,309 @@
-var N = (c, r, f) => new Promise((n, p) => {
-  var C = (a) => {
+var g = (c, h, n) => new Promise((C, l) => {
+  var y = (a) => {
     try {
-      t(f.next(a));
-    } catch (v) {
-      p(v);
+      u(n.next(a));
+    } catch (t) {
+      l(t);
     }
-  }, y = (a) => {
+  }, w = (a) => {
     try {
-      t(f.throw(a));
-    } catch (v) {
-      p(v);
+      u(n.throw(a));
+    } catch (t) {
+      l(t);
     }
-  }, t = (a) => a.done ? n(a.value) : Promise.resolve(a.value).then(C, y);
-  t((f = f.apply(c, r)).next());
+  }, u = (a) => a.done ? C(a.value) : Promise.resolve(a.value).then(y, w);
+  u((n = n.apply(c, h)).next());
 });
-import { jsx as e, jsxs as s, Fragment as b } from "react/jsx-runtime";
-import { useRef as T, useState as o, useContext as ce, useEffect as I, useCallback as me } from "react";
-import { Modal as B, Button as h, Space as G, Spin as fe, Tabs as ue, message as U } from "antd";
+import { jsx as e, jsxs as s, Fragment as N } from "react/jsx-runtime";
+import { useRef as X, useState as d, useContext as ye, useEffect as I, useCallback as we } from "react";
+import { Modal as Y, Button as f, Space as Z, Spin as Me, Tabs as ke, message as _ } from "antd";
 import "./style.less.mjs";
-import { useKeyPress as $ } from "ahooks";
-import { LoadingOutlined as he, MenuUnfoldOutlined as ve, MenuFoldOutlined as pe } from "@ant-design/icons";
-import ge from "../../../DatasetContext/index.mjs";
-import Ne from "../../VLMAnnotation/index.mjs";
-import H from "../../Iconfont/index.mjs";
-import Ce from "../VLMDetail/index.mjs";
-const Oe = (c) => {
+import { useKeyPress as ee } from "ahooks";
+import { LoadingOutlined as xe, MenuUnfoldOutlined as be, MenuFoldOutlined as Ie } from "@ant-design/icons";
+import qe from "../../../DatasetContext/index.mjs";
+import Ve from "../../VLMAnnotation/index.mjs";
+import Qe from "../../VQAAnnotation/VQAAnnotationPanel.mjs";
+import ie from "../../Iconfont/index.mjs";
+import Le from "../VLMDetail/index.mjs";
+const Se = (c) => {
+  var z, E, K, B, G, S, U, $;
   const {
-    isVlmModalVisible: r,
-    setIsVlmModalVisible: f,
-    record: n,
-    onNextClick: p,
-    onPrevClick: C,
-    listLoading: y = !1,
-    datasetId: t,
-    versionId: a,
-    handleMarkClick: v,
-    updateSampleList: J,
-    filterMode: W = "",
-    onFilterModeChange: X,
-    samples: Y,
+    vlmMarkTmpl: h,
+    isVlmModalVisible: n,
+    setIsVlmModalVisible: C,
+    record: l,
+    onNextClick: y,
+    onPrevClick: w,
+    listLoading: u = !1,
+    datasetId: a,
+    versionId: t,
+    handleMarkClick: ae,
+    updateSampleList: ne,
+    filterMode: te = "",
+    onFilterModeChange: le,
+    samples: se,
     onSave: q,
-    selectSampleId: Z,
-    onClose: Q,
-    pagination: _,
-    currentIndex: ee
-  } = c, ie = T({}), M = T(null), [i, ae] = o(), [te, V] = o(!1), [u, w] = o(c.initialMode || "preview"), [ne, k] = o([]), [L, R] = o(0), [le, O] = o(!1), [S, x] = o(!0), [j, A] = o(!1), {
+    selectSampleId: re,
+    onClose: V,
+    pagination: oe,
+    currentIndex: de
+  } = c, ce = X({}), M = X(null), [i, me] = d(), [ve, Q] = d(!1), [v, k] = d(c.initialMode || "preview"), [fe, x] = d([]), [L, A] = d(0), [ue, O] = d(!1), [R, b] = d(!0), [P, j] = d(!1), {
     services: {
-      cvforceDatahub: d
+      cvforceDatahub: o
     },
     featuresConfig: {
       datasetManager: D
     } = {
       datasetManager: !1
     }
-  } = ce(ge), g = !(te || y), F = () => {
-    g && C();
-  }, P = () => {
-    g && p();
+  } = ye(qe), p = !(ve || u), F = () => {
+    p && w();
+  }, T = () => {
+    p && y();
   };
-  $("rightarrow", () => {
-    P();
-  }), $("leftarrow", () => {
+  ee("rightarrow", () => {
+    T();
+  }), ee("leftarrow", () => {
     F();
-  }), I(() => {
-    r ? c.initialMode && w(c.initialMode) : w("preview");
-  }, [r, c.initialMode]), I(() => {
-    if (r && t && a && u === "preview") {
-      if (V(!0), !(n != null && n.id))
+  });
+  const he = h === "VQA";
+  I(() => {
+    n ? c.initialMode && k(c.initialMode) : k("preview");
+  }, [n, c.initialMode]), I(() => {
+    if (n && a && t && v === "preview") {
+      if (Q(!0), !(l != null && l.id))
         return;
-      d == null || d.getDatasetsIdVersionsVidSsamplesSid(t, a, n.id).then((l) => {
-        ae(l);
-      }).finally(() => V(!1));
+      o == null || o.getDatasetsIdVersionsVidSsamplesSid(a, t, l.id).then((r) => {
+        me(r);
+      }).finally(() => Q(!1));
     }
-  }, [r, n, u]);
-  const se = me(() => N(void 0, null, function* () {
-    if (!(!t || !a || !(i != null && i.id))) {
+  }, [n, l, v]);
+  const pe = we(() => g(void 0, null, function* () {
+    if (!(!a || !t || !(i != null && i.id))) {
       O(!0);
       try {
-        const l = yield d == null ? void 0 : d.postGenerateVlmKieQa(t, a, {
+        const r = yield o == null ? void 0 : o.postGenerateVlmKieQa(a, t, {
           sampleIds: [i.id]
         });
-        k(l || []), R(0);
-      } catch (l) {
-        console.error("QA生成失败", l), k([]);
+        x(r || []), A(0);
+      } catch (r) {
+        console.error("QA生成失败", r), x([]);
       } finally {
         O(!1);
       }
     }
-  }), [t, a, i]);
+  }), [a, t, i]);
   I(() => {
-    if (!r || u !== "preview" || !(i != null && i.id))
+    if (!n || v !== "preview" || !(i != null && i.id))
       return;
     if (((i == null ? void 0 : i.labels) || []).length)
-      A(!1), se();
+      j(!1), pe();
     else {
-      k([]), x(!1), A(!0);
+      x([]), b(!1), j(!0);
       return;
     }
   }, [i]);
-  const re = () => N(void 0, null, function* () {
-    if (u === "annotation" && M.current) {
+  const ge = () => g(void 0, null, function* () {
+    if (v === "annotation" && M.current) {
       yield M.current.handleModalCancel();
       return;
     } else
-      Q();
-  }), de = () => {
-    w("annotation"), v((i == null ? void 0 : i.id) || (n == null ? void 0 : n.id));
-  }, oe = () => {
-    B.confirm({
+      V();
+  }), Ne = () => {
+    k("annotation"), ae((i == null ? void 0 : i.id) || (l == null ? void 0 : l.id));
+  }, Ce = () => {
+    Y.confirm({
       title: "确认要删除此数据吗？",
       content: "删除后不可恢复，请慎重操作",
       onOk: () => {
-        !t || !a || !(i != null && i.id) || d == null || d.delDatasetsIdVersionsVidSamplesId(t, a, i.id).then(() => {
-          U.success("删除成功"), f(!1), J();
-        }).catch(() => U.error("删除失败"));
+        !a || !t || !(i != null && i.id) || o == null || o.delDatasetsIdVersionsVidSamplesId(a, t, i.id).then(() => {
+          _.success("删除成功"), C(!1), ne();
+        }).catch(() => _.error("删除失败"));
       }
     });
   };
-  return /* @__PURE__ */ e(B, {
+  return /* @__PURE__ */ e(Y, {
     title: /* @__PURE__ */ e("div", {
       className: "vlm-sample-detail-title",
       children: /* @__PURE__ */ e("span", {
-        children: u === "preview" ? "详情" : "标注"
+        children: v === "preview" ? "详情" : "标注"
       })
     }),
     width: "100%",
     wrapClassName: "vlm-modal-wrap",
     className: "vlm-modal",
-    open: r,
+    open: n,
     destroyOnClose: !0,
-    onCancel: () => re(),
+    onCancel: () => ge(),
     maskClosable: !1,
     centered: !0,
     footer: null,
-    children: u === "preview" ? /* @__PURE__ */ s(b, {
+    children: v === "preview" ? /* @__PURE__ */ s(N, {
       children: [/* @__PURE__ */ e("div", {
         className: "vlm-detail",
         children: /* @__PURE__ */ s("div", {
           className: "detail-content",
-          children: [/* @__PURE__ */ e(h, {
+          children: [/* @__PURE__ */ e(f, {
             className: "select-prev",
-            icon: /* @__PURE__ */ e(H, {
+            icon: /* @__PURE__ */ e(ie, {
               type: "icon-zuojiantou"
             }),
             type: "text",
-            disabled: !g,
+            disabled: !p,
             onClick: () => F()
           }), /* @__PURE__ */ e("div", {
             className: "select-main",
-            ref: ie,
-            children: i && /* @__PURE__ */ e(Ne, {
+            ref: ce,
+            children: i && /* @__PURE__ */ e(Ve, {
               sample: i,
               labelDefinitions: [],
               hideRightSide: !0,
               readOnly: !0,
-              ocrRecognize: () => N(void 0, null, function* () {
+              ocrRecognize: () => g(void 0, null, function* () {
                 return [];
               })
             })
-          }), S && /* @__PURE__ */ s(b, {
-            children: [/* @__PURE__ */ s("div", {
-              className: "qa-preview-container",
+          }), he ? /* @__PURE__ */ e("div", {
+            className: "vqa-preview-panel",
+            children: /* @__PURE__ */ e(Qe, {
+              question: (B = (K = (E = (z = i == null ? void 0 : i.content) == null ? void 0 : z.conversations) == null ? void 0 : E[0]) == null ? void 0 : K.value) != null ? B : "",
+              answer: ($ = (U = (S = (G = i == null ? void 0 : i.content) == null ? void 0 : G.conversations) == null ? void 0 : S[1]) == null ? void 0 : U.value) != null ? $ : "",
+              readOnly: !0
+            })
+          }) : /* @__PURE__ */ s(N, {
+            children: [R && /* @__PURE__ */ s(N, {
               children: [/* @__PURE__ */ s("div", {
-                className: "qa-preview-header",
-                children: [/* @__PURE__ */ e("div", {
-                  className: "qa-preview-title",
-                  children: "QA预览"
-                }), /* @__PURE__ */ e(G, {
-                  children: le && /* @__PURE__ */ e(fe, {
-                    indicator: /* @__PURE__ */ e(he, {
-                      spin: !0,
-                      style: {
-                        fontSize: 16
-                      }
+                className: "qa-preview-container",
+                children: [/* @__PURE__ */ s("div", {
+                  className: "qa-preview-header",
+                  children: [/* @__PURE__ */ e("div", {
+                    className: "qa-preview-title",
+                    children: "QA预览"
+                  }), /* @__PURE__ */ e(Z, {
+                    children: ue && /* @__PURE__ */ e(Me, {
+                      indicator: /* @__PURE__ */ e(xe, {
+                        spin: !0,
+                        style: {
+                          fontSize: 16
+                        }
+                      })
                     })
-                  })
+                  })]
+                }), /* @__PURE__ */ e("div", {
+                  className: "qa-preview-content",
+                  children: (() => {
+                    var H, J;
+                    const r = ((H = fe[0]) == null ? void 0 : H.qas) || [];
+                    return !(((J = i == null ? void 0 : i.labels) == null ? void 0 : J.length) > 0) || r.length === 0 ? /* @__PURE__ */ e("div", {
+                      className: "result-empty",
+                      children: "暂无结果"
+                    }) : /* @__PURE__ */ s(N, {
+                      children: [/* @__PURE__ */ e(ke, {
+                        activeKey: String(L),
+                        onChange: (m) => A(Number(m)),
+                        items: r.map((m, W) => ({
+                          key: String(W),
+                          label: m.name || `模版${W + 1}`
+                        }))
+                      }), /* @__PURE__ */ e("div", {
+                        className: "qa-preview-result",
+                        children: (() => {
+                          const m = r[L];
+                          return m ? /* @__PURE__ */ s("div", {
+                            className: "result-item-container",
+                            children: [/* @__PURE__ */ s("div", {
+                              className: "result-item-content",
+                              children: [/* @__PURE__ */ e("div", {
+                                className: "label",
+                                children: "问题"
+                              }), /* @__PURE__ */ e("div", {
+                                className: "value question",
+                                children: m.q
+                              })]
+                            }), /* @__PURE__ */ s("div", {
+                              className: "result-item-content",
+                              children: [/* @__PURE__ */ e("div", {
+                                className: "label",
+                                children: "回答"
+                              }), /* @__PURE__ */ e("div", {
+                                className: "value answer",
+                                children: m.a
+                              })]
+                            })]
+                          }) : /* @__PURE__ */ e("div", {
+                            className: "result-empty",
+                            children: "暂无结果"
+                          });
+                        })()
+                      })]
+                    });
+                  })()
                 })]
               }), /* @__PURE__ */ e("div", {
-                className: "qa-preview-content",
-                children: (() => {
-                  var z, E;
-                  const l = ((z = ne[0]) == null ? void 0 : z.qas) || [];
-                  return !(((E = i == null ? void 0 : i.labels) == null ? void 0 : E.length) > 0) || l.length === 0 ? /* @__PURE__ */ e("div", {
-                    className: "result-empty",
-                    children: "暂无结果"
-                  }) : /* @__PURE__ */ s(b, {
-                    children: [/* @__PURE__ */ e(ue, {
-                      activeKey: String(L),
-                      onChange: (m) => R(Number(m)),
-                      items: l.map((m, K) => ({
-                        key: String(K),
-                        label: m.name || `模版${K + 1}`
-                      }))
-                    }), /* @__PURE__ */ e("div", {
-                      className: "qa-preview-result",
-                      children: (() => {
-                        const m = l[L];
-                        return m ? /* @__PURE__ */ s("div", {
-                          className: "result-item-container",
-                          children: [/* @__PURE__ */ s("div", {
-                            className: "result-item-content",
-                            children: [/* @__PURE__ */ e("div", {
-                              className: "label",
-                              children: "问题"
-                            }), /* @__PURE__ */ e("div", {
-                              className: "value question",
-                              children: m.q
-                            })]
-                          }), /* @__PURE__ */ s("div", {
-                            className: "result-item-content",
-                            children: [/* @__PURE__ */ e("div", {
-                              className: "label",
-                              children: "回答"
-                            }), /* @__PURE__ */ e("div", {
-                              className: "value answer",
-                              children: m.a
-                            })]
-                          })]
-                        }) : /* @__PURE__ */ e("div", {
-                          className: "result-empty",
-                          children: "暂无结果"
-                        });
-                      })()
-                    })]
-                  });
-                })()
+                className: "qa-preview-collapsed",
+                children: /* @__PURE__ */ e(f, {
+                  type: "link",
+                  icon: /* @__PURE__ */ e(be, {}),
+                  disabled: P,
+                  onClick: () => b(!1)
+                })
               })]
-            }), /* @__PURE__ */ e("div", {
+            }), !R && /* @__PURE__ */ e("div", {
               className: "qa-preview-collapsed",
-              children: /* @__PURE__ */ e(h, {
+              children: /* @__PURE__ */ e(f, {
                 type: "link",
-                icon: /* @__PURE__ */ e(ve, {}),
-                disabled: j,
-                onClick: () => x(!1)
+                icon: /* @__PURE__ */ e(Ie, {}),
+                disabled: P,
+                onClick: () => b(!0)
               })
             })]
-          }), !S && /* @__PURE__ */ e("div", {
-            className: "qa-preview-collapsed",
-            children: /* @__PURE__ */ e(h, {
-              type: "link",
-              icon: /* @__PURE__ */ e(pe, {}),
-              disabled: j,
-              onClick: () => x(!0)
-            })
-          }), /* @__PURE__ */ e(h, {
+          }), /* @__PURE__ */ e(f, {
             className: "select-next",
-            icon: /* @__PURE__ */ e(H, {
+            icon: /* @__PURE__ */ e(ie, {
               type: "icon-youjiantou"
             }),
             type: "text",
-            disabled: !g,
-            onClick: () => P()
+            disabled: !p,
+            onClick: () => T()
           })]
         })
       }), /* @__PURE__ */ e("div", {
         className: "vlm-mark-action",
-        children: /* @__PURE__ */ s(G, {
-          children: [/* @__PURE__ */ e(h, {
+        children: /* @__PURE__ */ s(Z, {
+          children: [/* @__PURE__ */ e(f, {
             type: "primary",
-            onClick: de,
+            onClick: Ne,
             children: "标注"
-          }), D && D.delete && /* @__PURE__ */ e(h, {
+          }), D && D.delete && /* @__PURE__ */ e(f, {
             danger: !0,
             type: "ghost",
-            onClick: oe,
+            onClick: Ce,
             children: "删除数据"
           })]
         })
       })]
     }) : (
       // 标注模式：使用VLMDetail组件
-      q && /* @__PURE__ */ e(Ce, {
+      q && /* @__PURE__ */ e(Le, {
         ref: M,
-        visible: r,
-        selectSampleId: Z,
-        datasetId: t,
-        versionId: a,
+        visible: n,
+        selectSampleId: re,
+        datasetId: a,
+        versionId: t,
+        vlmMarkTmpl: h,
         onClose: () => {
-          Q();
+          V();
         },
         onSave: q,
-        externalFilterMode: W,
-        onFilterModeChange: X,
-        externalSamples: Y,
-        externalPagination: _,
-        externalCurrentIndex: ee
+        externalFilterMode: te,
+        onFilterModeChange: le,
+        externalSamples: se,
+        externalPagination: oe,
+        externalCurrentIndex: de
       })
     )
   });
 };
 export {
-  Oe as default
+  Se as default
 };