import json import os import gradio as gr from modules import errors from modules.ui_components import ToolButton class UiLoadsave: """allows saving and restorig default values for gradio components""" def __init__(self, filename): self.filename = filename self.ui_settings = {} self.component_mapping = {} self.error_loading = False self.finalized_ui = False self.ui_defaults_view = None self.ui_defaults_apply = None self.ui_defaults_review = None try: if os.path.exists(self.filename): self.ui_settings = self.read_from_file() except Exception as e: self.error_loading = True errors.display(e, "loading settings") def add_component(self, path, x): """adds component to the registry of tracked components""" assert not self.finalized_ui def apply_field(obj, field, condition=None, init_field=None): key = f"{path}/{field}" if getattr(obj, 'custom_script_source', None) is not None: key = f"customscript/{obj.custom_script_source}/{key}" if getattr(obj, 'do_not_save_to_config', False): return saved_value = self.ui_settings.get(key, None) if saved_value is None: self.ui_settings[key] = getattr(obj, field) elif condition and not condition(saved_value): pass else: setattr(obj, field, saved_value) if init_field is not None: init_field(saved_value) if field == 'value' and key not in self.component_mapping: self.component_mapping[key] = x if type(x) in [gr.Slider, gr.Radio, gr.Checkbox, gr.Textbox, gr.Number, gr.Dropdown, ToolButton, gr.Button] and x.visible: apply_field(x, 'visible') if type(x) == gr.Slider: apply_field(x, 'value') apply_field(x, 'minimum') apply_field(x, 'maximum') apply_field(x, 'step') if type(x) == gr.Radio: apply_field(x, 'value', lambda val: val in x.choices) if type(x) == gr.Checkbox: apply_field(x, 'value') if type(x) == gr.Textbox: apply_field(x, 'value') if type(x) == gr.Number: apply_field(x, 'value') if type(x) == gr.Dropdown: def check_dropdown(val): if getattr(x, 'multiselect', False): return all(value in x.choices for value in val) else: return val in x.choices apply_field(x, 'value', check_dropdown, getattr(x, 'init_field', None)) def check_tab_id(tab_id): tab_items = list(filter(lambda e: isinstance(e, gr.TabItem), x.children)) if type(tab_id) == str: tab_ids = [t.id for t in tab_items] return tab_id in tab_ids elif type(tab_id) == int: return 0 <= tab_id < len(tab_items) else: return False if type(x) == gr.Tabs: apply_field(x, 'selected', check_tab_id) def add_block(self, x, path=""): """adds all components inside a gradio block x to the registry of tracked components""" if hasattr(x, 'children'): if isinstance(x, gr.Tabs) and x.elem_id is not None: # Tabs element can't have a label, have to use elem_id instead self.add_component(f"{path}/Tabs@{x.elem_id}", x) for c in x.children: self.add_block(c, path) elif x.label is not None: self.add_component(f"{path}/{x.label}", x) elif isinstance(x, gr.Button) and x.value is not None: self.add_component(f"{path}/{x.value}", x) def read_from_file(self): with open(self.filename, "r", encoding="utf8") as file: return json.load(file) def write_to_file(self, current_ui_settings): with open(self.filename, "w", encoding="utf8") as file: json.dump(current_ui_settings, file, indent=4) def dump_defaults(self): """saves default values to a file unless tjhe file is present and there was an error loading default values at start""" if self.error_loading and os.path.exists(self.filename): return self.write_to_file(self.ui_settings) def iter_changes(self, current_ui_settings, values): """ given a dictionary with defaults from a file and current values from gradio elements, returns an iterator over tuples of values that are not the same between the file and the current; tuple contents are: path, old value, new value """ for (path, component), new_value in zip(self.component_mapping.items(), values): old_value = current_ui_settings.get(path) choices = getattr(component, 'choices', None) if isinstance(new_value, int) and choices: if new_value >= len(choices): continue new_value = choices[new_value] if new_value == old_value: continue if old_value is None and new_value == '' or new_value == []: continue yield path, old_value, new_value def ui_view(self, *values): text = [""] for path, old_value, new_value in self.iter_changes(self.read_from_file(), values): if old_value is None: old_value = "None" text.append(f"") if len(text) == 1: text.append("") text.append("") return "".join(text) def ui_apply(self, *values): num_changed = 0 current_ui_settings = self.read_from_file() for path, _, new_value in self.iter_changes(current_ui_settings.copy(), values): num_changed += 1 current_ui_settings[path] = new_value if num_changed == 0: return "No changes." self.write_to_file(current_ui_settings) return f"Wrote {num_changed} changes." def create_ui(self): """creates ui elements for editing defaults UI, without adding any logic to them""" gr.HTML( f"This page allows you to change default values in UI elements on other tabs.
" f"Make your changes, press 'View changes' to review the changed default values,
" f"then press 'Apply' to write them to {self.filename}.
" f"New defaults will apply after you restart the UI.
" ) with gr.Row(): self.ui_defaults_view = gr.Button(value='View changes', elem_id="ui_defaults_view", variant="secondary") self.ui_defaults_apply = gr.Button(value='Apply', elem_id="ui_defaults_apply", variant="primary") self.ui_defaults_review = gr.HTML("") def setup_ui(self): """adds logic to elements created with create_ui; all add_block class must be made before this""" assert not self.finalized_ui self.finalized_ui = True self.ui_defaults_view.click(fn=self.ui_view, inputs=list(self.component_mapping.values()), outputs=[self.ui_defaults_review]) self.ui_defaults_apply.click(fn=self.ui_apply, inputs=list(self.component_mapping.values()), outputs=[self.ui_defaults_review])
PathOld valueNew value
{path}{old_value}{new_value}
No changes