From cb1fca3a7ad8d672125e44322d8045f8c46fe2ba Mon Sep 17 00:00:00 2001
From: Jonas Krauss <jonas.krauss@stockpulse.de>
Date: Mon, 1 Apr 2024 22:38:53 +0200
Subject: [PATCH 1/4] node to create wildcard label prompt from string list

---
 __init__.py                  |  2 ++
 modules/impact/util_nodes.py | 60 ++++++++++++++++++++++++++++++++++++
 2 files changed, 62 insertions(+)

diff --git a/__init__.py b/__init__.py
index 83717de3..62f9c08e 100644
--- a/__init__.py
+++ b/__init__.py
@@ -313,6 +313,7 @@ def wildcard_load():
     "ImpactNeg": ImpactNeg,
     "ImpactConditionalStopIteration": ImpactConditionalStopIteration,
     "ImpactStringSelector": StringSelector,
+    "WildcardPromptFromStringList": WildcardPromptFromStringList,
 
     "RemoveNoiseMask": RemoveNoiseMask,
 
@@ -433,6 +434,7 @@ def wildcard_load():
     "ImpactMakeImageList": "Make Image List",
     "ImpactMakeImageBatch": "Make Image Batch",
     "ImpactStringSelector": "String Selector",
+    "WildcardPromptFromStringList": "Wildcard Prompt from String List",
     "ImpactIsNotEmptySEGS": "SEGS isn't Empty",
     "SetDefaultImageForSEGS": "Set Default Image for SEGS",
     "RemoveImageFromSEGS": "Remove Image from SEGS",
diff --git a/modules/impact/util_nodes.py b/modules/impact/util_nodes.py
index 1484d520..85ee2b17 100644
--- a/modules/impact/util_nodes.py
+++ b/modules/impact/util_nodes.py
@@ -5,6 +5,7 @@
 import comfy
 import sys
 import nodes
+import re
 
 
 class GeneralSwitch:
@@ -485,3 +486,62 @@ def doit(self, strings, multiline, select):
                 selected = lines[select % len(lines)]
 
         return (selected, )
+
+
+class WildcardPromptFromStringList:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "string_list": ("STRING", {"forceInput": True}),
+                "prefix_all": ("STRING", {"multiline": False}),
+                "postfix_all": ("STRING", {"multiline": False}),
+                "restrict_to_tags": ("STRING", {"multiline": False}),
+                "exclude_tags": ("STRING", {"multiline": False})
+            },
+        }
+
+    INPUT_IS_LIST = True
+    RETURN_TYPES = ("STRING", "STRING",)
+    RETURN_NAMES = ("wildcard", "segs_labels",)
+    FUNCTION = "doit"
+
+    CATEGORY = "ImpactPack/Util"
+
+    def doit(self, string_list, prefix_all, postfix_all, restrict_to_tags, exclude_tags):
+        # need to access as list due to INPUT_IS_LIST
+        # some sanity checks and normalization for later processing
+        if prefix_all[0] == None: prefix_all[0] = ""
+        if postfix_all[0] == None: postfix_all[0] = ""
+        if restrict_to_tags[0] == None: restrict_to_tags[0] = ""
+        if exclude_tags[0] == None: exclude_tags[0] = ""
+        if not isinstance(restrict_to_tags[0], list):
+            restrict_to_tags[0] = restrict_to_tags[0].split(", ")
+        if not isinstance(exclude_tags[0], list):
+            exclude_tags[0] = exclude_tags[0].split(", ")
+
+        # build the wildcard prompt per list entry
+        output = ["[LAB]"]
+        labels = []
+        for x in string_list:
+            label = str(len(labels) + 1)
+            labels.append(label)
+            x = x.split(", ")
+            # restrict to tags
+            if restrict_to_tags[0][0] != "":
+                x = list(set(x) & set(restrict_to_tags[0]))
+            # remove tags
+            if exclude_tags[0][0] != "":
+                x = list(set(x) - set(exclude_tags[0]))
+            # next row: <LABEL> <PREFIX> <TAGS> <POSTFIX>
+            prompt_for_seg = f'[{label}] {prefix_all[0]} {", ".join(x)} {postfix_all[0]}'.strip()
+            output.append(prompt_for_seg)
+        output = "\n".join(output)
+
+        # clean string: fixup double spaces, commas etc.
+        output = re.sub(r' ,', ',', output)
+        output = re.sub(r'  +', ' ', output)
+        output = re.sub(r',,+', ',', output)
+        output = re.sub(r'\n, ', '\n', output)
+
+        return (output, ", ".join(labels),)

