Update app.py

app.py

@@ -2,6 +2,7 @@ import cv2
 import numpy as np
 import json
 import gradio as gr
+import os
 
 # ---------------- Helper functions ----------------
 def get_rotated_rect_corners(x, y, w, h, rotation_deg):
@@ -87,7 +88,8 @@ def homography_all_detectors(flat_file, persp_file, json_file):
     persp_gray = preprocess_gray_clahe(persp_img)
 
     methods = ["SIFT", "ORB", "BRISK", "KAZE", "AKAZE"]
-    outputs = []
+    gallery_images = []
+    download_files = []
 
     for method in methods:
         kp1, kp2, good_matches = detect_and_match(flat_gray, persp_gray, method=method)
@@ -113,27 +115,37 @@ def homography_all_detectors(flat_file, persp_file, json_file):
         # Convert BGR -> RGB for display
         result_rgb = cv2.cvtColor(persp_debug, cv2.COLOR_BGR2RGB)
 
-        outputs.append((result_rgb, f"{method} Result"))
+        # Save result for download
+        file_name = f"result_{method.lower()}.png"
+        cv2.imwrite(file_name, cv2.cvtColor(result_rgb, cv2.COLOR_RGB2BGR))
 
-    return outputs
+        gallery_images.append((result_rgb, f"{method} Result"))
+        download_files.append(file_name)
 
-# ---------------- Gradio UI ----------------
-with gr.Blocks() as demo:
-    gr.Markdown("## Homography ROI Demo 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"])
+    # return gallery + 5 download files (pad with None if less)
+    while len(download_files) < 5:
+        download_files.append(None)
 
-    output_gallery = gr.Gallery(
-        label="Perspective ROI Results", 
-        columns=2, 
-        height=400, 
-        show_label=True
-    )
+    return [gallery_images] + download_files[:5]
 
-    run_btn = gr.Button("Run Homography")
-    run_btn.click(homography_all_detectors, inputs=[flat_input, persp_input, json_input], outputs=output_gallery)
-
-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()