Update app.py

app.py

@@ -122,30 +122,38 @@ def homography_all_detectors(flat_file, persp_file, json_file):
         gallery_images.append((result_rgb, f"{method} Result"))
         download_files.append(file_name)
 
-    # return gallery + 5 download files (pad with None if less)
+    # pad download files list to 5
     while len(download_files) < 5:
         download_files.append(None)
 
-    return [gallery_images] + download_files[:5]
+    return flat_img[:, :, ::-1], persp_img[:, :, ::-1], [gallery_images] + download_files[:5]
 
 # ---------------- 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()
+with gr.Blocks() as demo:
+    gr.Markdown("## Homography ROI Projection with Multiple Feature Detectors")
+
+    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"])
+
+    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)
+
+    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")
+    run_btn.click(
+        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]
+    )
+
+demo.launch()