Fix image distortion: preserve aspect ratio, remove stretched side-by-side

app.py

@@ -254,9 +254,7 @@ def process_image(image_rgb, procedure, intensity):
             f"Processing time: {elapsed:.2f}s\n"
             f"Mode: TPS (CPU)"
         )
-        side_by_side = np.hstack([image_rgb_512, composited_rgb])
-
-        return wireframe_rgb, mask_vis, composited_rgb, side_by_side, info
+        return wireframe_rgb, mask_vis, composited_rgb, image_rgb_512, info
 
     except Exception as exc:
         logger.error("Processing failed: %s\n%s", exc, traceback.format_exc())
@@ -483,7 +481,7 @@ with gr.Blocks(
                     out_mask = gr.Image(label="Surgical Mask", height=256)
                 with gr.Row():
                     out_result = gr.Image(label="Predicted Result", height=256)
-                    out_sidebyside = gr.Image(label="Before / After", height=256)
+                    out_sidebyside = gr.Image(label="Original", height=256)
 
         # -- Example images --
         if EXAMPLE_IMAGES: