Add RemoveTextBetween node to remove text between specified strings; update version to 1.0.2

Author: bash-j

mikey_nodes.py

@@ -4178,6 +4178,25 @@ def concat(self, delimiter, text1, text2, text3, text4, text5):
         text = delimiter.join(texts)
         return (text,)
 
+class RemoveTextBetween:
+    # removes text between two strings
+    # default start is <think> and end is </think>
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {'required': {'text': ('STRING', {'multiline': True, 'default': 'Input Text Here', 'dynamicPrompts': False})},
+                'optional': {'start': ('STRING', {'default': '<think>'}),
+                             'end': ('STRING', {'default': '</think>'})}}
+    
+    RETURN_TYPES = ('STRING',)
+    FUNCTION = 'remove'
+    CATEGORY = 'Mikey/Text'
+
+    def remove(self, text, start, end):
+        # search and replace
+        # remove text between start and end
+        text = re.sub(re.escape(start) + '.*?' + re.escape(end), '', text, flags=re.DOTALL).lstrip().rstrip()
+        return (text,)
+
 class OobaPrompt:
     @classmethod
     def INPUT_TYPES(s):
@@ -5390,6 +5409,7 @@ def save(self, model, vae, filename_prefix, clip=None, prompt=None, extra_pnginf
     'SRFloatPromptInput': SRFloatPromptInput,
     'TextPreserve': TextPreserve,
     'TextConcat': TextConcat,
+    'RemoveTextBetween': RemoveTextBetween,
     'OobaPrompt': OobaPrompt,
     'WildcardOobaPrompt': WildcardOobaPrompt,
     'LMStudioPrompt': LMStudioPrompt,
@@ -5465,6 +5485,7 @@ def save(self, model, vae, filename_prefix, clip=None, prompt=None, extra_pnginf
     'SRFloatPromptInput': 'SR Float Prompt Input (Mikey)',
     'TextPreserve': 'Text Preserve (Mikey)',
     'TextConcat': 'Text Concat (Mikey)',
+    'RemoveTextBetween': 'Remove Text Between (Mikey)',
     'OobaPrompt': 'OobaPrompt (Mikey)',
     'WildcardOobaPrompt': 'Wildcard OobaPrompt (Mikey)',
     'LMStudioPrompt': 'LM Studio Prompt (Mikey)',

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "mikey_nodes"
 description = "Collection of convenient nodes. Wildcard, style, image, llm, haldclut, metadata, and more."
-version = "1.0.1"
+version = "1.0.2"
 license = { text = "MIT License" }
 
 [project.urls]