fixed so that {prompt} can be anywhere in style

This commit is contained in:
RnDMonkey 2022-09-30 00:37:18 -07:00 committed by AUTOMATIC1111
parent ff4df06c57
commit 5c0c778a65
1 changed files with 5 additions and 1 deletions

View File

@ -290,7 +290,11 @@ def apply_filename_pattern(x, p, seed, prompt):
if "[prompt_no_styles]" in x:
prompt_no_style = prompt
for style in shared.prompt_styles.get_style_prompts(p.styles):
prompt_no_style = prompt_no_style.replace(style.replace("{prompt}", ""), "")
if len(style) > 0:
style_parts = [y for y in style.split("{prompt}")]
for part in style_parts:
prompt_no_style = prompt_no_style.replace(part, "").replace(", ,", ",").strip().strip(',')
prompt_no_style = prompt_no_style.replace(style, "").strip().strip(',').strip()
x = x.replace("[prompt_no_styles]", sanitize_filename_part(prompt_no_style, replace_spaces=False))
x = x.replace("[prompt_spaces]", sanitize_filename_part(prompt, replace_spaces=False))