102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
import os
|
|
import re
|
|
import gradio as gr
|
|
from modules.shared import script_path
|
|
from modules import shared
|
|
|
|
re_param_code = r"\s*([\w ]+):\s*([^,]+)(?:,|$)"
|
|
re_param = re.compile(re_param_code)
|
|
re_params = re.compile(r"^(?:" + re_param_code + "){3,}$")
|
|
re_imagesize = re.compile(r"^(\d+)x(\d+)$")
|
|
type_of_gr_update = type(gr.update())
|
|
|
|
|
|
def parse_generation_parameters(x: str):
|
|
"""parses generation parameters string, the one you see in text field under the picture in UI:
|
|
```
|
|
girl with an artist's beret, determined, blue eyes, desert scene, computer monitors, heavy makeup, by Alphonse Mucha and Charlie Bowater, ((eyeshadow)), (coquettish), detailed, intricate
|
|
Negative prompt: ugly, fat, obese, chubby, (((deformed))), [blurry], bad anatomy, disfigured, poorly drawn face, mutation, mutated, (extra_limb), (ugly), (poorly drawn hands), messy drawing
|
|
Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model hash: 45dee52b
|
|
```
|
|
|
|
returns a dict with field values
|
|
"""
|
|
|
|
res = {}
|
|
|
|
prompt = ""
|
|
negative_prompt = ""
|
|
|
|
done_with_prompt = False
|
|
|
|
*lines, lastline = x.strip().split("\n")
|
|
if not re_params.match(lastline):
|
|
lines.append(lastline)
|
|
lastline = ''
|
|
|
|
for i, line in enumerate(lines):
|
|
line = line.strip()
|
|
if line.startswith("Negative prompt:"):
|
|
done_with_prompt = True
|
|
line = line[16:].strip()
|
|
|
|
if done_with_prompt:
|
|
negative_prompt += ("" if negative_prompt == "" else "\n") + line
|
|
else:
|
|
prompt += ("" if prompt == "" else "\n") + line
|
|
|
|
if len(prompt) > 0:
|
|
res["Prompt"] = prompt
|
|
|
|
if len(negative_prompt) > 0:
|
|
res["Negative prompt"] = negative_prompt
|
|
|
|
for k, v in re_param.findall(lastline):
|
|
m = re_imagesize.match(v)
|
|
if m is not None:
|
|
res[k+"-1"] = m.group(1)
|
|
res[k+"-2"] = m.group(2)
|
|
else:
|
|
res[k] = v
|
|
|
|
return res
|
|
|
|
|
|
def connect_paste(button, paste_fields, input_comp, js=None):
|
|
def paste_func(prompt):
|
|
if not prompt and not shared.cmd_opts.hide_ui_dir_config:
|
|
filename = os.path.join(script_path, "params.txt")
|
|
if os.path.exists(filename):
|
|
with open(filename, "r", encoding="utf8") as file:
|
|
prompt = file.read()
|
|
|
|
params = parse_generation_parameters(prompt)
|
|
res = []
|
|
|
|
for output, key in paste_fields:
|
|
if callable(key):
|
|
v = key(params)
|
|
else:
|
|
v = params.get(key, None)
|
|
|
|
if v is None:
|
|
res.append(gr.update())
|
|
elif isinstance(v, type_of_gr_update):
|
|
res.append(v)
|
|
else:
|
|
try:
|
|
valtype = type(output.value)
|
|
val = valtype(v)
|
|
res.append(gr.update(value=val))
|
|
except Exception:
|
|
res.append(gr.update())
|
|
|
|
return res
|
|
|
|
button.click(
|
|
fn=paste_func,
|
|
_js=js,
|
|
inputs=[input_comp],
|
|
outputs=[x[0] for x in paste_fields],
|
|
)
|