Make extra networks button togglable

2 files changed, 9 insertions, 2 deletions

diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 01df5e90..50e2093e 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -219,10 +219,10 @@ def create_ui(container, button, tabname):
 

     def toggle_visibility(is_visible):

         is_visible = not is_visible

-        return is_visible, gr.update(visible=is_visible)

+        return is_visible, gr.update(visible=is_visible), gr.update(variant=("primary" if is_visible else "tool"))

 

     state_visible = gr.State(value=False)

-    button.click(fn=toggle_visibility, inputs=[state_visible], outputs=[state_visible, container])

+    button.click(fn=toggle_visibility, inputs=[state_visible], outputs=[state_visible, container, button])

 

     def refresh():

         res = []

diff --git a/style.css b/style.css
index 2f26ad02..e82aebf3 100644
--- a/style.css
+++ b/style.css
@@ -968,3 +968,10 @@ footer {
 [id*='_prompt_container'] > div {

 	margin: 0!important;

 }

+

+button[id$='_extra_networks'] {

+    margin: 0.6em 0em 0.55em 0;

+    max-width: 2.5em;

+    min-width: 2.5em !important;

+    height: 2.4em;

+}