From 0b7589225ac8da1a38399b4b059984f4fb26ce5b Mon Sep 17 00:00:00 2001
From: Antonio Cordero Balcazar <antonio.cordero@gmail.com>
Date: Thu, 11 May 2023 16:12:40 +0200
Subject: [PATCH] Initial version

---
 .gitignore                |  1 +
 README.md                 |  7 ++++++
 scripts/sendtonegative.py | 45 +++++++++++++++++++++++++++++++++++++++
 3 files changed, 53 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 README.md
 create mode 100644 scripts/sendtonegative.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..0575a0b
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+scripts/__pycache__
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..0ad8105
--- /dev/null
+++ b/README.md
@@ -0,0 +1,7 @@
+# Send To Negative
+
+## Purpose
+
+This extension allows the use of negative weights in the prompt and moves them to the negative prompt. This allows useful tricks when using a wildcard extension since you can add negative parts from the positive prompt. It will process recursive weights (one inside of another), but won't calculate the multiplied weights.
+
+The extension must be loaded after the wildcard extension. With the "Dynamic Prompts" extension this happens by default due to both extensions default folder names.
diff --git a/scripts/sendtonegative.py b/scripts/sendtonegative.py
new file mode 100644
index 0000000..c05c89c
--- /dev/null
+++ b/scripts/sendtonegative.py
@@ -0,0 +1,45 @@
+import logging
+from modules.processing import StableDiffusionProcessing
+import modules.scripts as scripts
+import re
+
+VERSION = '0.1'
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+class Script(scripts.Script):
+    def __init__(self):
+        pass
+
+    def title(self):
+        return f"SendToNegative v{VERSION}"
+
+    def show(self, is_img2img):
+        return scripts.AlwaysVisible
+
+    def process(self, p: StableDiffusionProcessing, *args, **kwargs):
+        try:
+            find = r"(((?<!\\)\([^():]+:)-(\d+(?:\.\d+)?(?<!\\)\)))" # format (xxxx:-nnn) TODO : support [] ? accumulate recursive weights?
+            regex = re.compile(find, re.S)
+            for i in range(len(p.all_prompts)):
+                # Extract from the prompt the terms with negative weights and add them to the negative prompt
+                current_prompt = p.all_prompts[i]
+                current_negative_prompt = p.all_negative_prompts[i]
+                already_applied = []
+                while True:
+                    matches = regex.findall(current_prompt)
+                    if len(matches) == 0:
+                        break
+                    for match in matches:
+                        if match[0] not in already_applied:
+                            logger.debug("Found negative term: %s", match[0])
+                            current_negative_prompt = match[1] + match[2] + " " + current_negative_prompt
+                            current_prompt = current_prompt.replace(match[0], "")
+                            already_applied.append(match[0])
+                        else:
+                            logger.debug("Found repeated negative term: %s", match[0])
+                p.all_prompts[i] = current_prompt
+                p.all_negative_prompts[i] = current_negative_prompt
+        except Exception as e:
+            logger.exception(e)