import base64 import io import os from PIL import Image import gradio as gr from modules.paths import data_path from modules import shared, gr_tempdir, script_callbacks, images from modules.infotext import parse, mapping, quote, unquote # pylint: disable=unused-import type_of_gr_update = type(gr.update()) paste_fields = {} registered_param_bindings = [] debug = shared.log.trace if os.environ.get('SD_PASTE_DEBUG', None) is not None else lambda *args, **kwargs: None debug('Trace: PASTE') parse_generation_parameters = parse # compatibility infotext_to_setting_name_mapping = mapping # compatibility class ParamBinding: def __init__(self, paste_button, tabname, source_text_component=None, source_image_component=None, source_tabname=None, override_settings_component=None, paste_field_names=None): self.paste_button = paste_button self.tabname = tabname self.source_text_component = source_text_component self.source_image_component = source_image_component self.source_tabname = source_tabname self.override_settings_component = override_settings_component self.paste_field_names = paste_field_names or [] debug(f'ParamBinding: {vars(self)}') def reset(): paste_fields.clear() def image_from_url_text(filedata): if filedata is None: return None if type(filedata) == list and len(filedata) > 0 and type(filedata[0]) == dict and filedata[0].get("is_file", False): filedata = filedata[0] if type(filedata) == dict and filedata.get("is_file", False): filename = filedata["name"] is_in_right_dir = gr_tempdir.check_tmp_file(shared.demo, filename) if is_in_right_dir: filename = filename.rsplit('?', 1)[0] if not os.path.exists(filename): shared.log.error(f'Image file not found: {filename}') image = Image.new('RGB', (512, 512)) image.info['parameters'] = f'Image file not found: {filename}' return image image = Image.open(filename) geninfo, _items = images.read_info_from_image(image) image.info['parameters'] = geninfo return image else: shared.log.warning(f'File access denied: {filename}') return None if type(filedata) == list: if len(filedata) == 0: return None filedata = filedata[0] if type(filedata) == dict: shared.log.warning('Incorrect filedata received') return None if filedata.startswith("data:image/png;base64,"): filedata = filedata[len("data:image/png;base64,"):] if filedata.startswith("data:image/webp;base64,"): filedata = filedata[len("data:image/webp;base64,"):] if filedata.startswith("data:image/jpeg;base64,"): filedata = filedata[len("data:image/jpeg;base64,"):] filedata = base64.decodebytes(filedata.encode('utf-8')) image = Image.open(io.BytesIO(filedata)) images.read_info_from_image(image) return image def add_paste_fields(tabname, init_img, fields, override_settings_component=None): paste_fields[tabname] = {"init_img": init_img, "fields": fields, "override_settings_component": override_settings_component} # backwards compatibility for existing extensions import modules.ui if tabname == 'txt2img': modules.ui.txt2img_paste_fields = fields elif tabname == 'img2img': modules.ui.img2img_paste_fields = fields elif tabname == 'control': modules.ui.control_paste_fields = fields def create_buttons(tabs_list): buttons = {} for tab in tabs_list: name = tab if name == 'txt2img': name = 'Text' elif name == 'img2img': name = 'Image' elif name == 'inpaint': name = 'Inpaint' elif name == 'extras': name = 'Process' elif name == 'control': name = 'Control' buttons[tab] = gr.Button(f"➠ {name}", elem_id=f"{tab}_tab") return buttons def bind_buttons(buttons, send_image, send_generate_info): """old function for backwards compatibility; do not use this, use register_paste_params_button""" for tabname, button in buttons.items(): source_text_component = send_generate_info if isinstance(send_generate_info, gr.components.Component) else None source_tabname = send_generate_info if isinstance(send_generate_info, str) else None bindings = ParamBinding(paste_button=button, tabname=tabname, source_text_component=source_text_component, source_image_component=send_image, source_tabname=source_tabname) register_paste_params_button(bindings) def register_paste_params_button(binding: ParamBinding): registered_param_bindings.append(binding) def connect_paste_params_buttons(): binding: ParamBinding for binding in registered_param_bindings: if binding.tabname not in paste_fields: debug(f"Not not registered: tab={binding.tabname}") continue destination_image_component = paste_fields[binding.tabname]["init_img"] fields = paste_fields[binding.tabname]["fields"] override_settings_component = binding.override_settings_component or paste_fields[binding.tabname]["override_settings_component"] destination_width_component = next(iter([field for field, name in fields if name == "Size-1"] if fields else []), None) destination_height_component = next(iter([field for field, name in fields if name == "Size-2"] if fields else []), None) if binding.source_image_component and destination_image_component: if isinstance(binding.source_image_component, gr.Gallery): func = send_image_and_dimensions if destination_width_component else image_from_url_text jsfunc = "extract_image_from_gallery" else: func = send_image_and_dimensions if destination_width_component else lambda x: x jsfunc = None binding.paste_button.click( fn=func, _js=jsfunc, inputs=[binding.source_image_component], outputs=[destination_image_component, destination_width_component, destination_height_component] if destination_width_component else [destination_image_component], show_progress=False, ) if binding.source_text_component is not None and fields is not None: connect_paste(binding.paste_button, fields, binding.source_text_component, override_settings_component, binding.tabname) if binding.source_tabname is not None and fields is not None and binding.source_tabname in paste_fields: paste_field_names = ['Prompt', 'Negative prompt', 'Steps', 'Face restoration'] + (["Seed"] if shared.opts.send_seed else []) + binding.paste_field_names if "fields" in paste_fields[binding.source_tabname] and paste_fields[binding.source_tabname]["fields"] is not None: binding.paste_button.click( fn=lambda *x: x, inputs=[field for field, name in paste_fields[binding.source_tabname]["fields"] if name in paste_field_names], outputs=[field for field, name in fields if name in paste_field_names], ) binding.paste_button.click( fn=None, _js=f"switch_to_{binding.tabname}", inputs=[], outputs=[], show_progress=False, ) def send_image_and_dimensions(x): img = x if isinstance(x, Image.Image) else image_from_url_text(x) if shared.opts.send_size and isinstance(img, Image.Image): w = img.width h = img.height else: w = gr.update() h = gr.update() return img, w, h def create_override_settings_dict(text_pairs): res = {} params = {} for pair in text_pairs: k, v = pair.split(":", maxsplit=1) params[k] = v.strip() for param_name, setting_name in mapping: value = params.get(param_name, None) if value is None: continue res[setting_name] = shared.opts.cast_value(setting_name, value) return res def connect_paste(button, local_paste_fields, input_comp, override_settings_component, tabname): def paste_func(prompt): if prompt is None or len(prompt.strip()) == 0 and not shared.cmd_opts.hide_ui_dir_config: filename = os.path.join(data_path, "params.txt") if os.path.exists(filename): with open(filename, "r", encoding="utf8") as file: prompt = file.read() shared.log.debug(f'Paste prompt: type="params" prompt="{prompt}"') else: prompt = '' else: shared.log.debug(f'Paste prompt: type="current" prompt="{prompt}"') params = parse(prompt) script_callbacks.infotext_pasted_callback(prompt, params) res = [] applied = {} for output, key in local_paste_fields: if callable(key): v = key(params) else: v = params.get(key, None) if v is None: res.append(gr.update()) # triggers update for each gradio component even if there are no updates elif isinstance(v, type_of_gr_update): res.append(v) applied[key] = v else: try: valtype = type(output.value) if valtype == bool and v == "False": val = False else: val = valtype(v) res.append(gr.update(value=val)) applied[key] = val except Exception: res.append(gr.update()) debug(f"Parse apply: {applied}") return res if override_settings_component is not None: def paste_settings(params): params.pop('Prompt', None) params.pop('Negative prompt', None) if not params: gr.Dropdown.update(value=[], choices=[], visible=False) vals = {} for param_name, setting_name in infotext_to_setting_name_mapping: v = params.get(param_name, None) if v is None: continue if shared.opts.disable_weights_auto_swap: if setting_name == "sd_model_checkpoint" or setting_name == 'sd_model_refiner' or setting_name == 'sd_backend' or setting_name == 'sd_vae': continue v = shared.opts.cast_value(setting_name, v) current_value = getattr(shared.opts, setting_name, None) if v == current_value: continue if type(current_value) == str and v == os.path.splitext(current_value)[0]: continue vals[param_name] = v vals_pairs = [f"{k}: {v}" for k, v in vals.items()] if len(vals_pairs) > 0: shared.log.debug(f'Settings overrides: {vals_pairs}') return gr.Dropdown.update(value=vals_pairs, choices=vals_pairs, visible=len(vals_pairs) > 0) local_paste_fields = local_paste_fields + [(override_settings_component, paste_settings)] button.click( fn=paste_func, inputs=[input_comp], outputs=[x[0] for x in local_paste_fields], show_progress=False, ) button.click( fn=None, _js=f"recalculate_prompts_{tabname}", inputs=[], outputs=[], show_progress=False, )