fix: widget index

myshell-ai · Sep 20, 2024 · 4e2d43c · 4e2d43c
1 parent 516a028
commit 4e2d43c
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/web/apps/web/src/stores/workflow/utils/data-transformer.ts b/web/apps/web/src/stores/workflow/utils/data-transformer.ts
@@ -216,19 +216,13 @@ export const genReactFlow: (
 
     // widget节点
     if (workflow.blocks) {
-      const widgetIndexMap = new Map<string, number>();
       if (isArray(workflow.blocks)) {
         workflow.blocks.forEach(block => {
           const id = block.name as NodeId;
           const type = NodeTypeEnum.widget;
           const widgetName = (block as any).widget_class_name;
-          let widgetIndex: number;
-          if (widgetIndexMap.get(widgetName)) {
-            widgetIndex = widgetIndexMap.get(widgetName) || 1;
-          } else {
-            widgetIndex = 1;
-            widgetIndexMap.set(widgetName, widgetIndex);
-          }
+          const widgetIndex =
+            nodes.filter(item => item.data.name === widgetName).length + 1;
           const idx = comfyui.nodes?.findIndex(
             (item: any) => `node_${item.id}` === block.name,
           );