Update app.py

app.py

@@ -57,6 +57,31 @@ def fit_to_ratio(image: Image.Image, ratio_name: str, long_side: int) -> Image.I
     return image.resize((width, height), Image.LANCZOS)
 
 
+def get_tile_dimensions(ratio_name: str, tile_preset: str):
+    long_side = int(tile_preset)
+    rw, rh = RATIO_MAP[ratio_name]
+
+    if rw >= rh:
+        tile_width = long_side
+        tile_height = round(long_side * rh / rw)
+    else:
+        tile_height = long_side
+        tile_width = round(long_side * rw / rh)
+
+    tile_width = max(2, tile_width - (tile_width % 2))
+    tile_height = max(2, tile_height - (tile_height % 2))
+
+    return tile_width, tile_height
+
+
+def update_tile_display(ratio_name: str, tile_preset: str):
+    tile_width, tile_height = get_tile_dimensions(ratio_name, tile_preset)
+    return (
+        f"**Tile Width:** {tile_width}px  \n"
+        f"**Tile Height:** {tile_height}px"
+    )
+
+
 # ---------------------------------
 # GPU function
 # ---------------------------------
@@ -65,11 +90,10 @@ def fit_to_ratio(image: Image.Image, ratio_name: str, long_side: int) -> Image.I
 def upscale(
     image,
     enhancer_type,
-    tile_width,
-    tile_height,
     use_ratio,
     ratio_name,
     target_long_side,
+    tile_preset,
 ):
     if image is None:
         return None, None
@@ -79,13 +103,15 @@ def upscale(
     if use_ratio:
         img = fit_to_ratio(img, ratio_name, int(target_long_side))
 
+    tile_width, tile_height = get_tile_dimensions(ratio_name, tile_preset)
+
     upscaler = get_upscaler(enhancer_type)
 
     out = upscaler(
         img,
         tiling=True,
-        tile_width=int(tile_width),
-        tile_height=int(tile_height),
+        tile_width=tile_width,
+        tile_height=tile_height,
     )
 
     if not isinstance(out, Image.Image):
@@ -117,19 +143,6 @@ with gr.Blocks() as demo:
         label="Enhancer Type"
     )
 
-    with gr.Row():
-        tile_width = gr.Number(
-            value=1024,
-            label="Tile Width",
-            precision=0
-        )
-
-        tile_height = gr.Number(
-            value=1024,
-            label="Tile Height",
-            precision=0
-        )
-
     with gr.Group():
         use_ratio = gr.Checkbox(
             label="Use Aspect Ratio Preset",
@@ -148,6 +161,19 @@ with gr.Blocks() as demo:
             precision=0
         )
 
+    with gr.Group():
+        gr.Markdown("### Tile Size Preset")
+
+        tile_preset = gr.Radio(
+            choices=["512", "768", "1024"],
+            value="768",
+            label="Preset Size"
+        )
+
+        tile_display = gr.Markdown(
+            value=update_tile_display("1:1", "768")
+        )
+
     run_button = gr.Button("Enhance 4x")
 
     output_image = gr.Image(
@@ -160,16 +186,27 @@ with gr.Blocks() as demo:
         label="Download Full PNG"
     )
 
+    ratio_name.change(
+        fn=update_tile_display,
+        inputs=[ratio_name, tile_preset],
+        outputs=tile_display
+    )
+
+    tile_preset.change(
+        fn=update_tile_display,
+        inputs=[ratio_name, tile_preset],
+        outputs=tile_display
+    )
+
     run_button.click(
         fn=upscale,
         inputs=[
             input_image,
             enhancer_type,
-            tile_width,
-            tile_height,
             use_ratio,
             ratio_name,
             target_long_side,
+            tile_preset,
         ],
         outputs=[
             output_image,