Update app.py

app.py

@@ -71,7 +71,7 @@ def detect_and_match(img1_gray, img2_gray, method="SIFT", ratio_thresh=0.78):
             good.append(m)
     return kp1, kp2, good
 
-# ---------------- Main Homography Function (unchanged logic) ----------------
+# ---------------- Main Homography Function ----------------
 def homography_all_detectors(flat_file, persp_file, json_file):
     flat_img = cv2.imread(flat_file.name)
     persp_img = cv2.imread(persp_file.name)
@@ -122,70 +122,30 @@ def homography_all_detectors(flat_file, persp_file, json_file):
         gallery_images.append((result_rgb, f"{method} Result"))
         download_files.append(file_name)
 
-    # pad to 5 download slots
+    # return gallery + 5 download files (pad with None if less)
     while len(download_files) < 5:
         download_files.append(None)
 
-    # return: gallery + 5 files
     return [gallery_images] + download_files[:5]
 
-# ---------------- Preview in the SAME Gallery (on upload) ----------------
-def preview_uploaded_gallery(flat_file, persp_file):
-    """
-    Returns a gallery list that shows the uploaded images with captions,
-    without running homography. Used on upload change events.
-    """
-    items = []
-    if flat_file is not None:
-        flat_img = cv2.imread(flat_file.name)
-        if flat_img is not None:
-            items.append((cv2.cvtColor(flat_img, cv2.COLOR_BGR2RGB), "Flat (Uploaded)"))
-    if persp_file is not None:def get_rotated_rect_corners(x, y, w, h, rotation_deg):
-    r
-        persp_img = cv2.imread(persp_file.name)
-        if persp_img is not None:
-            items.append((cv2.cvtColor(persp_img, cv2.COLOR_BGR2RGB), "Perspective (Uploaded)"))
-    return items
-
-# ---------------- Gradio UI (keeps the look of previous interface) ----------------
-with gr.Blocks() as demo:
-    gr.Markdown("# Homography ROI Projection with Multiple Feature Detectors")
-    gr.Markdown("Upload flat & perspective images plus `mockup.json`. The gallery shows your uploads immediately, and detector results after you click **Run**. Download each result separately.")
-
-    with gr.Row():
-        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"])
-
-    # Single gallery (used both for previews and results)
-    output_gallery = gr.Gallery(label="Gallery", show_label=True, columns=2, height=440)
-
-    # Individual download buttons (same as before)
-    download_sift  = gr.File(label="Download SIFT Result")
-    download_orb   = gr.File(label="Download ORB Result")
-    download_brisk = gr.File(label="Download BRISK Result")
-    download_kaze  = gr.File(label="Download KAZE Result")
-    download_akaze = gr.File(label="Download AKAZE Result")
-
-    run_btn = gr.Button("Run Homography")
-
-    # Show previews in the SAME gallery as soon as files are uploaded/changed
-    flat_input.change(
-        preview_uploaded_gallery,
-        inputs=[flat_input, persp_input],
-        outputs=[output_gallery]
-    )
-    persp_input.change(
-        preview_uploaded_gallery,
-        inputs=[flat_input, persp_input],
-        outputs=[output_gallery]
-    )
-
-    # On run, compute and replace gallery with detector results + populate downloads
-    run_btn.click(
-        fn=homography_all_detectors,
-        inputs=[flat_input, persp_input, json_input],
-        outputs=[output_gallery, download_sift, download_orb, download_brisk, download_kaze, download_akaze]
-    )
-
-demo.launch()
+# ---------------- Gradio UI ----------------
+iface = gr.Interface(
+    fn=homography_all_detectors,
+    inputs=[
+        gr.File(label="Upload Flat Image", file_types=[".jpg",".png",".jpeg"]),
+        gr.File(label="Upload Perspective Image", file_types=[".jpg",".png",".jpeg"]),
+        gr.File(label="Upload mockup.json", file_types=[".json"])
+    ],
+    outputs=[
+        gr.Gallery(label="Results (per Detector)", show_label=True),
+        gr.File(label="Download SIFT Result"),
+        gr.File(label="Download ORB Result"),
+        gr.File(label="Download BRISK Result"),
+        gr.File(label="Download KAZE Result"),
+        gr.File(label="Download AKAZE Result")
+    ],
+    title="Homography ROI Projection with Multiple Feature Detectors",
+    description="Upload flat & perspective images with mockup.json. The system will project ROI using SIFT, ORB, BRISK, KAZE, and AKAZE. Each result can be viewed and downloaded."
+)
+
+iface.launch()