import os import re import gradio as gr from modules.shared import script_path 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: 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], )