Update app.py

app.py

@@ -232,23 +232,42 @@ def match_left_to_right(left_boxes, right_boxes):
 
 def visualize(img, left, right, mapping):
     vis = img.copy()
+
+    # --- Draw left (test) boxes ---
     for i, tb in enumerate(left):
-        x,y,w,h = tb
-        cv2.rectangle(vis, (x,y), (x+w,y+h), (255,0,0), 2)
-        cv2.putText(vis, f"T{i+1}", (x,y-5), cv2.FONT_HERSHEY_SIMPLEX, 0.45, (255,0,0), 1)
+        x, y, w, h = tb
+        cv2.rectangle(vis, (x, y), (x + w, y + h), (255, 0, 0), 2)
+        cv2.putText(
+            vis, f"T{i+1}", (x, y - 5),
+            cv2.FONT_HERSHEY_SIMPLEX, 0.45, (255, 0, 0), 1
+        )
+
+    # --- Draw right (reference) boxes (skip synthetic ones) ---
     for j, rb in enumerate(right):
-        color = (255,0,255) if rb.get('synthetic', False) else (255,0,255)
-        cv2.rectangle(vis, (rb['x'], rb['y']), (rb['x']+rb['w'], rb['y']+rb['h']), color, 1)
+        if rb.get('synthetic', False):
+            continue  # skip synthetic boxes to avoid double-drawing
+        color = (255, 0, 255)  # magenta for right boxes
+        cv2.rectangle(
+            vis,
+            (rb['x'], rb['y']),
+            (rb['x'] + rb['w'], rb['y'] + rb['h']),
+            color,
+            1
+        )
+    # --- Draw red matching lines ---
     for i, tb in enumerate(left):
         key = f"test_box_{i+1}"
         tx, ty, tw, th = tb
-        tcx, tcy = int(tx + tw/2), int(ty + th/2)
-        for rb in mapping[key]["matched_refs"]:
-            rcx = int(rb["x"] + rb["w"]/2)
-            rcy = int(rb["y"] + rb["h"]/2)
-            cv2.line(vis, (tcx,tcy), (rcx,rcy), (0,0,255), 1)
+        tcx, tcy = int(tx + tw / 2), int(ty + th / 2)
+        for rb in mapping.get(key, {}).get("matched_refs", []):
+            rcx = int(rb["x"] + rb["w"] / 2)
+            rcy = int(rb["y"] + rb["h"] / 2)
+            cv2.line(vis, (tcx, tcy), (rcx, rcy), (0, 0, 255), 1)
+
+    # Convert BGR → RGB for Gradio display
     vis_rgb = cv2.cvtColor(vis, cv2.COLOR_BGR2RGB)
     return vis_rgb
+
 def keep_one_box_per_row(rects, reference_rects=None, row_tol=ROW_TOL):
     """
     Keep only one representative box per row.