Update image input

2024-10-14 17:32:47 +08:00
parent d2573428f5
commit 261620ea3a
2 changed files with 17 additions and 2 deletions
@@ -35,4 +35,18 @@ app.registerExtension({
 			}
    });
  },
-});
+});app.registerExtension({
+  name: "Comfy.ShellAgent.UploadImage",
+  async beforeRegisterNodeDef(nodeType, nodeData, app) {
+    if (nodeData.name === "ShellAgentPluginInputImage") {
+      if (
+        nodeData?.input?.required?.default_value?.[1]?.image_upload === true
+      ) {
+        nodeData.input.required.upload = [
+          "IMAGEUPLOAD",
+          { widget: "default_value" },
+        ];
+      }
+    }
+  },
+});