mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-06-07 21:20:49 +00:00
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")
|
||
|
|
}))
|