From 255be75d30f41e089e499ec1c8462d6bf64dec24 Mon Sep 17 00:00:00 2001
From: aperullo <18688190+aperullo@users.noreply.github.com>
Date: Tue, 11 Oct 2022 06:16:57 -0400
Subject: [PATCH] Error if prompt missing SR token to prevent mis-gens (#2209)

---
 scripts/xy_grid.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/scripts/xy_grid.py b/scripts/xy_grid.py
index 42e1489c..10a82dc9 100644
--- a/scripts/xy_grid.py
+++ b/scripts/xy_grid.py
@@ -27,9 +27,16 @@ def apply_field(field):
 
 
 def apply_prompt(p, x, xs):
+
+    orig_prompt = p.prompt
+    orig_negative_prompt = p.negative_prompt
+
     p.prompt = p.prompt.replace(xs[0], x)
     p.negative_prompt = p.negative_prompt.replace(xs[0], x)
 
+    if p.prompt == orig_prompt and p.negative_prompt == orig_negative_prompt:
+        raise RuntimeError(f"Prompt S/R did not find {xs[0]} in prompt or negative prompt. Did you forget to add the token?")
+
 
 def apply_order(p, x, xs):
     token_order = []