From e4a7ffbb7f40d8bf0d28afb282c48c7a10a490a7 Mon Sep 17 00:00:00 2001
From: AlekPet <alexepetrof@gmail.com>
Date: Wed, 6 Sep 2023 17:12:45 +0300
Subject: [PATCH] update

---
 ExtrasNode/extras_node.py    |  4 ++--
 ExtrasNode/js/extras_node.js | 13 +++++++++----
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/ExtrasNode/extras_node.py b/ExtrasNode/extras_node.py
index 4df2050..8b22912 100644
--- a/ExtrasNode/extras_node.py
+++ b/ExtrasNode/extras_node.py
@@ -12,11 +12,11 @@ class PreviewTextNode:
             "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
             }
 
-    RETURN_TYPES = ()
+    RETURN_TYPES = ("STRING",)
     OUTPUT_NODE = True
     FUNCTION = "preview_text"
 
     CATEGORY = "AlekPet Nodes/extras"
 
     def preview_text(self, text, prompt=None, extra_pnginfo=None):
-        return {"ui": {"string": [text,]}}
+        return {"ui": {"string": [text,]}, "result": (text,)}
diff --git a/ExtrasNode/js/extras_node.js b/ExtrasNode/js/extras_node.js
index 1968e0f..4f1bb28 100644
--- a/ExtrasNode/js/extras_node.js
+++ b/ExtrasNode/js/extras_node.js
@@ -4,11 +4,12 @@ import { ComfyWidgets } from "/scripts/widgets.js";
 app.registerExtension({
   name: "Comfy.ExtrasNode",
   async beforeRegisterNodeDef(nodeType, nodeData, app) {
+    // --- Preview Text Node
     if (nodeData.name === "PreviewTextNode") {
       // Node Created
       const onNodeCreated = nodeType.prototype.onNodeCreated;
       nodeType.prototype.onNodeCreated = function () {
-        const r = onNodeCreated
+        const ret = onNodeCreated
           ? onNodeCreated.apply(this, arguments)
           : undefined;
 
@@ -19,7 +20,7 @@ app.registerExtension({
 
         console.log(`Create PreviewTextNode: ${nodeName}`);
 
-        ComfyWidgets.STRING(
+        const wi = ComfyWidgets.STRING(
           this,
           nodeName,
           [
@@ -32,13 +33,15 @@ app.registerExtension({
           ],
           app
         );
-        return r;
+        wi.widget.inputEl.readOnly = true;
+
+        return ret;
       };
 
       // On drawforeground
       const onDrawForeground = nodeType.prototype.onDrawForeground;
       nodeType.prototype.onDrawForeground = function () {
-        onDrawForeground?.apply(this, arguments);
+        const ret = onDrawForeground?.apply(this, arguments);
 
         const output = app.nodeOutputs[this.id + ""];
         if (
@@ -48,7 +51,9 @@ app.registerExtension({
         ) {
           this.widgets[0].inputEl.value = output.string[0];
         }
+        return ret;
       };
     }
+    // --- Preview Text Node
   },
 });