From 261620ea3a3825f5ce4b98e884c7c4cd17eac539 Mon Sep 17 00:00:00 2001
From: shanexi <shane@myshell.ai>
Date: Mon, 14 Oct 2024 17:32:42 +0800
Subject: [PATCH] Update image input

---
 comfy-nodes/input_image.py |  3 ++-
 web/shellagent.js          | 16 +++++++++++++++-
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/comfy-nodes/input_image.py b/comfy-nodes/input_image.py
index cb24158..3fa5a8b 100755
--- a/comfy-nodes/input_image.py
+++ b/comfy-nodes/input_image.py
@@ -20,7 +20,8 @@ class ShellAgentPluginInputImage:
                     {"multiline": False, "default": "input_image"},
                 ),
                 "default_value": (
-                    "STRING", {"image_upload": True, "default": files[0] if len(files) else ""},
+                    # "STRING", {"image_upload": True, "default": files[0] if len(files) else ""},
+                    sorted(files), {"image_upload": True}
                 ),
             },
             "optional": {
diff --git a/web/shellagent.js b/web/shellagent.js
index bdc31be..ee3111b 100644
--- a/web/shellagent.js
+++ b/web/shellagent.js
@@ -35,4 +35,18 @@ app.registerExtension({
 			}
     });
   },
-});
\ No newline at end of file
+});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" },
+        ];
+      }
+    }
+  },
+});