@glodon-aiot/dataset-annotation 3.11.1-shapshot.1765331459 → 3.13.0-alpha.28

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

@@ -0,0 +1,455 @@
+var Ge = Object.defineProperty, $e = Object.defineProperties;
+var Pe = Object.getOwnPropertyDescriptors;
+var Ie = Object.getOwnPropertySymbols;
+var je = Object.prototype.hasOwnProperty, Be = Object.prototype.propertyIsEnumerable;
+var be = (w, u, o) => u in w ? Ge(w, u, { enumerable: !0, configurable: !0, writable: !0, value: o }) : w[u] = o, U = (w, u) => {
+  for (var o in u || (u = {}))
+    je.call(u, o) && be(w, o, u[o]);
+  if (Ie)
+    for (var o of Ie(u))
+      Be.call(u, o) && be(w, o, u[o]);
+  return w;
+}, P = (w, u) => $e(w, Pe(u));
+var z = (w, u, o) => new Promise((x, g) => {
+  var J = (q) => {
+    try {
+      r(o.next(q));
+    } catch (E) {
+      g(E);
+    }
+  }, R = (q) => {
+    try {
+      r(o.throw(q));
+    } catch (E) {
+      g(E);
+    }
+  }, r = (q) => q.done ? x(q.value) : Promise.resolve(q.value).then(J, R);
+  r((o = o.apply(w, u)).next());
+});
+import { jsx as l, jsxs as B, Fragment as Ee } from "react/jsx-runtime";
+import { Modal as ke, Space as Ae, Button as te, message as N, Tabs as de, Spin as Fe } from "antd";
+import { useContext as Qe, useRef as j, useState as m, useEffect as W } from "react";
+import "./style.less.mjs";
+import Ke from "../../DataEmpty/index.mjs";
+import Oe from "../../../DatasetContext/index.mjs";
+import He from "./AttributeDetail/index.mjs";
+import We from "./QADetail/index.mjs";
+import Je from "../../VLMAnnotation/index.mjs";
+import Xe from "axios";
+const {
+  TabPane: K
+} = de, lt = (w) => {
+  const {
+    visible: u,
+    selectSampleId: o,
+    datasetId: x,
+    versionId: g,
+    // vmlMarkTmpl,
+    onClose: J,
+    onSave: R
+  } = w, {
+    services: {
+      cvforceDatahub: r
+    },
+    token: q,
+    apiUrl: E
+  } = Qe(Oe), F = j(null);
+  j(null), j({});
+  const [i, I] = m(), C = j(void 0), [Ye, fe] = m(0), [y, G] = m([]), [ne, M] = m(-1), [f, T] = m({
+    current: 1,
+    pageSize: 100
+    // 与主列表保持一致
+  }), [ue, ge] = m(0), [he, pe] = m(0), [b, ie] = m(() => o ? "" : "nocode"), [me, O] = m(), [oe, xe] = m([]), [h, X] = m("textAnnotation"), [Y, Q] = m(), [Z, L] = m(!1), _ = j(0), re = j(0), [se, De] = m(!1), [ye, ee] = m(!1), [ae, ce] = m(!1), [Se, Ce] = m(!1), $ = j(null), D = j(null), [Ve, Le] = m(() => Date.now()), [ze, Ne] = m(() => Date.now());
+  W(() => {
+    u && (!x || !g || r == null || r.getDatasetsIdVersionsVidConfigs(x, g).then((e) => {
+      xe(e || []), (e == null ? void 0 : e.length) > 0 ? X("textAnnotation") : X("attributeDefinition");
+    }));
+  }, [u]), W(() => {
+    const e = f.current || 1, t = f.pageSize || 100, n = f.type;
+    if (!x || !g)
+      return;
+    const a = ++_.current;
+    ce(!0);
+    const p = y || [], s = o || me;
+    r == null || r.getDatasetsIdVersionsVidSamples(x, g, {
+      labels: b ? [b] : void 0,
+      pageNo: e,
+      pageSize: t
+    }).then((k) => {
+      if (a !== _.current)
+        return;
+      const d = k.recordList || [], c = e === 1 && !n ? d : n === "prev" ? [...d, ...p] : n === "next" ? [...p, ...d] : d;
+      G(c), T(P(U({}, f), {
+        current: e,
+        pageSize: t,
+        total: k.totalResults,
+        type: void 0
+      }));
+      let v = 0, S = c[0];
+      if (n === "next" && d.length)
+        v = p.length, S = d[0];
+      else if (n === "prev" && d.length)
+        v = d.length - 1, S = d[d.length - 1];
+      else if (s) {
+        const A = c.findIndex((H) => H.id === s);
+        if (A !== -1)
+          v = A, S = c[A];
+        else {
+          r == null || r.getDatasetsIdVersionsVidSsamplesSid(x, g, s).then((H) => {
+            a === _.current && H && (G([H, ...c]), I(H), M(0), fe(c.length + 1), O(void 0));
+          });
+          return;
+        }
+      }
+      S && (I(S), M(v), fe(c.length), me && O(void 0));
+    }).finally(() => {
+      a === _.current && ce(!1);
+    }), V();
+  }, [f.current, b, x, g]);
+  const V = () => {
+    !x || !g || r == null || r.getDatasetsIdVersionsVidLabelcount(x, g).then((e) => {
+      const t = e.find((a) => a.id === "allcodeid"), n = e.find((a) => a.id === "nocodeid");
+      ge(t ? t.count : 0), pe(n ? n.count : 0);
+    });
+  }, le = () => {
+    y && G((e) => {
+      var k, d, c, v;
+      const t = C.current || i, n = [...e];
+      let a;
+      if (t) {
+        const S = n.findIndex((A) => A.id === t.id);
+        S > -1 && n.splice(S, 1, t);
+      }
+      const p = !t || (k = t.labels) != null && k.length && b === "hascode" || !((d = t.labels) != null && d.length) && b === "nocode" || b === "" ? ne + 1 : ne;
+      if (t && ((c = t.labels) != null && c.length && b === "nocode" || !((v = t.labels) != null && v.length) && b === "hascode")) {
+        const S = n.findIndex((A) => A.id === t.id);
+        S > -1 && (n.splice(S, 1), a = [...n], T((A) => P(U({}, A), {
+          total: A.total ? A.total - 1 : A.total
+        })));
+      }
+      const s = a || n;
+      if (p >= s.length) {
+        const S = (f.current - 1) * f.pageSize + p;
+        f.total && S >= f.total ? N.error("已经是最后一个了") : T({
+          current: f.current + 1,
+          pageSize: f.pageSize,
+          type: "next"
+        });
+      } else
+        setTimeout(() => {
+          M(p);
+          const S = s[p];
+          I(S);
+        }, 0);
+      return s;
+    });
+  }, ve = () => {
+    var s, k;
+    if (!y)
+      return;
+    const e = C.current || i, t = [...y];
+    let n;
+    if (e && y) {
+      const d = t.findIndex((c) => c.id === e.id);
+      if (d > -1 && t.splice(d, 1, e), (s = e.labels) != null && s.length && b === "nocode" || !((k = e.labels) != null && k.length) && b === "hascode") {
+        const c = t.findIndex((v) => v.id === e.id);
+        c > -1 && (t.splice(c, 1), n = [...t], T((v) => P(U({}, v), {
+          total: v.total ? v.total - 1 : v.total
+        })));
+      }
+    }
+    const a = n || t, p = ne - 1;
+    if (p < 0)
+      (f.current - 1) * f.pageSize + p < 0 ? N.error("已经是第一个了") : T({
+        current: f.current - 1,
+        pageSize: f.pageSize,
+        type: "prev"
+      });
+    else {
+      M(p);
+      const d = a[p];
+      I(d);
+    }
+    n && G(n);
+  };
+  W(() => {
+    if (!u || !x || !g || !i)
+      return;
+    const e = ++re.current;
+    Ce(!0), r == null || r.getDatasetsIdVersionsVidSsamplesSid(x, g, i.id).then((t) => {
+      e === re.current && I(t);
+    }).finally(() => {
+      e === re.current && Ce(!1);
+    });
+  }, [u, i == null ? void 0 : i.id]);
+  const Re = () => z(void 0, null, function* () {
+    h === "attributeDefinition" && $.current && !(yield $.current.checkDirtyAndConfirm()) || h === "qaGeneration" && D.current && !(yield D.current.checkDirtyAndConfirm()) || J();
+  });
+  W(() => {
+    F.current && (F.current.width = F.current.offsetWidth, F.current.height = F.current.offsetHeight);
+  }, [F.current]);
+  const qe = () => z(void 0, null, function* () {
+    if (Z)
+      return;
+    if (L(!0), h === "qaGeneration" && D.current && !(yield D.current.checkDirtyAndConfirm())) {
+      L(!1);
+      return;
+    }
+    const e = C.current || i;
+    if (!e) {
+      L(!1);
+      return;
+    }
+    console.log("onPrevClick sampleToUse", e), h === "textAnnotation" && Y ? (ee(!0), R(e).then((t) => {
+      t ? (Q(void 0), ve(), V()) : N.error("保存失败");
+    }).finally(() => {
+      ee(!1), L(!1);
+    })) : (ve(), L(!1));
+  }), Me = () => z(void 0, null, function* () {
+    if (Z)
+      return;
+    if (L(!0), h === "qaGeneration" && D.current && !(yield D.current.checkDirtyAndConfirm())) {
+      L(!1);
+      return;
+    }
+    const e = C.current || i;
+    if (!e) {
+      L(!1);
+      return;
+    }
+    console.log("onNextClick sampleToUse", e), h === "textAnnotation" && Y ? (ee(!0), R(e).then((t) => {
+      t ? (Q(void 0), le(), V()) : N.error("保存失败");
+    }).finally(() => {
+      ee(!1), L(!1);
+    })) : (le(), L(!1));
+  });
+  W(() => {
+    C.current = i;
+  }, [i]);
+  const Te = (e) => (console.log("ocrRecognize apiUrl", E), Xe.post(`${E}/aishop/v2/infer?serviceId=Actpkb4pi3vxg`, {
+    pages: [{
+      image: e,
+      no: 0
+    }]
+  }, {
+    headers: {
+      Authorization: `Bearer ${q}`
+    }
+  }).then((t) => (console.log("ocrRecognize res", t.data.data), t.data.data.pages[0].layout.filter((a) => a.content))).catch((t) => {
+    console.error("ocrRecognize error", t);
+  })), Ue = () => z(void 0, null, function* () {
+    const e = C.current || i;
+    !e || !x || !g || ke.confirm({
+      title: "确认要删除此数据吗？",
+      content: "删除后不可恢复，请慎重操作",
+      onOk: () => z(void 0, null, function* () {
+        console.log("handleDeleteClick currentSample", e);
+        try {
+          yield r == null ? void 0 : r.delDatasetsIdVersionsVidSamplesId(x, g, e.id), N.success("删除成功"), o ? J() : G((t) => {
+            var d;
+            const n = t.findIndex((c) => c.id === e.id);
+            if (n === -1)
+              return t;
+            const a = [...t.slice(0, n), ...t.slice(n + 1)], p = f.pageSize || 1, s = Math.max(((d = f.total) != null ? d : t.length) - 1, 0), k = a.length === 0 && s > f.current * p;
+            if (T((c) => U(P(U({}, c), {
+              total: s
+            }), k ? {
+              current: c.current + 1,
+              pageSize: p,
+              type: "next"
+            } : {
+              type: k ? "next" : void 0
+            })), a.length) {
+              const c = Math.min(n, a.length - 1);
+              M(c), I(a[c]);
+            } else
+              M(-1), I(void 0);
+            return a;
+          }), V();
+        } catch (t) {
+          N.error("删除失败");
+        }
+      })
+    });
+  }), we = () => z(void 0, null, function* () {
+    if (!x || !g || !(i != null && i.id))
+      return;
+    const e = yield r == null ? void 0 : r.getDatasetsIdVersionsVidSsamplesSid(x, g, i.id);
+    e && (I(e), C.current = e);
+  });
+  return /* @__PURE__ */ l(ke, {
+    title: "标注",
+    width: "100%",
+    wrapClassName: "vlm-modal-wrap",
+    className: "vlm-modal",
+    open: u,
+    destroyOnClose: !0,
+    onCancel: () => Re(),
+    maskClosable: !1,
+    keyboard: !1,
+    centered: !0,
+    footer: h !== "attributeDefinition" ? /* @__PURE__ */ B("div", {
+      style: {
+        display: "flex",
+        justifyContent: "space-between",
+        alignItems: "center"
+      },
+      children: [/* @__PURE__ */ l("div", {
+        style: {
+          flex: 1
+        }
+      }), /* @__PURE__ */ B(Ae, {
+        size: 24,
+        children: [!o && (y == null ? void 0 : y.length) > 0 && /* @__PURE__ */ l(te, {
+          type: "default",
+          disabled: Se || ae || ye || Z,
+          onClick: qe,
+          children: "上一张"
+        }, "prev"), !o && (y == null ? void 0 : y.length) > 0 && /* @__PURE__ */ l(te, {
+          type: "default",
+          disabled: Se || ae || ye || Z,
+          onClick: Me,
+          children: "下一张"
+        }, "next")]
+      }), /* @__PURE__ */ l("div", {
+        style: {
+          flex: 1,
+          display: "flex",
+          justifyContent: "flex-end"
+        },
+        children: /* @__PURE__ */ l(Ae, {
+          children: h === "textAnnotation" && (y == null ? void 0 : y.length) > 0 ? /* @__PURE__ */ B(Ee, {
+            children: [/* @__PURE__ */ l(te, {
+              disabled: !i,
+              type: "primary",
+              onClick: () => {
+                const e = C.current || i;
+                e && (console.log("save currentSample", e), R(e).then((t) => z(void 0, null, function* () {
+                  t ? (N.success("保存成功"), Q(void 0), o ? yield we() : le(), V()) : N.error("保存失败");
+                })));
+              },
+              children: "保存标注"
+            }, "save"), /* @__PURE__ */ l(te, {
+              type: "default",
+              onClick: Ue,
+              children: "删除数据"
+            })]
+          }) : null
+        })
+      })]
+    }) : null,
+    children: /* @__PURE__ */ B("div", {
+      className: "vlm-records-content",
+      children: [/* @__PURE__ */ l("div", {
+        className: "records-content-tabs-container",
+        children: !o && /* @__PURE__ */ B(de, {
+          className: "records-content-tabs",
+          activeKey: b,
+          onChange: (e) => z(void 0, null, function* () {
+            var n, a, p;
+            let t = !1;
+            if (!(h === "attributeDefinition" && $.current && (t = !!((a = (n = $.current).hasUnsavedChanges) != null && a.call(n)), !(yield $.current.checkDirtyAndConfirm()))) && !(h === "qaGeneration" && D.current && !(yield D.current.checkDirtyAndConfirm()))) {
+              if (h === "textAnnotation" && Y) {
+                const s = C.current || i;
+                if (s && R) {
+                  if (!(yield R(s))) {
+                    N.error("保存失败，已阻止切换筛选");
+                    return;
+                  }
+                  I(s), C.current = s, Q(void 0), V == null || V(), !o && b === "nocode" && ((p = s == null ? void 0 : s.labels) != null && p.length) && (G([]), I(s), M(-1), O(s.id), T(P(U({}, f), {
+                    current: 1,
+                    total: void 0,
+                    type: void 0
+                  })), ie("hascode"));
+                }
+              }
+              t && Le(Date.now()), Ne(Date.now()), h !== "attributeDefinition" && (ce(!0), G([]), I(void 0), M(-1), O(void 0)), T(P(U({}, f), {
+                current: 1,
+                total: void 0,
+                type: void 0
+              })), ie(e);
+            }
+          }),
+          children: [/* @__PURE__ */ l(K, {
+            tab: `全部（${ue}）`
+          }, ""), /* @__PURE__ */ l(K, {
+            tab: `未标注（${he}）`
+          }, "nocode"), /* @__PURE__ */ l(K, {
+            tab: `已标注（${ue - he}）`
+          }, "hascode")]
+        })
+      }), y != null && y.length ? /* @__PURE__ */ B("div", {
+        className: "sample-detail",
+        children: [/* @__PURE__ */ l("div", {
+          className: "content-mode-buttons",
+          children: /* @__PURE__ */ B(de, {
+            activeKey: h,
+            onChange: (e) => z(void 0, null, function* () {
+              var t;
+              if (!(h === "attributeDefinition" && $.current && !(yield $.current.checkDirtyAndConfirm())) && !(h === "qaGeneration" && D.current && !(yield D.current.checkDirtyAndConfirm()))) {
+                if (h === "textAnnotation" && R && Y && (C.current || i)) {
+                  const n = C.current || i;
+                  if (!(yield R(n))) {
+                    N.error("保存失败，已阻止切换页签");
+                    return;
+                  }
+                  I(n), C.current = n, Q(void 0), V == null || V(), !o && b === "nocode" && ((t = n == null ? void 0 : n.labels) != null && t.length) && (G([]), I(n), M(-1), O(n.id), T(P(U({}, f), {
+                    current: 1,
+                    total: void 0,
+                    type: void 0
+                  })), ie("hascode"));
+                }
+                e === "textAnnotation" && (i != null && i.id) && (yield we()), X(e);
+              }
+            }),
+            children: [/* @__PURE__ */ l(K, {
+              tab: "属性定义",
+              disabled: se
+            }, "attributeDefinition"), /* @__PURE__ */ l(K, {
+              tab: "文本标注",
+              disabled: se
+            }, "textAnnotation"), /* @__PURE__ */ l(K, {
+              tab: "QA生成",
+              disabled: se
+            }, "qaGeneration")]
+          })
+        }), h === "textAnnotation" && i && /* @__PURE__ */ l("div", {
+          className: "container",
+          children: /* @__PURE__ */ l(Je, {
+            sample: i,
+            labelDefinitions: oe,
+            onChange: (e) => {
+              console.log("VLMAnnotation onChange", e), C.current = e, Q(e);
+            },
+            ocrRecognize: Te
+          })
+        }), h === "attributeDefinition" && /* @__PURE__ */ l(He, {
+          ref: $,
+          datasetId: x,
+          versionId: g,
+          onAttributesChange: (e) => {
+            xe(e), e.length > 0 && oe.length === 0 && X("textAnnotation");
+          },
+          onLoadingChange: (e) => {
+            De(e);
+          }
+        }, `attribute-${Ve}`), h === "qaGeneration" && /* @__PURE__ */ l(We, {
+          ref: D,
+          sample: i,
+          labels: oe,
+          datasetId: x,
+          visionId: g
+        }, `qa-${ze}`)]
+      }) : ae ? /* @__PURE__ */ l("div", {
+        style: {
+          padding: "48px 0",
+          textAlign: "center"
+        },
+        children: /* @__PURE__ */ l(Fe, {})
+      }) : /* @__PURE__ */ l(Ke, {})]
+    })
+  });
+};
+export {
+  lt as default
+};