From 43e1bbbef3fa0000fdcda9a4959b76771f207719 Mon Sep 17 00:00:00 2001
From: Jonas Krauss <jonas.krauss@stockpulse.de>
Date: Sun, 21 Apr 2024 09:41:35 +0200
Subject: [PATCH 2/4] use `is` for checking on None value

---
 modules/impact/util_nodes.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/modules/impact/util_nodes.py b/modules/impact/util_nodes.py
index 85ee2b17..9dc771db 100644
--- a/modules/impact/util_nodes.py
+++ b/modules/impact/util_nodes.py
@@ -511,10 +511,10 @@ def INPUT_TYPES(s):
     def doit(self, string_list, prefix_all, postfix_all, restrict_to_tags, exclude_tags):
         # need to access as list due to INPUT_IS_LIST
         # some sanity checks and normalization for later processing
-        if prefix_all[0] == None: prefix_all[0] = ""
-        if postfix_all[0] == None: postfix_all[0] = ""
-        if restrict_to_tags[0] == None: restrict_to_tags[0] = ""
-        if exclude_tags[0] == None: exclude_tags[0] = ""
+        if prefix_all[0] is None: prefix_all[0] = ""
+        if postfix_all[0] is None: postfix_all[0] = ""
+        if restrict_to_tags[0] is None: restrict_to_tags[0] = ""
+        if exclude_tags[0] is None: exclude_tags[0] = ""
         if not isinstance(restrict_to_tags[0], list):
             restrict_to_tags[0] = restrict_to_tags[0].split(", ")
         if not isinstance(exclude_tags[0], list):

From e41788aafa0a14eaeb70bb7dfce99be17cc0d027 Mon Sep 17 00:00:00 2001
From: Jonas Krauss <jonas.krauss@stockpulse.de>
Date: Wed, 1 May 2024 11:21:39 +0200
Subject: [PATCH 3/4] add StringListToString node

---
 __init__.py                  |  2 ++
 modules/impact/util_nodes.py | 65 ++++++++++++++++++++++++++----------
 2 files changed, 49 insertions(+), 18 deletions(-)

diff --git a/__init__.py b/__init__.py
index 62f9c08e..34489471 100644
--- a/__init__.py
+++ b/__init__.py
@@ -313,6 +313,7 @@ def wildcard_load():
     "ImpactNeg": ImpactNeg,
     "ImpactConditionalStopIteration": ImpactConditionalStopIteration,
     "ImpactStringSelector": StringSelector,
+    "StringListToString": StringListToString,
     "WildcardPromptFromStringList": WildcardPromptFromStringList,
 
     "RemoveNoiseMask": RemoveNoiseMask,
@@ -434,6 +435,7 @@ def wildcard_load():
     "ImpactMakeImageList": "Make Image List",
     "ImpactMakeImageBatch": "Make Image Batch",
     "ImpactStringSelector": "String Selector",
+    "StringListToString": "String List to String",
     "WildcardPromptFromStringList": "Wildcard Prompt from String List",
     "ImpactIsNotEmptySEGS": "SEGS isn't Empty",
     "SetDefaultImageForSEGS": "Set Default Image for SEGS",
diff --git a/modules/impact/util_nodes.py b/modules/impact/util_nodes.py
index 9dc771db..f81fb952 100644
--- a/modules/impact/util_nodes.py
+++ b/modules/impact/util_nodes.py
@@ -488,12 +488,39 @@ def doit(self, strings, multiline, select):
         return (selected, )
 
 
-class WildcardPromptFromStringList:
+class StringListToString:
     @classmethod
     def INPUT_TYPES(s):
         return {
             "required": {
+                "join_with": ("STRING", {"default": "\\n"}),
                 "string_list": ("STRING", {"forceInput": True}),
+            }
+        }
+
+    INPUT_IS_LIST = True
+    RETURN_TYPES = ("STRING",)
+    FUNCTION = "doit"
+
+    CATEGORY = "ImpactPack/Util"
+
+    def doit(self, join_with, string_list):
+        # convert \\n to newline character
+        if join_with[0] == "\\n":
+            join_with[0] = "\n"
+
+        joined_text = join_with[0].join(string_list)
+
+        return (joined_text,)
+
+
+class WildcardPromptFromStringList:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "string": ("STRING", {"forceInput": True}),
+                "delimiter": ("STRING", {"multiline": False, "default": "\\n" }),
                 "prefix_all": ("STRING", {"multiline": False}),
                 "postfix_all": ("STRING", {"multiline": False}),
                 "restrict_to_tags": ("STRING", {"multiline": False}),
@@ -501,40 +528,42 @@ def INPUT_TYPES(s):
             },
         }
 
