Update app.py

app.py

@@ -58,7 +58,7 @@ def homography_all_detectors(flat_file, persp_file, json_file, xml_file):
     xml_points = parse_xml_points(xml_file.name)
 
     methods = ["SIFT","ORB","BRISK","KAZE","AKAZE"]
-    gallery_paths = []
+    gallery_files = []
     download_files = []
 
     for method in methods:
@@ -82,31 +82,32 @@ def homography_all_detectors(flat_file, persp_file, json_file, xml_file):
         xml_mapped = cv2.perspectiveTransform(xml_points.reshape(-1,1,2),H).reshape(-1,2)
         for px,py in xml_mapped: cv2.circle(xml_gt_img,(int(px),int(py)),5,(0,0,255),-1)
 
-        # Merge 2x2 grid (original size)
+        # Merge 2x2 grid (original resolution)
         top = np.hstack([flat_img, match_img])
         bottom = np.hstack([persp_roi, xml_gt_img])
         combined_grid = np.vstack([top, bottom])
 
-        # Save combined grid
+        # Save combined grid as file
         base_name = os.path.splitext(os.path.basename(persp_file))[0]
         file_name = f"{base_name}_{method.lower()}.png"
         cv2.imwrite(file_name, combined_grid)
-        gallery_paths.append(file_name)
+        gallery_files.append(file_name)
         download_files.append(file_name)
 
+    # Ensure 5 download slots
     while len(download_files)<5: download_files.append(None)
-    return gallery_paths, download_files[0], download_files[1], download_files[2], download_files[3], download_files[4]
+    return gallery_files, download_files[0], download_files[1], download_files[2], download_files[3], download_files[4]
 
 iface = gr.Interface(
     fn=homography_all_detectors,
     inputs=[
-        gr.Image(label="Upload Flat Image",type="filepath"),
-        gr.Image(label="Upload Perspective Image",type="filepath"),
-        gr.File(label="Upload mockup.json",file_types=[".json"]),
-        gr.File(label="Upload XML file",file_types=[".xml"])
+        gr.Image(label="Upload Flat Image", type="filepath"),
+        gr.Image(label="Upload Perspective Image", type="filepath"),
+        gr.File(label="Upload mockup.json", file_types=[".json"]),
+        gr.File(label="Upload XML file", file_types=[".xml"])
     ],
     outputs=[
-        gr.Gallery(label="Results per Detector",show_label=True),
+        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"),
@@ -114,7 +115,7 @@ iface = gr.Interface(
         gr.File(label="Download AKAZE Result")
     ],
     title="Homography ROI Projection with Feature Matching & XML GT",
-    description="Flat + Perspective images with mockup.json & XML. Shows 4 views per detector. Original resolution kept."
+    description="Shows 4 views per detector. Original resolution kept; click on Gallery images to open full."
 )
 
 iface.launch()