Initial commit

2026-07-17 00:31:36 +02:00
commit d38647157e
10 changed files with 508 additions and 0 deletions
@@ -0,0 +1,77 @@
+import random
+import re
+
+
+class MultilineRandomChanceNode:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "text": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "A professional photo of a {female::5|male} mage, casting a {fire::5|ice::2|lightning} spell, cinematic lighting.",
+                    },
+                ),
+                "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}),
+            },
+            "optional": {
+                "regex_pattern": ("STRING", {"default": ",\\s*,"}),
+                "regex_replacement": ("STRING", {"default": ","}),
+            },
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("selected_text",)
+    FUNCTION = "process_random_text"
+    CATEGORY = "TextRandomizer/Text"
+
+    def _parse_weights(self, options_list, separator="::"):
+        choices, weights = [], []
+        for opt in options_list:
+            if separator in opt:
+                parts = opt.rsplit(separator, 1)
+                choice_text = parts[0].strip()
+                try:
+                    weight = float(parts[1].strip())
+                except ValueError:
+                    weight = 1.0
+            else:
+                choice_text = opt.strip()
+                weight = 1.0
+            choices.append(choice_text)
+            weights.append(weight)
+        return choices, weights
+
+    def _replace_brackets(self, match):
+        content = match.group(1)
+        raw_options = content.split("|")
+        choices, weights = self._parse_weights(raw_options, separator="::")
+        if choices:
+            return random.choices(choices, weights=weights, k=1)[0]
+        return ""
+
+    def process_random_text(self, text, seed, regex_pattern="", regex_replacement=""):
+        random.seed(seed)
+        lines = [line.strip() for line in text.split("\n") if line.strip()]
+        if not lines:
+            return ("",)
+
+        base_choices, base_weights = self._parse_weights(lines, separator="::")
+        selected_line = random.choices(base_choices, weights=base_weights, k=1)[0]
+
+        bracket_pattern = re.compile(r"\{([^{}]+)\}")
+        loop_guard = 0
+        while bracket_pattern.search(selected_line) and loop_guard < 10:
+            selected_line = bracket_pattern.sub(self._replace_brackets, selected_line)
+            loop_guard += 1
+
+        if regex_pattern:
+            try:
+                compiled_regex = re.compile(regex_pattern)
+                selected_line = compiled_regex.sub(regex_replacement, selected_line)
+            except re.error:
+                pass
+
+        return (selected_line,)