Update app.py

app.py

@@ -72,7 +72,8 @@ def process_images(flat_img, persp_img, json_file):
 
     # Load JSON
     try:
-        data = json.load(open(json_file.name))
+        with open(json_file.name, 'r') as f:
+            data = json.load(f)
     except Exception as e:
         print("JSON read error:", e)
         return []
@@ -105,14 +106,19 @@ def process_images(flat_img, persp_img, json_file):
         roi_corners_flat = get_rotated_rect_corners(roi_x, roi_y, roi_w, roi_h, roi_rot_deg)
         roi_corners_persp = cv2.perspectiveTransform(roi_corners_flat.reshape(-1,1,2), H).reshape(-1,2)
 
-        # Draw ROI on Perspective image only
+        # Draw ROI & label
+        flat_out = flat_img.copy()
         persp_out = persp_img.copy()
+        cv2.polylines(flat_out, [roi_corners_flat.astype(int)], True, (255,0,0), 3)
         cv2.polylines(persp_out, [roi_corners_persp.astype(int)], True, (0,255,0), 3)
+        cv2.putText(flat_out, det, (10,30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,255,255), 2)
+        cv2.putText(persp_out, det, (10,30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,255,255), 2)
 
-        # Add detector label
-        cv2.putText(persp_out, f"{det}", (10,30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,255,255), 2)
+        # Convert to RGB for Gradio
+        flat_rgb = cv2.cvtColor(flat_out, cv2.COLOR_BGR2RGB)
+        persp_rgb = cv2.cvtColor(persp_out, cv2.COLOR_BGR2RGB)
 
-        gallery_images.append(persp_out)
+        gallery_images.extend([flat_rgb, persp_rgb])
 
     return gallery_images
 
@@ -125,10 +131,10 @@ iface = gr.Interface(
         gr.File(type="filepath", label="JSON File")
     ],
     outputs=[
-        gr.Gallery(label="Detector-wise Perspective ROI Projection", show_label=True)
+        gr.Gallery(label="Detector Results (Flat + Perspective per Detector)")
     ],
     title="Feature Detection with ROI Projection",
-    description="Shows SIFT, ORB, BRISK, AKAZE, KAZE feature-based ROI projections on the Perspective image only."
+    description="Shows SIFT, ORB, BRISK, AKAZE, KAZE feature-based ROI projections. Each detector outputs Flat and Perspective images with labels."
 )
 
 iface.launch()