Update utils/bubble_utils.py

utils/bubble_utils.py

@@ -33,29 +33,41 @@ def match_translations_to_bubbles(translations, bubble_polygons, min_overlap=0.1
     return translations
 
 # ======================== Debug Visualization ============================
-
-def visualize_all_debug(img, translations, bubble_polygons, 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)
-      - Original OCR polygons (RED)
+      - Bounding boxes (ORANGE)
+      - OCR polygons (RED)
       - Corrected polygons (GREEN)
       - OCR centers (YELLOW)
-    Saves a PNG for inspection.
     """
     debug = img.copy()
     draw = ImageDraw.Draw(debug, "RGBA")
 
-    # Bubbles (blue)
+    # -------------------------------
+    # Draw Bubble bounding boxes
+    # -------------------------------
+    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 bp in bubble_polygons:
         if not 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))
 
+    # -------------------------------
     # OCR polygons
+    # -------------------------------
     for t in translations:
         orig = t.get("original_polygon")
         corr = t.get("polygon")
@@ -67,8 +79,7 @@ def visualize_all_debug(img, translations, bubble_polygons, step_name="debug", p
 
         poly_for_center = corr or orig
         if poly_for_center:
-            center = np.mean(poly_for_center, axis=0)
-            cx, cy = int(center[0]), int(center[1])
+            cx, cy = np.mean(poly_for_center, axis=0)
             draw.ellipse((cx - 3, cy - 3, cx + 3, cy + 3), fill=(255, 255, 0, 220))
 
     out_path = f"{prefix}_{step_name}.png"
@@ -76,7 +87,6 @@ def visualize_all_debug(img, translations, bubble_polygons, step_name="debug", p
     print(f"📌 Saved debug visualization → {out_path}")
     return out_path
 
-
 # ===================== Main Bubble Translation Pipeline ===================
 def bubble_pipeline_single(file_obj, num_chunks=1, polygon_strategy="hybrid", debug=True):
     """
@@ -98,12 +108,12 @@ def bubble_pipeline_single(file_obj, num_chunks=1, polygon_strategy="hybrid", de
     # -------------------------------------------------------
     # 2. Detect rectangles + refine shapes
     # -------------------------------------------------------
-    bubble_polygons, interior_polygons = detect_and_refine_bubbles(full_img)
+    bubble_polygons, interior_polygons, bubble_boxes = detect_and_refine_bubbles(full_img)
     print(f"🔍 Found {len(bubble_polygons)} refined bubbles")
 
     if debug:
         p = visualize_all_debug(
-            full_img, [], bubble_polygons,
+            full_img, [], bubble_polygons, bubble_boxes=bubble_boxes,
             step_name="bubbles_only", prefix="bubble_dbg"
         )
         debug_files.append(p)
@@ -136,7 +146,7 @@ def bubble_pipeline_single(file_obj, num_chunks=1, polygon_strategy="hybrid", de
 
     if debug:
         p = visualize_all_debug(
-            full_img, translations, bubble_polygons,
+            full_img, translations, bubble_polygons, bubble_boxes=bubble_boxes,
             step_name="after_correction", prefix="bubble_dbg"
         )
         debug_files.append(p)