Update app.py

app.py

@@ -31,11 +31,21 @@ def run_pipeline(flat_img, pers_img, mockup_json, xml_gt):
     json_pts = get_json_corners(mockup_json)
     gt_pts   = extract_points_from_xml(xml_gt)
     pred_pts = extract_points_from_xml(xml_pred_path)
-
-    match_json_gt   = draw_feature_matching(img_json.copy(), json_pts, img_xml.copy(), gt_pts, draw_boxes=True)
-    match_json_pred = draw_feature_matching(img_json.copy(), json_pts, img_xml.copy(), pred_pts, draw_boxes=True)
+    color = (0, 255, 0)  # Green for boxes
+    color2 = (0, 0, 255) # Blue for lines
+    match_json_gt   = draw_feature_matching(img_json.copy(), json_pts, img_xml.copy(), gt_pts, color,draw_boxes=True)
+    match_json_pred = draw_feature_matching(img_json.copy(), json_pts, img_xml.copy(), pred_pts, color2,draw_boxes=True)
 
     stacked = stack_images_side_by_side(match_json_gt, match_json_pred)
+    # Add vertical center line
+    h, w, _ = stacked.shape
+    center_x = w // 2
+    cv2.line(stacked, (center_x, 0), (center_x, h), (255, 0, 0), 4)  # blue line
+
+    # Add text labels
+    font = cv2.FONT_HERSHEY_SIMPLEX
+    cv2.putText(stacked, "Ground Truth", (50, 50), font, 2, (0, 255, 0), 3, cv2.LINE_AA)
+    cv2.putText(stacked, "Our Result", (center_x + 50, 50), font, 2, (0, 0, 255), 3, cv2.LINE_AA)
 
     # Save result
     cv2.imwrite(result_path, cv2.cvtColor(stacked, cv2.COLOR_RGB2BGR))