FIX (#7)

- FIX (d6ccb8b55c12b595b3eed28ae8b50ea60fc9419d)

app.py

@@ -4,26 +4,17 @@ import numpy as np
 from PIL import Image
 
 def flatten_image(img, points):
-    """
-    img: PIL.Image
-    points: list of (x, y) tuples in order [TL, TR, BR, BL]
-    """
     if img is None or not points or len(points) != 4:
         return None, "Please click exactly 4 points (TL, TR, BR, BL)."
 
-    # Convert to numpy array
     image_np = np.array(img)
     h, w = image_np.shape[:2]
-
-    # Convert input points to float32 numpy array
     src_pts = np.array(points, dtype=np.float32)
 
-    # Compute output rectangle size using distances
     width_top = np.linalg.norm(src_pts[0] - src_pts[1])
     width_bottom = np.linalg.norm(src_pts[3] - src_pts[2])
     height_left = np.linalg.norm(src_pts[0] - src_pts[3])
     height_right = np.linalg.norm(src_pts[1] - src_pts[2])
-
     max_width = int(max(width_top, width_bottom))
     max_height = int(max(height_left, height_right))
 
@@ -34,21 +25,16 @@ def flatten_image(img, points):
         [0, max_height - 1]
     ], dtype=np.float32)
 
-    # Compute homography
     M = cv2.getPerspectiveTransform(src_pts, dst_pts)
-
-    # Apply perspective warp
     warped = cv2.warpPerspective(image_np, M, (max_width, max_height), flags=cv2.INTER_CUBIC)
-
-    warped_pil = Image.fromarray(warped)
-    return warped_pil, None
+    return Image.fromarray(warped), None
 
 
 with gr.Blocks() as demo:
     gr.Markdown("## 📸 Perspective Flatten Tool\nUpload an image, click 4 corners (Top-Left → Top-Right → Bottom-Right → Bottom-Left), then flatten!")
 
     with gr.Row():
-        input_image = gr.Image(label="Upload Image", tool="select", type="pil")
+        input_image = gr.Image(label="Upload Image", type="pil")
         output_image = gr.Image(label="Flattened Output")
 
     coords = gr.State([])
@@ -56,17 +42,17 @@ with gr.Blocks() as demo:
     def collect_points(evt: gr.SelectData, points):
         if points is None:
             points = []
-        points.append(evt.index)  # evt.index returns (x, y)
+        points.append(evt.index)  # evt.index = (x, y)
         if len(points) > 4:
-            points = points[-4:]  # keep only last 4
+            points = points[-4:]
         return points, f"Selected {len(points)}/4 points: {points}"
 
     points_output = gr.Textbox(label="Selected Points", interactive=False)
 
+    # The .select event still works; we just don’t declare tool="select"
     input_image.select(fn=collect_points, inputs=coords, outputs=[coords, points_output])
 
     flatten_btn = gr.Button("🔄 Flatten Image")
-
     error_box = gr.Textbox(label="Messages", interactive=False)
 
     flatten_btn.click(fn=flatten_image, inputs=[input_image, coords], outputs=[output_image, error_box])