From 3e80b2d8245d46b8a587ee0686b29865b3be312a Mon Sep 17 00:00:00 2001
From: Danamir <danamir@danamir.com>
Date: Tue, 8 Aug 2023 13:13:57 +0200
Subject: [PATCH] feat: Use [PROMPT] keyword to insert the source prompt (#243)

---
 scripts/!adetailer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
index 228b984..a2747c9 100644
--- a/scripts/!adetailer.py
+++ b/scripts/!adetailer.py
@@ -244,6 +244,8 @@ class AfterDetailerScript(scripts.Script):
         for n in range(len(prompts)):
             if not prompts[n]:
                 prompts[n] = blank_replacement
+            elif "[PROMPT]" in prompts[n]:
+                prompts[n] = prompts[n].replace("[PROMPT]", f" {blank_replacement} ")
         return prompts
 
     def get_prompt(self, p, args: ADetailerArgs) -> tuple[list[str], list[str]]: