Update app.py

app.py

@@ -77,7 +77,10 @@ def add_title(img_bgr, title):
     return np.vstack([bar, img_bgr])
 
 def parse_xml_roi_points(xml_path):
-    """Parse XML, return list of polygons (Nx2 points)."""
+    """Parse your XML structure, return list of polygons (Nx2 points)."""
+    if xml_path is None:
+        return None
+        
     polys = []
     try:
         tree = ET.parse(xml_path)
@@ -106,13 +109,15 @@ def parse_xml_roi_points(xml_path):
 def homography_all_detectors(flat_file, persp_file, json_file, xml_file):
     flat_img = cv2.imread(flat_file)
     persp_img = cv2.imread(persp_file)
-    mockup = json.load(open(json_file.name))
+    
+    # FIX: Use the file paths directly (no .name needed)
+    mockup = json.load(open(json_file))
     roi = mockup["printAreas"][0]
     roi_x, roi_y = roi["position"]["x"], roi["position"]["y"]
     roi_w, roi_h = roi["width"], roi["height"]
     roi_rot_deg = roi["rotation"]
 
-    xml_polys = parse_xml_roi_points(xml_file.name) if xml_file else None
+    xml_polys = parse_xml_roi_points(xml_file) if xml_file else None
 
     flat_gray = preprocess_gray_clahe(flat_img)
     persp_gray = preprocess_gray_clahe(persp_img)
@@ -165,11 +170,14 @@ def homography_all_detectors(flat_file, persp_file, json_file, xml_file):
         results.append((cv2.cvtColor(composite, cv2.COLOR_BGR2RGB), f"{method} Grid"))
         download_files.append(file_name)
 
-    # pad list with "" instead of None (to avoid Gradio error)
     while len(download_files) < 5:
-        download_files.append("")
-
-    return [results] + download_files[:5]
+        download_files.append(None)
+    
+    # Return the results in the correct format
+    gallery_output = results
+    file_outputs = download_files[:5]
+    
+    return [gallery_output] + file_outputs
 
 # ---------------- Gradio UI ----------------
 iface = gr.Interface(
@@ -191,4 +199,4 @@ iface = gr.Interface(
     title="Homography ROI vs XML ROI",
     description="Each detector produces one 2×2 grid: Flat, Matches, Homography ROI, Ground Truth ROI."
 )
-iface.launch()
+iface.launch()