add option to change what's shown in quicksettings bar

2 files changed, 11 insertions, 9 deletions

diff --git a/modules/shared.py b/modules/shared.py
index 5f6101a4..4d3ed625 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -152,7 +152,6 @@ class OptionInfo:
         self.component_args = component_args

         self.onchange = onchange

         self.section = None

-        self.show_on_main_page = show_on_main_page

 

 

 def options_section(section_identifier, options_dict):

@@ -237,7 +236,7 @@ options_templates.update(options_section(('training', "Training"), {
 }))

 

 options_templates.update(options_section(('sd', "Stable Diffusion"), {

-    "sd_model_checkpoint": OptionInfo(None, "Stable Diffusion checkpoint", gr.Dropdown, lambda: {"choices": modules.sd_models.checkpoint_tiles()}, show_on_main_page=True),

+    "sd_model_checkpoint": OptionInfo(None, "Stable Diffusion checkpoint", gr.Dropdown, lambda: {"choices": modules.sd_models.checkpoint_tiles()}),

     "sd_hypernetwork": OptionInfo("None", "Stable Diffusion finetune hypernetwork", gr.Dropdown, lambda: {"choices": ["None"] + [x for x in hypernetworks.keys()]}),

     "img2img_color_correction": OptionInfo(False, "Apply color correction to img2img results to match original colors."),

     "save_images_before_color_correction": OptionInfo(False, "Save a copy of image before applying color correction to img2img results"),

@@ -250,6 +249,7 @@ options_templates.update(options_section(('sd', "Stable Diffusion"), {
     "filter_nsfw": OptionInfo(False, "Filter NSFW content"),

     'CLIP_stop_at_last_layers': OptionInfo(1, "Stop At last layers of CLIP model", gr.Slider, {"minimum": 1, "maximum": 12, "step": 1}),

     "random_artist_categories": OptionInfo([], "Allowed categories for random artists selection when using the Roll button", gr.CheckboxGroup, {"choices": artist_db.categories()}),

+    'quicksettings': OptionInfo("sd_model_checkpoint", "Quicksettings list"),

 }))

 

 options_templates.update(options_section(('interrogate', "Interrogate Options"), {

diff --git a/modules/ui.py b/modules/ui.py
index e07ee0e1..a0529860 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1305,6 +1305,9 @@ Requested path was: {f}
         settings_cols = 3

         items_per_col = int(len(opts.data_labels) * 0.9 / settings_cols)

 

+        quicksettings_names = [x.strip() for x in opts.quicksettings.split(",")]

+        quicksettings_names = set(x for x in quicksettings_names if x != 'quicksettings')

+

         quicksettings_list = []

 

         cols_displayed = 0

@@ -1329,7 +1332,7 @@ Requested path was: {f}
 

                     gr.HTML(elem_id="settings_header_text_{}".format(item.section[0]), value='<h1 class="gr-button-lg">{}</h1>'.format(item.section[1]))

 

-                if item.show_on_main_page:

+                if k in quicksettings_names:

                     quicksettings_list.append((i, k, item))

                     components.append(dummy_component)

                 else:

@@ -1338,7 +1341,11 @@ Requested path was: {f}
                     components.append(component)

                     items_displayed += 1

 

-        request_notifications = gr.Button(value='Request browser notifications', elem_id="request_notifications")

+        with gr.Row():

+            request_notifications = gr.Button(value='Request browser notifications', elem_id="request_notifications")

+            reload_script_bodies = gr.Button(value='Reload custom script bodies (No ui updates, No restart)', variant='secondary')

+            restart_gradio = gr.Button(value='Restart Gradio and Refresh components (Custom Scripts, ui.py, js and css only)', variant='primary')

+

         request_notifications.click(

             fn=lambda: None,

             inputs=[],

@@ -1346,10 +1353,6 @@ Requested path was: {f}
             _js='function(){}'

         )

 

-        with gr.Row():

-            reload_script_bodies = gr.Button(value='Reload custom script bodies (No ui updates, No restart)', variant='secondary')

-            restart_gradio = gr.Button(value='Restart Gradio and Refresh components (Custom Scripts, ui.py, js and css only)', variant='primary')

-

         def reload_scripts():

             modules.scripts.reload_script_body_only()

 

@@ -1364,7 +1367,6 @@ Requested path was: {f}
             shared.state.interrupt()

             settings_interface.gradio_ref.do_restart = True

 

-

         restart_gradio.click(

             fn=request_restart,

             inputs=[],