options to refresh list of models and hypernetworks

1 files changed, 29 insertions, 4 deletions

diff --git a/modules/ui.py b/modules/ui.py
index a0529860..0a58f6be 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -78,6 +78,8 @@ reuse_symbol = '\u267b\ufe0f'  # ♻️
 art_symbol = '\U0001f3a8'  # 🎨

 paste_symbol = '\u2199\ufe0f'  # ↙

 folder_symbol = '\U0001f4c2'  # 📂

+refresh_symbol = '\U0001f504'  # 🔄

+

 

 def plaintext_to_html(text):

     text = "<p>" + "<br>\n".join([f"{html.escape(x)}" for x in text.split('\n')]) + "</p>"

@@ -1210,8 +1212,7 @@ def create_ui(wrap_gradio_gpu_call):
             outputs=[],

         )

 

-

-    def create_setting_component(key):

+    def create_setting_component(key, is_quicksettings=False):

         def fun():

             return opts.data[key] if key in opts.data else opts.data_labels[key].default

 

@@ -1231,7 +1232,31 @@ def create_ui(wrap_gradio_gpu_call):
         else:

             raise Exception(f'bad options item type: {str(t)} for key {key}')

 

-        return comp(label=info.label, value=fun, **(args or {}))

+        if info.refresh is not None:

+            if is_quicksettings:

+                res = comp(label=info.label, value=fun, **(args or {}))

+                refresh_button = gr.Button(value=refresh_symbol, elem_id="refresh_"+key)

+            else:

+                with gr.Row(variant="compact"):

+                    res = comp(label=info.label, value=fun, **(args or {}))

+                    refresh_button = gr.Button(value=refresh_symbol, elem_id="refresh_" + key)

+

+            def refresh():

+                info.refresh()

+                refreshed_args = info.component_args() if callable(info.component_args) else info.component_args

+                res.choices = refreshed_args["choices"]

+                return gr.update(**(refreshed_args or {}))

+

+            refresh_button.click(

+                fn=refresh,

+                inputs=[],

+                outputs=[res],

+            )

+        else:

+            res = comp(label=info.label, value=fun, **(args or {}))

+

+

+        return res

 

     components = []

     component_dict = {}

@@ -1401,7 +1426,7 @@ Requested path was: {f}
     with gr.Blocks(css=css, analytics_enabled=False, title="Stable Diffusion") as demo:

         with gr.Row(elem_id="quicksettings"):

             for i, k, item in quicksettings_list:

-                component = create_setting_component(k)

+                component = create_setting_component(k, is_quicksettings=True)

                 component_dict[k] = component

 

         settings_interface.gradio_ref = demo