@glodon-aiot/dataset-annotation 3.18.0-alpha.11 → 3.18.0-alpha.13

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

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