mirror of
https://github.com/openvinotoolkit/stable-diffusion-webui.git
synced 2024-12-15 07:03:06 +03:00
df02498d03
split some code from ui.py into ui_settings.py ui_gradio_edxtensions.py add before_process callback for scripts add ability for alwayson scripts to specify section and let user reorder those sections
49 lines
1.9 KiB
Python
49 lines
1.9 KiB
Python
import gradio as gr
|
|
from modules import scripts, shared, ui_components, ui_settings
|
|
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 = []
|
|
|
|
with gr.Blocks() as interface:
|
|
with gr.Accordion("Options", open=False) if shared.opts.extra_options_accordion and len(shared.opts.extra_options) > 0 else gr.Group(), gr.Row():
|
|
for setting_name in shared.opts.extra_options:
|
|
with FormColumn():
|
|
comp = ui_settings.create_setting_component(setting_name)
|
|
|
|
self.comps.append(comp)
|
|
self.setting_names.append(setting_name)
|
|
|
|
def get_settings_values():
|
|
return [ui_settings.get_value_for_setting(key) for key in self.setting_names]
|
|
|
|
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": shared.OptionInfo([], "Options in main UI", ui_components.DropdownMulti, lambda: {"choices": list(shared.opts.data_labels.keys())}).js("info", "settingsHintsShowQuicksettings").info("setting entries that also appear in txt2img/img2img interfaces").needs_restart(),
|
|
"extra_options_accordion": shared.OptionInfo(False, "Place options in main UI into an accordion")
|
|
}))
|