Update app.py

app.py

@@ -152,6 +152,8 @@ def update_output_resolution(image, scale_choice, custom_scale):
         width, height = image.size
         if scale_choice == "Custom":
             scale = custom_scale
+        elif "%" in scale_choice:
+            scale = float(scale_choice.split()[-1].strip("()%")) / 100
         else:
             scale = float(scale_choice.split()[-1].strip("()x"))
         return f"Current resolution: {width}x{height}. Output resolution: {int(width*scale)}x{int(height*scale)}"
@@ -171,8 +173,16 @@ def update_scale_choices(image):
                 scale = max(w/width, h/height)
                 if scale > 1:
                     choices.append(f"{w}x{h} ({scale:.2f}x)")
+        
+        if not choices:  # If no common resolutions fit, use percentage-based options
+            choices = [
+                f"{width*2}x{height*2} (200%)",
+                f"{width*4}x{height*4} (400%)",
+                f"{width*8}x{height*8} (800%)"
+            ]
+        
         choices.append("Custom")
-        return gr.update(choices=choices, value=choices[1] if len(choices) > 1 else "Custom")
+        return gr.update(choices=choices, value=choices[0])
     return gr.update(choices=["Custom"], value="Custom")
 
 # Improved UI design
@@ -235,6 +245,8 @@ with gr.Blocks(css=css) as block:
     def get_scale_value(choice, custom):
         if choice == "Custom":
             return custom
+        if "%" in choice:
+            return float(choice.split()[-1].strip("()%")) / 100
         return float(choice.split()[-1].strip("()x"))
 
     inputs = [
@@ -243,7 +255,7 @@ with gr.Blocks(css=css) as block:
         tile_diffusion_stride
     ]
     run_button.click(
-        fn=lambda *args: process(*args[:1], args[1], get_scale_value(args[2], args[-1]), *args[3:]),
+        fn=lambda *args: process(args[0], args[1], get_scale_value(args[2], args[-1]), *args[3:-1]),
         inputs=inputs + [custom_scale],
         outputs=[result_gallery]
     )
@@ -276,4 +288,4 @@ with gr.Blocks(css=css) as block:
         outputs=[sr_scale]
     )
 
-block.launch()
+block.launch(share=True)