From 31bc856aa3c5508892eadbc557592b61e0e8ba56 Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Wed, 15 Nov 2023 20:22:19 +0000
Subject: [PATCH] fix imageupload widget requiring a specific name

---
 web/extensions/core/groupNode.js | 3 +++
 web/scripts/widgets.js           | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/web/extensions/core/groupNode.js b/web/extensions/core/groupNode.js
index af06aede1..1a02fa360 100644
--- a/web/extensions/core/groupNode.js
+++ b/web/extensions/core/groupNode.js
@@ -189,6 +189,9 @@ function buildNodeDef(config, nodeName, defs, source = "workflow") {
 			if (inputName === "seed" || inputName === "noise_seed") {
 				inputDef = [...inputDef];
 				inputDef[1] = { control_after_generate: `${prefix} control_after_generate`, ...inputDef[1] };
+			} else if (inputDef[0] === "IMAGEUPLOAD") {
+				inputDef = [...inputDef];
+				inputDef[1] = { widget: `${prefix} ${inputDef[1]?.widget ?? "image"}`, ...inputDef[1] };
 			}
 			newDef.input.required[name] = inputDef;
 		}
diff --git a/web/scripts/widgets.js b/web/scripts/widgets.js
index 9bb04acf4..837a39271 100644
--- a/web/scripts/widgets.js
+++ b/web/scripts/widgets.js
@@ -376,7 +376,7 @@ export const ComfyWidgets = {
 		return { widget: node.addWidget("combo", inputName, defaultValue, () => {}, { values: type }) };
 	},
 	IMAGEUPLOAD(node, inputName, inputData, app) {
-		const imageWidget = node.widgets.find((w) => w.name === "image");
+		const imageWidget = node.widgets.find((w) => w.name === (inputData[1]?.widget ?? "image"));
 		let uploadWidget;
 
 		function showImage(name) {