Update image input

This commit is contained in:
shanexi
2024-10-14 17:32:47 +08:00
parent d2573428f5
commit 261620ea3a
2 changed files with 17 additions and 2 deletions
+2 -1
View File
@@ -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": {
+15 -1
View File
@@ -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" },
];
}
}
},
});