From b4398a09d366de3d12734ffae393263c8350a545 Mon Sep 17 00:00:00 2001
From: Chris <chris.goringe@gmail.com>
Date: Mon, 11 Sep 2023 14:55:20 +1000
Subject: [PATCH] pass UI_OUTPUT  to front end and add to prototype

---
 server.py          | 1 +
 web/scripts/app.js | 1 +
 2 files changed, 2 insertions(+)

diff --git a/server.py b/server.py
index d04060499..1029b42fe 100644
--- a/server.py
+++ b/server.py
@@ -399,6 +399,7 @@ class PromptServer():
             info['name'] = node_class
             info['display_name'] = nodes.NODE_DISPLAY_NAME_MAPPINGS[node_class] if node_class in nodes.NODE_DISPLAY_NAME_MAPPINGS.keys() else node_class
             info['description'] = obj_class.DESCRIPTION if hasattr(obj_class,'DESCRIPTION') else ''
+            info['ui_output'] = obj_class.UI_OUTPUT if hasattr(obj_class, 'UI_OUTPUT') else ''
             info['category'] = 'sd'
             if hasattr(obj_class, 'OUTPUT_NODE') and obj_class.OUTPUT_NODE == True:
                 info['output_node'] = True
diff --git a/web/scripts/app.js b/web/scripts/app.js
index 9db4e9230..f82ce586e 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -1273,6 +1273,7 @@ export class ComfyApp {
 				}
 			);
 			node.prototype.comfyClass = nodeData.name;
+			node.prototype.ui_output = nodeData.ui_output;
 
 			this.#addNodeContextMenuHandler(node);
 			this.#addDrawBackgroundHandler(node, app);