Update app.py

app.py

@@ -88,10 +88,16 @@ def homography_demo(flat_file, persp_file, json_file):
 
     for method in methods:
         kp1, kp2, good_matches = detect_and_match(flat_gray, persp_gray, method=method)
+        if kp1 is None or kp2 is None or len(good_matches) < 4:
+            continue
+        
         src_pts = np.float32([kp1[m.queryIdx].pt for m in good_matches]).reshape(-1,1,2)
         dst_pts = np.float32([kp2[m.trainIdx].pt for m in good_matches]).reshape(-1,1,2)
         H, mask = cv2.findHomography(src_pts, dst_pts, cv2.RANSAC, 5.0)
 
+        if H is None:
+            continue
+
         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)
 
@@ -100,7 +106,7 @@ def homography_demo(flat_file, persp_file, json_file):
         for (px, py) in roi_corners_persp:
             cv2.circle(persp_debug, (int(px), int(py)), 5, (255,0,0), -1)
 
-        outputs.append(cv2.cvtColor(persp_debug, cv2.COLOR_BGR2RGB))
+        outputs.append((f"{method} Result", cv2.cvtColor(persp_debug, cv2.COLOR_BGR2RGB)))
 
     return outputs
 
@@ -111,7 +117,10 @@ with gr.Blocks() as demo:
         flat_input = gr.File(label="Upload Flat Image", file_types=[".jpg",".png",".jpeg"])
         persp_input = gr.File(label="Upload Perspective Image", file_types=[".jpg",".png",".jpeg"])
         json_input = gr.File(label="Upload mockup.json", file_types=[".json"])
-    output_gallery = gr.Gallery(label="Perspective ROI Results", columns=2, height=400)
+    with gr.Row():
+        gr.Image(type="filepath", label="Flat Image Preview").style(height=200)
+        gr.Image(type="filepath", label="Perspective Image Preview").style(height=200)
+    output_gallery = gr.Gallery(label="Perspective ROI Results", show_label=True, columns=2, height=600)
     run_btn = gr.Button("Run Homography")
     run_btn.click(homography_demo, inputs=[flat_input, persp_input, json_input], outputs=output_gallery)