Update utils/bubble_utils.py

utils/bubble_utils.py

@@ -33,58 +33,104 @@ def match_translations_to_bubbles(translations, bubble_polygons, min_overlap=0.1
     return translations
 
 # ======================== Debug Visualization ============================
-def visualize_all_debug(img, translations, bubble_polygons, bubble_boxes=None, 
-                        step_name="debug", prefix="debug"):
+def visualize_all_debug(
+    img, translations, bubble_polygons, bubble_boxes=None,
+    step_name="debug", prefix="debug"
+):
     """
-    Draw:
-      - Bubble polygons (BLUE)
-      - Bounding boxes (ORANGE)
-      - OCR polygons (RED)
-      - Corrected polygons (GREEN)
-      - OCR centers (YELLOW)
+    Robust debug visualization.
+    Handles malformed polygons, empty lists, None values, and degenerate shapes.
     """
+
+    import numpy as np
+    from PIL import ImageDraw
+
     debug = img.copy()
     draw = ImageDraw.Draw(debug, "RGBA")
 
-    # -------------------------------
-    # Draw Bubble bounding boxes
-    # -------------------------------
+    # ==========================
+    # Helper: validate polygon
+    # ==========================
+    def valid_poly(poly):
+        if not poly or len(poly) < 3:
+            return False
+        # Filter invalid coords
+        cleaned = [(int(x), int(y)) for x, y in poly if isinstance(x, (int, float))]
+        # Must have >= 3 *distinct* points
+        return len(set(cleaned)) >= 3
+
+    # ==========================
+    # Draw bounding boxes (ORANGE)
+    # ==========================
     if bubble_boxes:
-        for (x1, y1, x2, y2) in bubble_boxes:
-            draw.rectangle((x1, y1, x2, y2), outline=(255, 165, 0, 180), width=3)
-
-    # -------------------------------
-    # Draw Bubble polygons
-    # -------------------------------
+        for box in bubble_boxes:
+            try:
+                x1, y1, x2, y2 = map(int, box)
+                draw.rectangle((x1, y1, x2, y2), outline=(255,165,0,180), width=3)
+            except Exception:
+                continue
+
+    # ==========================
+    # Draw bubble polygons (BLUE)
+    # ==========================
     for bp in bubble_polygons:
-        if not bp:
+        if not valid_poly(bp):
             continue
-        draw.polygon(bp, outline=(30, 144, 255, 200), width=4)
 
-        cx = int(np.mean([p[0] for p in bp]))
-        cy = int(np.mean([p[1] for p in bp]))
-        draw.ellipse((cx - 4, cy - 4, cx + 4, cy + 4), fill=(0, 255, 255, 220))
+        try:
+            draw.polygon(bp, outline=(30,144,255,200), width=4)
 
-    # -------------------------------
-    # OCR polygons
-    # -------------------------------
+            xs = [p[0] for p in bp]
+            ys = [p[1] for p in bp]
+            cx, cy = int(np.mean(xs)), int(np.mean(ys))
+            draw.ellipse((cx-4, cy-4, cx+4, cy+4), fill=(0,255,255,220))
+        except Exception:
+            continue
+
+    # ==========================
+    # Draw OCR polygons
+    # ==========================
     for t in translations:
         orig = t.get("original_polygon")
         corr = t.get("polygon")
 
-        if orig:
-            draw.polygon(orig, outline=(255, 50, 50, 180), width=3)
-        if corr:
-            draw.polygon(corr, outline=(50, 255, 50, 220), width=3)
+        # Draw original polygon (RED)
+        if valid_poly(orig):
+            try:
+                draw.polygon(orig, outline=(255,50,50,180), width=3)
+            except Exception:
+                pass
+
+        # Draw corrected polygon (GREEN)
+        if valid_poly(corr):
+            try:
+                draw.polygon(corr, outline=(50,255,50,220), width=3)
+            except Exception:
+                pass
+
+        # Center point (YELLOW)
+        poly_for_center = None
+        if valid_poly(corr):
+            poly_for_center = corr
+        elif valid_poly(orig):
+            poly_for_center = orig
 
-        poly_for_center = corr or orig
         if poly_for_center:
-            cx, cy = np.mean(poly_for_center, axis=0)
-            draw.ellipse((cx - 3, cy - 3, cx + 3, cy + 3), fill=(255, 255, 0, 220))
-
+            try:
+                xs = [p[0] for p in poly_for_center]
+                ys = [p[1] for p in poly_for_center]
+                cx, cy = int(np.mean(xs)), int(np.mean(ys))
+                draw.ellipse((cx-3, cy-3, cx+3, cy+3), fill=(255,255,0,220))
+            except Exception:
+                pass
+
+    # ==========================
+    # Save output
+    # ==========================
     out_path = f"{prefix}_{step_name}.png"
     debug.save(out_path)
     print(f"📌 Saved debug visualization → {out_path}")
+
     return out_path
 
 # ===================== Main Bubble Translation Pipeline ===================