package/dist/es/components/VLMAnnotationDetail/MarkVLMBoxModal/style.less.mjs

@@ -0,0 +1,4 @@
+const t = "";
+export {
+  t as default
+};

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

@@ -0,0 +1,149 @@
+var N = (p, i, l) => new Promise((o, r) => {
+  var u = (e) => {
+    try {
+      s(l.next(e));
+    } catch (d) {
+      r(d);
+    }
+  }, h = (e) => {
+    try {
+      s(l.throw(e));
+    } catch (d) {
+      r(d);
+    }
+  }, s = (e) => e.done ? o(e.value) : Promise.resolve(e.value).then(u, h);
+  s((l = l.apply(p, i)).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 "./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) => {
+  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), {
+    services: {
+      cvforceDatahub: n
+    },
+    featuresConfig: {
+      datasetManager: k
+    } = {
+      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);
+  }, P = () => {
+    M.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("删除失败"));
+      }
+    });
+  };
+  return /* @__PURE__ */ C(M, {
+    title: /* @__PURE__ */ t("div", {
+      className: "vlm-sample-detail-title",
+      children: /* @__PURE__ */ t("span", {
+        children: "详情"
+      })
+    }),
+    width: "100%",
+    wrapClassName: "vlm-modal-wrap",
+    className: "vlm-modal",
+    open: i,
+    destroyOnClose: !0,
+    onCancel: () => O(),
+    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 [];
+            })
+          })
+        }), /* @__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: "删除数据"
+        })]
+      })
+    })]
+  });
+};
+export {
+  X as default
+};

package/dist/es/components/VLMAnnotationDetail/VlmModal/style.less.mjs

@@ -0,0 +1,4 @@
+const t = "";
+export {
+  t as default
+};