@glodon-aiot/dataset-annotation 3.14.0-alpha.33 → 3.14.0-alpha.34

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

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