mirror of
https://github.com/openvinotoolkit/stable-diffusion-webui.git
synced 2024-12-15 15:13:45 +03:00
64311faa68
add VAE from main model into infotext, not from refiner model option to make scripts UI without gr.Group fix inconsistencies with refiner when usings samplers that do more denoising than steps
134 lines
3.7 KiB
Python
134 lines
3.7 KiB
Python
import gradio as gr
|
|
|
|
|
|
class FormComponent:
|
|
def get_expected_parent(self):
|
|
return gr.components.Form
|
|
|
|
|
|
gr.Dropdown.get_expected_parent = FormComponent.get_expected_parent
|
|
|
|
|
|
class ToolButton(FormComponent, gr.Button):
|
|
"""Small button with single emoji as text, fits inside gradio forms"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
classes = kwargs.pop("elem_classes", [])
|
|
super().__init__(*args, elem_classes=["tool", *classes], **kwargs)
|
|
|
|
def get_block_name(self):
|
|
return "button"
|
|
|
|
|
|
class FormRow(FormComponent, gr.Row):
|
|
"""Same as gr.Row but fits inside gradio forms"""
|
|
|
|
def get_block_name(self):
|
|
return "row"
|
|
|
|
|
|
class FormColumn(FormComponent, gr.Column):
|
|
"""Same as gr.Column but fits inside gradio forms"""
|
|
|
|
def get_block_name(self):
|
|
return "column"
|
|
|
|
|
|
class FormGroup(FormComponent, gr.Group):
|
|
"""Same as gr.Group but fits inside gradio forms"""
|
|
|
|
def get_block_name(self):
|
|
return "group"
|
|
|
|
|
|
class FormHTML(FormComponent, gr.HTML):
|
|
"""Same as gr.HTML but fits inside gradio forms"""
|
|
|
|
def get_block_name(self):
|
|
return "html"
|
|
|
|
|
|
class FormColorPicker(FormComponent, gr.ColorPicker):
|
|
"""Same as gr.ColorPicker but fits inside gradio forms"""
|
|
|
|
def get_block_name(self):
|
|
return "colorpicker"
|
|
|
|
|
|
class DropdownMulti(FormComponent, gr.Dropdown):
|
|
"""Same as gr.Dropdown but always multiselect"""
|
|
def __init__(self, **kwargs):
|
|
super().__init__(multiselect=True, **kwargs)
|
|
|
|
def get_block_name(self):
|
|
return "dropdown"
|
|
|
|
|
|
class DropdownEditable(FormComponent, gr.Dropdown):
|
|
"""Same as gr.Dropdown but allows editing value"""
|
|
def __init__(self, **kwargs):
|
|
super().__init__(allow_custom_value=True, **kwargs)
|
|
|
|
def get_block_name(self):
|
|
return "dropdown"
|
|
|
|
|
|
class InputAccordion(gr.Checkbox):
|
|
"""A gr.Accordion that can be used as an input - returns True if open, False if closed.
|
|
|
|
Actaully just a hidden checkbox, but creates an accordion that follows and is followed by the state of the checkbox.
|
|
"""
|
|
|
|
global_index = 0
|
|
|
|
def __init__(self, value, **kwargs):
|
|
self.accordion_id = kwargs.get('elem_id')
|
|
if self.accordion_id is None:
|
|
self.accordion_id = f"input-accordion-{InputAccordion.global_index}"
|
|
InputAccordion.global_index += 1
|
|
|
|
kwargs_checkbox = {
|
|
**kwargs,
|
|
"elem_id": f"{self.accordion_id}-checkbox",
|
|
"visible": False,
|
|
}
|
|
super().__init__(value, **kwargs_checkbox)
|
|
|
|
self.change(fn=None, _js='function(checked){ inputAccordionChecked("' + self.accordion_id + '", checked); }', inputs=[self])
|
|
|
|
kwargs_accordion = {
|
|
**kwargs,
|
|
"elem_id": self.accordion_id,
|
|
"label": kwargs.get('label', 'Accordion'),
|
|
"elem_classes": ['input-accordion'],
|
|
"open": value,
|
|
}
|
|
self.accordion = gr.Accordion(**kwargs_accordion)
|
|
|
|
def extra(self):
|
|
"""Allows you to put something into the label of the accordion.
|
|
|
|
Use it like this:
|
|
|
|
```
|
|
with InputAccordion(False, label="Accordion") as acc:
|
|
with acc.extra():
|
|
FormHTML(value="hello", min_width=0)
|
|
|
|
...
|
|
```
|
|
"""
|
|
|
|
return gr.Column(elem_id=self.accordion_id + '-extra', elem_classes='input-accordion-extra', min_width=0)
|
|
|
|
def __enter__(self):
|
|
self.accordion.__enter__()
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
self.accordion.__exit__(exc_type, exc_val, exc_tb)
|
|
|
|
def get_block_name(self):
|
|
return "checkbox"
|
|
|