@glodon-aiot/dataset-annotation 3.2.2-beta.3 → 3.2.10-beta.1

package/dist/es/components/OcrAnnotation/PreLabelOCR/PreLabelModal.mjs

@@ -0,0 +1,281 @@
+var x = (I, c, i) => new Promise((V, y) => {
+  var T = (d) => {
+    try {
+      m(i.next(d));
+    } catch (N) {
+      y(N);
+    }
+  }, t = (d) => {
+    try {
+      m(i.throw(d));
+    } catch (N) {
+      y(N);
+    }
+  }, m = (d) => d.done ? V(d.value) : Promise.resolve(d.value).then(T, t);
+  m((i = i.apply(I, c)).next());
+});
+import { jsx as a, jsxs as v } from "react/jsx-runtime";
+import { useState as f, useContext as Y, useEffect as O } from "react";
+import { Form as h, Modal as Z, Button as C, Select as E, Tooltip as G, Tag as ee, InputNumber as te, Space as z, Spin as ae } from "antd";
+import { LoadingOutlined as ne } from "@ant-design/icons";
+import "./style.less.mjs";
+import { detectionPreService as le, ocrPreService as se } from "../constant.mjs";
+import { PreLabelTaskStatus as p } from "@glodon-aiot/apis";
+import ie from "../../../DatasetContext/index.mjs";
+const re = [{
+  index: 0,
+  key: "PENDING",
+  color: "#0F64FF",
+  name: "待确认"
+}, {
+  index: 1,
+  key: "STARTED",
+  color: "#0F64FF",
+  name: "进行中"
+}, {
+  index: 2,
+  key: "SUCCESS",
+  color: "#2DC159",
+  name: "已完成"
+}], ve = (I) => {
+  var j;
+  const {
+    datasetId: c,
+    versionId: i,
+    isVisible: V,
+    onClose: y,
+    setIsTaskChange: T,
+    record: t,
+    datasetType: m
+  } = I, [d, N] = f([]), [A, M] = f([]), [S] = h.useForm(), [l, L] = f(), [F, q] = f(), [b, R] = f(""), k = h.useWatch("service", S), [B, _] = f(!1), {
+    services: {
+      cvforceDatahub: r,
+      cvforceModelService: g
+    },
+    featuresConfig: {
+      labelManager: oe
+    } = {
+      labelManager: !1
+    }
+  } = Y(ie), J = m === "detection" ? le : se, P = () => {
+    y();
+  };
+  O(() => {
+    $(), H();
+  }, []), O(() => {
+    t && L(t), t != null && t.lastTaskResult && q(JSON.parse(t == null ? void 0 : t.lastTaskResult));
+  }, [t]), O(() => {
+    var e, n, s, o, u;
+    S.setFieldsValue({
+      service: {
+        value: (e = l == null ? void 0 : l.params) == null ? void 0 : e.serviceId,
+        label: (n = l == null ? void 0 : l.params) == null ? void 0 : n.serviceName
+      },
+      data: (s = l == null ? void 0 : l.params) == null ? void 0 : s.data,
+      confidence: (u = (o = l == null ? void 0 : l.params) == null ? void 0 : o.confidence) != null ? u : 0.6
+    }), R((l == null ? void 0 : l.state) || "PENDING");
+  }, [l]), O(() => {
+    !k || !k.value || m !== "ocr" && Promise.all([U(), W(k.value)]).then(([e, n]) => {
+      if (!e)
+        return;
+      const s = e.filter((o) => {
+        if (!(n != null && n.find((u) => u === o)))
+          return o;
+      });
+      _(!!s.length), S.validateFields(["service"]);
+    });
+  }, [k]);
+  const U = () => {
+    if (!(!c || !i))
+      return r == null ? void 0 : r.getDatasetsIdVersionsVidLabelcount(c, i).then((e) => e.filter((s) => s.label !== "nocode").map((s) => s.label));
+  }, W = (e) => g == null ? void 0 : g.getModelServiceById(e).then((n) => {
+    const s = n.deployOptions.confidence;
+    return s == null ? void 0 : s.map((u) => u.label);
+  }), $ = () => {
+    g == null || g.openModelServices({
+      signName: J,
+      state: "Running"
+    }).then((e) => {
+      const s = (e || []).map((o) => {
+        var u;
+        return {
+          label: o.name,
+          value: o.id,
+          confidence: ((u = o.deployOptions.confidence) == null ? void 0 : u.map((X) => X.label)) || []
+        };
+      });
+      N(s);
+    });
+  }, H = () => {
+    !c || !i || r == null || r.getDatasetsIdVersionsVidSamples(c, i, {
+      labels: "nocode",
+      pageNo: 1,
+      pageSize: 1
+    }).then((e) => {
+      const n = e.recordList || [];
+      M([{
+        label: "未标注",
+        value: "nolabel",
+        disabled: !n.length
+      }]);
+    });
+  }, w = () => {
+    !c || !i || !(l != null && l.taskId) || r == null || r.delPreLabelTask(c, i, l.taskId).then((e) => {
+      T((n) => !n), R("PENDING");
+    });
+  }, K = () => {
+    w();
+  }, Q = () => {
+    w(), S.resetFields(), L(void 0);
+  }, D = (e) => {
+    if (!c || !i || !e.service)
+      return;
+    const n = {
+      serviceId: e.service.value,
+      serviceName: e.service.label,
+      data: "nolabel",
+      confidence: e.confidence
+    };
+    r == null || r.postPreLabelTask(c, i, n).then((s) => {
+      T((o) => !o);
+    });
+  };
+  return /* @__PURE__ */ a(Z, {
+    title: "批量预标注",
+    wrapClassName: "batch-ocr-modal-wrap",
+    className: "batch-ocr-modal",
+    open: V,
+    destroyOnClose: !0,
+    onCancel: P,
+    maskClosable: !1,
+    centered: !0,
+    footer: b === p.Pending ? [/* @__PURE__ */ a(C, {
+      onClick: P,
+      children: "取消"
+    }, "cancel"), /* @__PURE__ */ a(C, {
+      type: "primary",
+      htmlType: "submit",
+      form: "preLabelForm",
+      onClick: D,
+      children: "确定"
+    }, "confirm")] : (t == null ? void 0 : t.state) === p.Started ? [/* @__PURE__ */ a(C, {
+      onClick: K,
+      children: "终止"
+    }, "abort"), /* @__PURE__ */ a(C, {
+      onClick: P,
+      children: "返回"
+    }, "cancel")] : [/* @__PURE__ */ a(C, {
+      onClick: Q,
+      children: "新建任务"
+    }, "abort"), /* @__PURE__ */ a(C, {
+      onClick: P,
+      children: "返回"
+    }, "cancel")],
+    children: /* @__PURE__ */ v(h, {
+      onFinish: D,
+      name: "preLabelForm",
+      labelAlign: "right",
+      labelCol: {
+        style: {
+          width: 80
+        }
+      },
+      form: S,
+      children: [/* @__PURE__ */ a(h.Item, {
+        label: "选择服务",
+        name: "service",
+        rules: [{
+          required: !0
+        }, {
+          validator: (e, n) => x(void 0, null, function* () {
+            const s = S.getFieldValue("service");
+            if (!s.value || !s.label)
+              return Promise.reject(new Error("请选择服务"));
+          })
+        }, {
+          validator: (e, n) => x(void 0, null, function* () {
+            if (B && m === "detection")
+              return Promise.reject(new Error("所选服务与当前数据集标签不符"));
+          })
+        }],
+        children: /* @__PURE__ */ a(E, {
+          labelInValue: !0,
+          disabled: (t == null ? void 0 : t.state) === p.Started,
+          children: d.map((e) => /* @__PURE__ */ a(E.Option, {
+            value: e.value,
+            disabled: e.disabled,
+            children: /* @__PURE__ */ a(G, {
+              title: e.confidence.length ? e.confidence.map((n) => /* @__PURE__ */ a(ee, {
+                children: n
+              }, n)) : null,
+              overlayClassName: "service-tag-tooltip",
+              children: e.label
+            })
+          }, e.value))
+        })
+      }), m === "detection" && /* @__PURE__ */ a(h.Item, {
+        label: "置信度",
+        name: "confidence",
+        initialValue: 0.6,
+        children: /* @__PURE__ */ a(te, {
+          max: 1,
+          min: 0,
+          precision: 2,
+          step: 0.01,
+          disabled: (t == null ? void 0 : t.state) === p.Started
+        })
+      }), /* @__PURE__ */ a(h.Item, {
+        label: "选择数据",
+        name: "data",
+        rules: [{
+          required: !0,
+          message: "请选择数据"
+        }],
+        initialValue: "nolabel",
+        children: /* @__PURE__ */ a(E, {
+          disabled: (t == null ? void 0 : t.state) === p.Started,
+          children: A.map((e) => /* @__PURE__ */ a(E.Option, {
+            value: e.value,
+            disabled: e.disabled,
+            children: /* @__PURE__ */ a(G, {
+              title: e.disabled ? "该数据集下不存在未标注的数据" : "当前数据集版本下所有未标注的数据",
+              children: e.label
+            })
+          }, e.value))
+        })
+      }), /* @__PURE__ */ a(h.Item, {
+        label: "标注状态",
+        children: /* @__PURE__ */ v(z, {
+          className: "mark-status",
+          children: [b === p.Started ? /* @__PURE__ */ a(ae, {
+            indicator: /* @__PURE__ */ a(ne, {
+              style: {
+                fontSize: 12
+              },
+              spin: !0
+            })
+          }) : /* @__PURE__ */ a("span", {
+            className: `status-tip status-tip-${b || "PENDING"}`
+          }), /* @__PURE__ */ a("span", {
+            children: ((j = re.find((e) => e.key == b)) == null ? void 0 : j.name) || "待确认"
+          }), (b === p.Started || b === p.Success) && /* @__PURE__ */ v(z, {
+            style: {
+              marginLeft: 14
+            },
+            size: 14,
+            children: [/* @__PURE__ */ v("span", {
+              children: [F.successCount || 0, "/", F.allCount || 0]
+            }), /* @__PURE__ */ v("span", {
+              children: ["已标注数据：", F.successCount || 0, " 张"]
+            })]
+          }), b === p.Success && /* @__PURE__ */ v("span", {
+            children: ["，失败：", F.failCount || 0, " 张"]
+          })]
+        })
+      })]
+    })
+  });
+};
+export {
+  ve as default
+};

