add suffix and prefix support

2023-12-01 09:06:14 +05:30
parent b197e26c13
commit 9f29ddd5ce
1 changed files with 4 additions and 0 deletions
@@ -48,6 +48,8 @@ class Wildcard:
                "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}),
                "input_files": (input_files, {"default": input_files[0]}),
                "input_text": (["*"] + items, {"default": "*"}),
+                "prefix": ("STRING", {"default": "", "dynamicPrompts": False}),
+                "suffix": ("STRING", {"default": "", "dynamicPrompts": False}),
                "output_text": (
                    "STRING",
                    {"multiline": True, "dynamicPrompts": False, "default": ""},
@@ -67,6 +69,8 @@ class Wildcard:
            random.seed(kwargs["seed"])
            output_text = random.choice(items)

+        output_text = f"{kwargs['prefix']} {output_text} {kwargs['suffix']}".strip()
+
        return {
            "ui": {"output_text": output_text},
            "result": (output_text,),