Merge pull request #99 from sn0w12/main

Update WidgetToString to use 'any_input' link if no id or title provided.

nodes/nodes.py

@@ -717,9 +717,9 @@ class WidgetToString:
                          "any_input": (any, {}),
                          "node_title": ("STRING", {"multiline": False}),
                          },
-            
             "hidden": {"extra_pnginfo": "EXTRA_PNGINFO",
-                       "prompt": "PROMPT"},
+                       "prompt": "PROMPT",
+                       "unique_id": "UNIQUE_ID",},
         }
 
     RETURN_TYPES = ("STRING", )
@@ -727,18 +727,20 @@ class WidgetToString:
     CATEGORY = "KJNodes/text"
     DESCRIPTION = """
 Selects a node and it's specified widget and outputs the value as a string.  
+If no node id or title is provided it will use the 'any_input' link and use that node.  
 To see node id's, enable node id display from Manager badge menu.  
 Alternatively you can search with the node title. Node titles ONLY exist if they  
 are manually edited!  
-The 'any_input' is purely for making sure the node you want the value from exists in the workflow,  
+The 'any_input' is required for making sure the node you want the value from exists in the workflow.
-it has no other function than place this node at right spot in the workflow execution order.
 """
 
-    def get_widget_value(self, id, widget_name, extra_pnginfo, prompt, return_all=False, any_input=None, node_title=""):
+    def get_widget_value(self, id, widget_name, extra_pnginfo, prompt, unique_id, return_all=False, any_input=None, node_title=""):
         workflow = extra_pnginfo["workflow"]
         #print(json.dumps(workflow, indent=4))
         results = []
         node_id = None  # Initialize node_id to handle cases where no match is found
+        link_id = None
+        link_to_node_map = {}
 
         for node in workflow["nodes"]:
             if node_title:
@@ -748,9 +750,30 @@ it has no other function than place this node at right spot in the workflow exec
                         break
                 else:
                     print("Node title not found.")
-            elif node["id"] == id:
+            elif id != 0:
+                if node["id"] == id:
                     node_id = id
                     break
+            elif any_input is not None:
+                if node["type"] == "WidgetToString" and node["id"] == int(unique_id) and not link_id:
+                    for node_input in node["inputs"]:
+                        link_id = node_input["link"]
+                    
+                # Construct a map of links to node IDs for future reference
+                node_outputs = node.get("outputs", None)
+                if not node_outputs:
+                    continue
+                for output in node_outputs:
+                    node_links = output.get("links", None)
+                    if not node_links:
+                        continue
+                    for link in node_links:
+                        link_to_node_map[link] = node["id"]
+                        if link_id and link == link_id:
+                            break
+        
+        if link_id:
+            node_id = link_to_node_map.get(link_id, None)
 
         if node_id is None:
             raise ValueError("No matching node found for the given title or id")
@@ -763,9 +786,9 @@ it has no other function than place this node at right spot in the workflow exec
                 v = str(values["inputs"][widget_name])  # Convert to string here
                 return (v, )
             else:
-                raise NameError(f"Widget not found: {id}.{widget_name}")
+                raise NameError(f"Widget not found: {node_id}.{widget_name}")
         if not results:
-            raise NameError(f"Node not found: {id}")
+            raise NameError(f"Node not found: {node_id}")
         return (', '.join(results).strip(', '), )
 
 class DummyOut: