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 ----------------
+# ---------------- Main Homography Function (unchanged logic) ----------------
 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,38 +122,70 @@ def homography_all_detectors(flat_file, persp_file, json_file):
         gallery_images.append((result_rgb, f"{method} Result"))
         download_files.append(file_name)
 
-    # pad download files list to 5
+    # pad to 5 download slots
     while len(download_files) < 5:
         download_files.append(None)
 
-    return flat_img[:, :, ::-1], persp_img[:, :, ::-1], [gallery_images] + download_files[:5]
-
-# ---------------- Gradio UI ----------------
+    # 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("# 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"])
+        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"])
 
-    with gr.Row():
-        flat_preview = gr.Image(label="Flat Image Preview")
-        persp_preview = gr.Image(label="Perspective Image Preview")
-
-    output_gallery = gr.Gallery(label="Detector Results", show_label=True, columns=2, height=400)
+    # Single gallery (used both for previews and results)
+    output_gallery = gr.Gallery(label="Gallery", show_label=True, columns=2, height=440)
 
-    download_sift = gr.File(label="Download SIFT Result")
-    download_orb = gr.File(label="Download ORB Result")
+    # 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_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, 
+        fn=homography_all_detectors,
         inputs=[flat_input, persp_input, json_input],
-        outputs=[flat_preview, persp_preview, output_gallery, download_sift, download_orb, download_brisk, download_kaze, download_akaze]
+        outputs=[output_gallery, download_sift, download_orb, download_brisk, download_kaze, download_akaze]
     )
 
 demo.launch()