From 87702febe08bddb9e8b952204c459927532ec6e6 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Fri, 19 May 2023 19:04:07 +0300
Subject: [PATCH] allow hiding buttons in ui-config.json

---
 modules/ui_loadsave.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/modules/ui_loadsave.py b/modules/ui_loadsave.py
index 728fec9ea..0052a5cce 100644
--- a/modules/ui_loadsave.py
+++ b/modules/ui_loadsave.py
@@ -55,7 +55,7 @@ class UiLoadsave:
             if field == 'value' and key not in self.component_mapping:
                 self.component_mapping[key] = x
 
-        if type(x) in [gr.Slider, gr.Radio, gr.Checkbox, gr.Textbox, gr.Number, gr.Dropdown, ToolButton] and x.visible:
+        if type(x) in [gr.Slider, gr.Radio, gr.Checkbox, gr.Textbox, gr.Number, gr.Dropdown, ToolButton, gr.Button] and x.visible:
             apply_field(x, 'visible')
 
         if type(x) == gr.Slider:
@@ -109,6 +109,8 @@ class UiLoadsave:
                 self.add_block(c, path)
         elif x.label is not None:
             self.add_component(f"{path}/{x.label}", x)
+        elif isinstance(x, gr.Button) and x.value is not None:
+            self.add_component(f"{path}/{x.value}", x)
 
     def read_from_file(self):
         with open(self.filename, "r", encoding="utf8") as file:
-- 
GitLab