diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-01 10:08:40 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-01 10:08:40 +0000 |
| commit | e5f1a37cb9b537d95b2df47c96b4a4f7242fd294 (patch) | |
| tree | 21bd3560ed8bc561745b77a3f39ad927c48e0486 /modules | |
| parent | b46b97fa297b3a4a654da77cf98a775a2bcab4c7 (diff) | |
| download | stable-diffusion-webui-gfx803-e5f1a37cb9b537d95b2df47c96b4a4f7242fd294.tar.gz stable-diffusion-webui-gfx803-e5f1a37cb9b537d95b2df47c96b4a4f7242fd294.tar.bz2 stable-diffusion-webui-gfx803-e5f1a37cb9b537d95b2df47c96b4a4f7242fd294.zip | |
make refresh buttons look more nice
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/ui.py | 6 | ||||
| -rw-r--r-- | modules/ui_components.py | 18 |
2 files changed, 21 insertions, 3 deletions
diff --git a/modules/ui.py b/modules/ui.py index 4cc2ce4f..32fa80d1 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -19,7 +19,7 @@ import numpy as np from PIL import Image, PngImagePlugin
from modules.call_queue import wrap_gradio_gpu_call, wrap_queued_call, wrap_gradio_call
-from modules import sd_hijack, sd_models, localization, script_callbacks, ui_extensions, deepbooru
+from modules import sd_hijack, sd_models, localization, script_callbacks, ui_extensions, deepbooru, ui_components
from modules.paths import script_path
from modules.shared import opts, cmd_opts, restricted_opts
@@ -532,7 +532,7 @@ def create_refresh_button(refresh_component, refresh_method, refreshed_args, ele return gr.update(**(args or {}))
- refresh_button = gr.Button(value=refresh_symbol, elem_id=elem_id)
+ refresh_button = ui_components.ToolButton(value=refresh_symbol, elem_id=elem_id)
refresh_button.click(
fn=refresh,
inputs=[],
@@ -1476,7 +1476,7 @@ def create_ui(): res = comp(label=info.label, value=fun(), elem_id=elem_id, **(args or {}))
create_refresh_button(res, info.refresh, info.component_args, "refresh_" + key)
else:
- with gr.Row(variant="compact"):
+ with ui_components.FormRow():
res = comp(label=info.label, value=fun(), elem_id=elem_id, **(args or {}))
create_refresh_button(res, info.refresh, info.component_args, "refresh_" + key)
else:
diff --git a/modules/ui_components.py b/modules/ui_components.py new file mode 100644 index 00000000..d0519d2d --- /dev/null +++ b/modules/ui_components.py @@ -0,0 +1,18 @@ +import gradio as gr
+
+
+class ToolButton(gr.Button, gr.components.FormComponent):
+ """Small button with single emoji as text, fits inside gradio forms"""
+
+ def __init__(self, **kwargs):
+ super().__init__(variant="tool", **kwargs)
+
+ def get_block_name(self):
+ return "button"
+
+
+class FormRow(gr.Row, gr.components.FormComponent):
+ """Same as gr.Row but fits inside gradio forms"""
+
+ def get_block_name(self):
+ return "row"
|