Change extras "scale to" to sliders

1 files changed, 10 insertions, 4 deletions

diff --git a/scripts/postprocessing_upscale.py b/scripts/postprocessing_upscale.py
index 11eab31a..bc43719b 100644
--- a/scripts/postprocessing_upscale.py
+++ b/scripts/postprocessing_upscale.py
@@ -4,8 +4,9 @@ import numpy as np
 from modules import scripts_postprocessing, shared

 import gradio as gr

 

-from modules.ui_components import FormRow

+from modules.ui_components import FormRow, ToolButton

 

+switch_values_symbol = '\U000021C5' # ⇅

 

 upscale_cache = {}

 

@@ -25,9 +26,12 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
 

                     with gr.TabItem('Scale to', elem_id="extras_scale_to_tab") as tab_scale_to:

                         with FormRow():

-                            upscaling_resize_w = gr.Number(label="Width", value=512, precision=0, elem_id="extras_upscaling_resize_w")

-                            upscaling_resize_h = gr.Number(label="Height", value=512, precision=0, elem_id="extras_upscaling_resize_h")

-                            upscaling_crop = gr.Checkbox(label='Crop to fit', value=True, elem_id="extras_upscaling_crop")

+                            with gr.Column(elem_id="upscaling_column_size", scale=4):

+                                upscaling_resize_w = gr.Slider(minimum=64, maximum=2048, step=8, label="Width", value=512, elem_id="extras_upscaling_resize_w")

+                                upscaling_resize_h = gr.Slider(minimum=64, maximum=2048, step=8, label="Height", value=512, elem_id="extras_upscaling_resize_w")

+                            with gr.Column(elem_id="upscaling_dimensions_row", scale=1, elem_classes="dimensions-tools"):

+                                upscaling_res_switch_btn = ToolButton(value=switch_values_symbol, elem_id="upscaling_res_switch_btn")

+                                upscaling_crop = gr.Checkbox(label='Crop to fit', value=True, elem_id="extras_upscaling_crop")

 

             with FormRow():

                 extras_upscaler_1 = gr.Dropdown(label='Upscaler 1', elem_id="extras_upscaler_1", choices=[x.name for x in shared.sd_upscalers], value=shared.sd_upscalers[0].name)

@@ -36,6 +40,7 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
                 extras_upscaler_2 = gr.Dropdown(label='Upscaler 2', elem_id="extras_upscaler_2", choices=[x.name for x in shared.sd_upscalers], value=shared.sd_upscalers[0].name)

                 extras_upscaler_2_visibility = gr.Slider(minimum=0.0, maximum=1.0, step=0.001, label="Upscaler 2 visibility", value=0.0, elem_id="extras_upscaler_2_visibility")

 

+        upscaling_res_switch_btn.click(lambda w, h: (h, w), inputs=[upscaling_resize_w, upscaling_resize_h], outputs=[upscaling_resize_w, upscaling_resize_h], show_progress=False)

         tab_scale_by.select(fn=lambda: 0, inputs=[], outputs=[selected_tab])

         tab_scale_to.select(fn=lambda: 1, inputs=[], outputs=[selected_tab])

 

@@ -45,6 +50,7 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
             "upscale_to_width": upscaling_resize_w,

             "upscale_to_height": upscaling_resize_h,

             "upscale_crop": upscaling_crop,

+            "upscaling_res_switch_btn": upscaling_res_switch_btn,

             "upscaler_1_name": extras_upscaler_1,

             "upscaler_2_name": extras_upscaler_2,

             "upscaler_2_visibility": extras_upscaler_2_visibility,