@glodon-aiot/dataset-annotation 3.9.0-beta.9 → 3.10.0-alpha.11

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

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

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

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