package/dist/es/components/OcrAnnotation/PreLabelOCR/index.mjs

@@ -0,0 +1,44 @@
+import { jsxs as m, jsx as e } from "react/jsx-runtime";
+import { useState as c } from "react";
+import { Button as f, Spin as b } from "antd";
+import p from "./PreLabelModal.mjs";
+import { LoadingOutlined as h } from "@ant-design/icons";
+import "./style.less.mjs";
+import { PreLabelTaskStatus as u } from "@glodon-aiot/apis";
+const B = (o) => {
+  const {
+    datasetId: i,
+    versionId: r,
+    disabled: l,
+    setIsTaskChange: n,
+    preLabelTask: t,
+    datasetType: d
+  } = o, [a, s] = c(!1);
+  return /* @__PURE__ */ m("div", {
+    className: "batch-btn-panel",
+    children: [/* @__PURE__ */ e(f, {
+      className: "batch-mark-btn",
+      onClick: () => s(!0),
+      disabled: l,
+      children: "批量预标注"
+    }), (t == null ? void 0 : t.state) === u.Started && /* @__PURE__ */ e(b, {
+      indicator: /* @__PURE__ */ e(h, {
+        style: {
+          fontSize: 15,
+          marginLeft: 4
+        }
+      })
+    }), a && /* @__PURE__ */ e(p, {
+      datasetId: i,
+      versionId: r,
+      isVisible: a,
+      onClose: () => s(!1),
+      setIsTaskChange: n,
+      record: t,
+      datasetType: d
+    })]
+  });
+};
+export {
+  B as default
+};

package/dist/es/components/OcrAnnotation/PreLabelOCR/style.less.mjs

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

package/dist/es/components/OcrAnnotation/constant.mjs

@@ -0,0 +1,5 @@
+const e = "ocr-det", c = "obj-det";
+export {
+  c as detectionPreService,
+  e as ocrPreService
+};