feat: Use [PROMPT] keyword to insert the source prompt (Bing-su#243)

Cryptik-Rick · Aug 8, 2023 · 3e80b2d · 3e80b2d
1 parent f18e00b
commit 3e80b2d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
@@ -244,6 +244,8 @@ def _get_prompt(
         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]]: