From 153f3f386397e2019b7ace0e273e50712940e6a4 Mon Sep 17 00:00:00 2001
From: RookHyena <36494751+RookHyena@users.noreply.github.com>
Date: Mon, 6 May 2024 20:25:34 -0700
Subject: [PATCH] Prompt Weighting Toggle for sd_tag_batch.py

Added a toggle to enable or disable the prompt weighting. Prompt weighting is enabled by default, which is the vanilla behavior.

I found that prompt weighting sometimes caused problems and thought that it would be nice if users could choose whether interrogator prompts were weighted or not.
---
 scripts/sd_tag_batch.py | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/scripts/sd_tag_batch.py b/scripts/sd_tag_batch.py
index fb3c962..c1f608a 100644
--- a/scripts/sd_tag_batch.py
+++ b/scripts/sd_tag_batch.py
@@ -26,10 +26,11 @@ class Script(scripts.Script):
         prompt_weight = gr.Slider(
             0.0, 1.0, value=0.5, step=0.1, label="interrogator weight"
         )
+        use_weight = gr.Checkbox(label="Use Weighted Prompt", value=True)
         use_deepbooru = gr.Checkbox(label="Use deepbooru", value=True)
-        return [in_front, prompt_weight, use_deepbooru]
+        return [in_front, prompt_weight, use_deepbooru, use_weight]
 
-    def run(self, p, in_front, prompt_weight, use_deepbooru):
+    def run(self, p, in_front, prompt_weight, use_deepbooru, use_weight):
 
         # :|
         if in_front:
@@ -50,12 +51,20 @@ class Script(scripts.Script):
         Script.interrogator = prompt
 
         p.prompt = ""
-        if Script.original_prompt in ["Some", ""]:
-            p.prompt = Script.interrogator
-        elif in_front:
-            p.prompt = f"{Script.original_prompt}, ({Script.interrogator}:{prompt_weight})"
+        if use_weight:
+            if Script.original_prompt in ["Some", ""]:
+                p.prompt = Script.interrogator
+            elif in_front:
+                p.prompt = f"{Script.original_prompt}, ({Script.interrogator}:{prompt_weight})"
+            else:
+                p.prompt = f"({Script.interrogator}:{prompt_weight}), {Script.original_prompt}"
         else:
-            p.prompt = f"({Script.interrogator}:{prompt_weight}), {Script.original_prompt}"
+            if Script.original_prompt in ["Some", ""]:
+                p.prompt = Script.interrogator
+            elif in_front:
+                p.prompt = f"{Script.original_prompt}, {Script.interrogator}"
+            else:
+                p.prompt = f"{Script.interrogator}, {Script.original_prompt}"
 
         print(f"Prompt: {p.prompt}")
         return process_images(p)