-    INPUT_IS_LIST = True
     RETURN_TYPES = ("STRING", "STRING",)
     RETURN_NAMES = ("wildcard", "segs_labels",)
     FUNCTION = "doit"
 
     CATEGORY = "ImpactPack/Util"
 
-    def doit(self, string_list, prefix_all, postfix_all, restrict_to_tags, exclude_tags):
-        # need to access as list due to INPUT_IS_LIST
+    def doit(self, string, delimiter, prefix_all, postfix_all, restrict_to_tags, exclude_tags):
+        # convert \\n to newline character
+        if delimiter == "\\n":
+            delimiter = "\n"
+
         # some sanity checks and normalization for later processing
-        if prefix_all[0] is None: prefix_all[0] = ""
-        if postfix_all[0] is None: postfix_all[0] = ""
-        if restrict_to_tags[0] is None: restrict_to_tags[0] = ""
-        if exclude_tags[0] is None: exclude_tags[0] = ""
-        if not isinstance(restrict_to_tags[0], list):
-            restrict_to_tags[0] = restrict_to_tags[0].split(", ")
-        if not isinstance(exclude_tags[0], list):
-            exclude_tags[0] = exclude_tags[0].split(", ")
+        if prefix_all is None: prefix_all = ""
+        if postfix_all is None: postfix_all = ""
+        if restrict_to_tags is None: restrict_to_tags = ""
+        if exclude_tags is None: exclude_tags = ""
+        if not isinstance(restrict_to_tags, list):
+            restrict_to_tags = restrict_to_tags.split(", ")
+        if not isinstance(exclude_tags, list):
+            exclude_tags = exclude_tags.split(", ")
 
         # build the wildcard prompt per list entry
         output = ["[LAB]"]
         labels = []
-        for x in string_list:
+        for x in string.split(delimiter):
             label = str(len(labels) + 1)
             labels.append(label)
             x = x.split(", ")
             # restrict to tags
-            if restrict_to_tags[0][0] != "":
-                x = list(set(x) & set(restrict_to_tags[0]))
+            if restrict_to_tags != "":
+                x = list(set(x) & set(restrict_to_tags))
             # remove tags
-            if exclude_tags[0][0] != "":
-                x = list(set(x) - set(exclude_tags[0]))
+            if exclude_tags != "":
+                x = list(set(x) - set(exclude_tags))
             # next row: <LABEL> <PREFIX> <TAGS> <POSTFIX>
-            prompt_for_seg = f'[{label}] {prefix_all[0]} {", ".join(x)} {postfix_all[0]}'.strip()
+            prompt_for_seg = f'[{label}] {prefix_all} {", ".join(x)} {postfix_all}'.strip()
             output.append(prompt_for_seg)
         output = "\n".join(output)
 

From 659d9dfb0dcedef5690e6ba356fbcfaf99cd508e Mon Sep 17 00:00:00 2001
From: Jonas Krauss <jonas.krauss@stockpulse.de>
Date: Wed, 1 May 2024 11:49:34 +0200
Subject: [PATCH 4/4] remove `list` from name of wildcard creation node

---
 __init__.py                  | 4 ++--
 modules/impact/util_nodes.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index 34489471..c8c7a8e9 100644
--- a/__init__.py
+++ b/__init__.py
@@ -314,7 +314,7 @@ def wildcard_load():
     "ImpactConditionalStopIteration": ImpactConditionalStopIteration,
     "ImpactStringSelector": StringSelector,
     "StringListToString": StringListToString,
-    "WildcardPromptFromStringList": WildcardPromptFromStringList,
+    "WildcardPromptFromString": WildcardPromptFromString,
 
     "RemoveNoiseMask": RemoveNoiseMask,
 
@@ -436,7 +436,7 @@ def wildcard_load():
     "ImpactMakeImageBatch": "Make Image Batch",
     "ImpactStringSelector": "String Selector",
     "StringListToString": "String List to String",
-    "WildcardPromptFromStringList": "Wildcard Prompt from String List",
+    "WildcardPromptFromString": "Wildcard Prompt from String",
     "ImpactIsNotEmptySEGS": "SEGS isn't Empty",
     "SetDefaultImageForSEGS": "Set Default Image for SEGS",
     "RemoveImageFromSEGS": "Remove Image from SEGS",
diff --git a/modules/impact/util_nodes.py b/modules/impact/util_nodes.py
index f81fb952..d590e5e2 100644
--- a/modules/impact/util_nodes.py
+++ b/modules/impact/util_nodes.py
@@ -514,7 +514,7 @@ def doit(self, join_with, string_list):
         return (joined_text,)
 
 
-class WildcardPromptFromStringList:
+class WildcardPromptFromString:
     @classmethod
     def INPUT_TYPES(s):
         return {