Update app.py

app.py

@@ -69,7 +69,7 @@ sampler = SpacedSampler(model, var_type="fixed_small")
 def process(
     control_img: Image.Image,
     num_samples: int,
-    sr_scale: int,
+    sr_scale: float,
     strength: float,
     positive_prompt: str,
     negative_prompt: str,
@@ -159,14 +159,28 @@ def process(
 
     return preds
 
+def update_output_resolution(image):
+    if image is not None:
+        width, height = image.size
+        return f"Current resolution: {width}x{height}. Output resolution: {int(width*sr_scale.value)}x{int(height*sr_scale.value)}"
+    return "Upload an image to see the output resolution"
+
 block = gr.Blocks().queue()
 with block:
     with gr.Row():
         input_image = gr.Image(type="pil", label="Input Image")
+    
+    with gr.Row():
+        sr_scale = gr.Slider(label="SR Scale", minimum=1, maximum=8, value=4, step=0.1, info="Super-resolution scale factor.")
+    
+    output_resolution = gr.Markdown("Upload an image to see the output resolution")
+    
+    with gr.Row():
         run_button = gr.Button(value="Run")
-        with gr.Accordion("Options", open=True):
+    
+    with gr.Accordion("Options", open=False):
+        with gr.Column():
             num_samples = gr.Slider(label="Number Of Samples", minimum=1, maximum=12, value=1, step=1, info="Number of output images to generate.")
-            sr_scale = gr.Dropdown(label="SR Scale", choices=["2x", "4x", "8x"], value="4x", info="Super-resolution scale factor.")
             strength = gr.Slider(label="Control Strength", minimum=0.0, maximum=2.0, value=1.0, step=0.01, info="Strength of the control signal.")
             positive_prompt = gr.Textbox(label="Positive Prompt", value="", info="Positive text prompt to guide the image generation.")
             negative_prompt = gr.Textbox(
@@ -184,6 +198,7 @@ with block:
             tile_vae = gr.Checkbox(label="Tile VAE", value=True, info="Enable tiled VAE for large images.")
             vae_encoder_tile_size = gr.Slider(label="Encoder tile size", minimum=512, maximum=5000, value=1024, step=256, info="Size of each tile for the VAE encoder.")
             vae_decoder_tile_size = gr.Slider(label="Decoder tile size", minimum=64, maximum=512, value=224, step=128, info="Size of each tile for the VAE decoder.")
+    
     with gr.Column():
         result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery")
 
@@ -207,4 +222,15 @@ with block:
     ]
     run_button.click(fn=process, inputs=inputs, outputs=[result_gallery])
 
+    # Update output resolution when image is uploaded or SR scale is changed
+    input_image.change(update_output_resolution, inputs=[input_image], outputs=[output_resolution])
+    sr_scale.change(update_output_resolution, inputs=[input_image], outputs=[output_resolution])
+
+    # Disable SR scale slider when no image is uploaded
+    input_image.change(
+        lambda x: gr.update(interactive=x is not None),
+        inputs=[input_image],
+        outputs=[sr_scale]
+    )
+
 block.launch()