| import math | |
| import gradio as gr | |
| from modules import scripts, shared, ui_components, ui_settings, generation_parameters_copypaste | |
| from modules.ui_components import FormColumn | |
| class ExtraOptionsSection(scripts.Script): | |
| section = "extra_options" | |
| def __init__(self): | |
| self.comps = None | |
| self.setting_names = None | |
| def title(self): | |
| return "Extra options" | |
| def show(self, is_img2img): | |
| return scripts.AlwaysVisible | |
| def ui(self, is_img2img): | |
| self.comps = [] | |
| self.setting_names = [] | |
| self.infotext_fields = [] | |
| extra_options = shared.opts.extra_options_img2img if is_img2img else shared.opts.extra_options_txt2img | |
| mapping = {k: v for v, k in generation_parameters_copypaste.infotext_to_setting_name_mapping} | |
| with gr.Blocks() as interface: | |
| with gr.Accordion("Options", open=False) if shared.opts.extra_options_accordion and extra_options else gr.Group(): | |
| row_count = math.ceil(len(extra_options) / shared.opts.extra_options_cols) | |
| for row in range(row_count): | |
| with gr.Row(): | |
| for col in range(shared.opts.extra_options_cols): | |
| index = row * shared.opts.extra_options_cols + col | |
| if index >= len(extra_options): | |
| break | |
| setting_name = extra_options[index] | |
| with FormColumn(): | |
| comp = ui_settings.create_setting_component(setting_name) | |
| self.comps.append(comp) | |
| self.setting_names.append(setting_name) | |
| setting_infotext_name = mapping.get(setting_name) | |
| if setting_infotext_name is not None: | |
| self.infotext_fields.append((comp, setting_infotext_name)) | |
| def get_settings_values(): | |
| res = [ui_settings.get_value_for_setting(key) for key in self.setting_names] | |
| return res[0] if len(res) == 1 else res | |
| interface.load(fn=get_settings_values, inputs=[], outputs=self.comps, queue=False, show_progress=False) | |
| return self.comps | |
| def before_process(self, p, *args): | |
| for name, value in zip(self.setting_names, args): | |
| if name not in p.override_settings: | |
| p.override_settings[name] = value | |
| shared.options_templates.update(shared.options_section(('ui', "User interface"), { | |
| "extra_options_txt2img": shared.OptionInfo([], "Options in main UI - txt2img", ui_components.DropdownMulti, lambda: {"choices": list(shared.opts.data_labels.keys())}).js("info", "settingsHintsShowQuicksettings").info("setting entries that also appear in txt2img interfaces").needs_reload_ui(), | |
| "extra_options_img2img": shared.OptionInfo([], "Options in main UI - img2img", ui_components.DropdownMulti, lambda: {"choices": list(shared.opts.data_labels.keys())}).js("info", "settingsHintsShowQuicksettings").info("setting entries that also appear in img2img interfaces").needs_reload_ui(), | |
| "extra_options_cols": shared.OptionInfo(1, "Options in main UI - number of columns", gr.Number, {"precision": 0}).needs_reload_ui(), | |
| "extra_options_accordion": shared.OptionInfo(False, "Options in main UI - place into an accordion").needs_reload_ui() | |
| })) | |