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

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

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