Add files via upload

Author: if-ai

IFChatPromptNode.py

@@ -1,29 +1,59 @@
+import sys
+
 class IFDisplayText:
     def __init__(self):
         self.type = "output"
 
     @classmethod
     def INPUT_TYPES(cls):
-
         return {
             "required": {        
-                "text": ("STRING", {"forceInput": True}),     
-                },
+                "text": ("STRING", {"forceInput": True}),
+                "select": ("INT", {
+                    "default": 0, 
+                    "min": 0, 
+                    "max": sys.maxsize,  # No practical upper limit
+                    "step": 1,
+                    "tooltip": "Select which line to output (cycles through available lines)"
+                }),     
+            },
             "hidden": {},
-            }
+        }
 
-    RETURN_TYPES = ("STRING",)
+    RETURN_TYPES = ("STRING", "STRING", "INT", "STRING")
+    RETURN_NAMES = ("text", "text_list", "count", "selected")
+    OUTPUT_IS_LIST = (False, True, False, False)
     FUNCTION = "display_text"
     OUTPUT_NODE = True
     CATEGORY = "ImpactFrames💥🎞️"
 
-    def display_text(self, text):
+    def display_text(self, text, select):
         print("==================")
         print("IF_AI_tool_output:")
         print("==================")
         print(text)
-        return {"ui": {"string": [text,]}, "result": (text,)}
-    
+        
+        # Split text into lines and filter out empty lines
+        text_list = [line.strip() for line in text.split('\n') if line.strip()]
+        count = len(text_list)
+        
+        # Select line using modulo to handle cycling
+        if count == 0:
+            selected = text  # If no valid lines, return original text
+        else:
+            selected = text_list[select % count]
+        
+        # Return both UI update and the multiple outputs
+        return {
+            "ui": {"string": [text]}, 
+            "result": (
+                text,        # complete text
+                text_list,   # list of individual lines as separate string outputs
+                count,       # number of lines
+                selected    # selected line based on select input
+            )
+        }
+
 NODE_CLASS_MAPPINGS = {"IF_DisplayText": IFDisplayText}
 NODE_DISPLAY_NAME_MAPPINGS = {"IF_DisplayText": "IF Display Text📟"}
 

IFDisplayTextNode.py

@@ -10,19 +10,31 @@
 import tempfile
 from torchvision.transforms.functional import to_pil_image
 import folder_paths
-from server import PromptServer
-from aiohttp import web
-
-@PromptServer.instance.routes.post("/IF_ImagePrompt/get_models")
-async def get_models_endpoint(request):
-    data = await request.json()
-    engine = data.get("engine")
-    base_ip = data.get("base_ip")
-    port = data.get("port")
-
-    node = IFImagePrompt()
-    models = node.get_models(engine, base_ip, port)
-    return web.json_response(models)
+import sys
+
+# Add the ComfyUI directory to the Python path
+comfy_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
+sys.path.insert(0, comfy_path)
+
+try:
+    from server import PromptServer
+    from aiohttp import web
+
+    @PromptServer.instance.routes.post("/IF_ImagePrompt/get_models")
+    async def get_models_endpoint(request):
+        data = await request.json()
+        engine = data.get("engine")
+        base_ip = data.get("base_ip")
+        port = data.get("port")
+
+        node = IFImagePrompt()
+        models = node.get_models(engine, base_ip, port)
+        return web.json_response(models)
+except AttributeError:
+    print("PromptServer.instance not available. Skipping route decoration for IF_ImagePrompt.")
+    async def get_models_endpoint(request):
+        # Fallback implementation
+        return web.json_response({"error": "PromptServer.instance not available"})
 
 class IFImagePrompt:
 

IFImagePromptNode.py

@@ -0,0 +1,65 @@
+class IFJoinText:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "separator": ("STRING", {
+                    "multiline": False,
+                    "default": " ",
+                    "placeholder": "Text to insert between joined strings"
+                }),
+            },
+            "optional": {
+                "text1": ("STRING", {
+                    "multiline": False,
+                    "default": "",
+                    "forceInput": True,
+                    "placeholder": "First text input"
+                }),
+                "text2": ("STRING", {
+                    "multiline": False,
+                    "default": "",
+                    "forceInput": True,
+                    "placeholder": "Second text input"
+                }),
+                "text3": ("STRING", {
+                    "multiline": False,
+                    "default": "",
+                    "forceInput": True,
+                    "placeholder": "Third text input"
+                }),
+                "text4": ("STRING", {
+                    "multiline": False,
+                    "default": "",
+                    "forceInput": True,
+                    "placeholder": "Fourth text input"
+                }),
+            },
+        }
+    
+    RETURN_TYPES = ("STRING",)
+    FUNCTION = "join_text"
+    CATEGORY = "ImpactFrames💥🎞️"
+    
+    def join_text(self, separator=" ", text1="", text2="", text3="", text4=""):
+        # Collect all non-empty text inputs
+        texts = [t for t in [text1, text2, text3, text4] if t.strip()]
+        
+        # Join texts with separator
+        result = separator.join(texts)
+        
+        # Print for debugging
+        print("==================")
+        print("IF_JoinText output:")
+        print("==================")
+        print(result)
+        
+        return (result,)
+
+NODE_CLASS_MAPPINGS = {
+    "IF_JoinText": IFJoinText
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "IF_JoinText": "IF Join Text 📝"
+}