From cc4f0fdd79d4b46c95a9b8ec32f18da4c8a71322 Mon Sep 17 00:00:00 2001
From: shanexi <shane@myshell.ai>
Date: Wed, 16 Oct 2024 11:16:33 +0800
Subject: [PATCH] Choices

---
 comfy-nodes/input_text.py | 16 ++++++++--------
 web/shellagent.js         | 12 +++++-------
 2 files changed, 13 insertions(+), 15 deletions(-)

diff --git a/comfy-nodes/input_text.py b/comfy-nodes/input_text.py
index fb90d7a..50de2e4 100755
--- a/comfy-nodes/input_text.py
+++ b/comfy-nodes/input_text.py
@@ -22,10 +22,10 @@ class ShellAgentPluginInputText:
                     "STRING",
                     {"multiline": True, "default": ""},
                 ),
-                "choices": (
-                    "STRING",
-                    {"multiline": False, "default": ""},
-                ),
+                # "choices": (
+                #     "STRING",
+                #     {"multiline": False, "default": ""},
+                # ),
             }
         }
 
@@ -79,10 +79,10 @@ class ShellAgentPluginInputFloat:
                     "STRING",
                     {"default": ""},
                 ),
-                "choices": (
-                    "STRING",
-                    {"multiline": False, "default": ""},
-                ),
+                # "choices": (
+                #     "STRING",
+                #     {"multiline": False, "default": ""},
+                # ),
             }
         }
 
diff --git a/web/shellagent.js b/web/shellagent.js
index 2070a8e..f5b71c2 100644
--- a/web/shellagent.js
+++ b/web/shellagent.js
@@ -9,7 +9,6 @@ const getPresets = () => {
     items = JSON.parse(localStorage.getItem(id));
   } catch (error) { }
   if (!items || !items.length) {
-    // items = [{ name: "default negative", value: "worst quality" }];
     items = [""];
   }
   return items;
@@ -54,14 +53,13 @@ app.registerExtension({
   },
   async beforeRegisterNodeDef(nodeType, nodeData, app) {
 
-    if (nodeData.name === "ShellAgentPluginInputText") {
-
+    if (["ShellAgentPluginInputText", "ShellAgentPluginInputFloat"].indexOf(nodeData.name) > -1) {
       chainCallback(nodeType.prototype, "onNodeCreated", function () {
+        const widget = this.addWidget("combo", "choices", presets[0], () => { }, {
+          values: presets,
+        });
 
-        const widget = this.addWidget("combo", "value", presets[0], () => {}, {
-					values: presets,
-				});
-
+        console.log('nodeData', nodeData)
         this.addWidget('button', 'manage choices', null, () => {
           const container = document.createElement("div");
           Object.assign(container.style, {