From 27428d2cb7e59001f4fb54e3e143ffd0163d9cd5 Mon Sep 17 00:00:00 2001 From: Chris Date: Mon, 11 Sep 2023 15:35:10 +1000 Subject: [PATCH] fix reload of defaultInput widget --- comfy_extras/ui_decorator.py | 3 ++- web/extensions/core/widgetInputs.js | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/comfy_extras/ui_decorator.py b/comfy_extras/ui_decorator.py index c5bd339a1..03f84506a 100644 --- a/comfy_extras/ui_decorator.py +++ b/comfy_extras/ui_decorator.py @@ -27,7 +27,7 @@ def ui_signal(signals:str|list[str]): returns_tuple = returns_tuple[:-len(signals)] for i,key in enumerate(signals): - returns['ui']['key'] = popped_returns[i] + returns_ui[key] = popped_returns[i] return { "ui":returns_ui, "result": returns_tuple } clazz._ui_signal_decorated_function = _ui_signal_decorated_function @@ -35,6 +35,7 @@ def ui_signal(signals:str|list[str]): clazz.OUTPUT_NODE = True clazz.UI_OUTPUT = clazz.UI_OUTPUT+"," if hasattr(clazz, 'UI_OUTPUT') else "" clazz.UI_OUTPUT += ",".join(signals) + return clazz return decorator diff --git a/web/extensions/core/widgetInputs.js b/web/extensions/core/widgetInputs.js index 606605f0a..09179a57b 100644 --- a/web/extensions/core/widgetInputs.js +++ b/web/extensions/core/widgetInputs.js @@ -158,7 +158,7 @@ app.registerExtension({ if (this.inputs) { for (const input of this.inputs) { - if (input.widget && !input.widget.config[1]?.forceInput) { + if (input.widget && !input.widget.config[1]?.forceInput && !input.widget.config[1]?.defaultInput) { const w = this.widgets.find((w) => w.name === input.widget.name); if (w) { hideWidget(this, w);