Update utils/bubble_utils.py

utils/bubble_utils.py

@@ -5,7 +5,7 @@ import numpy as np
 from PIL import Image, ImageDraw
 
 from utils.image_utils import load_and_split_image, encode_image_to_html
-from utils.ocr_utils import extract_and_translate_chunk
+from utils.ocr_utils import extract_and_translate_chunk, extract_and_translate_with_masks
 from utils.polygon_utils import (
     draw_translated_text_convex,
     shrink_or_expand_polygon,
@@ -17,59 +17,6 @@ from utils.bubble_detect import detect_speech_bubbles_robust
 from utils.u2net_detector import detect_bubbles_u2net
 from utils.bubble_detect_rtdetr import detect_and_refine_bubbles, polygon_to_mask
 
-def extract_and_translate_with_masks(full_img, interior_polygons):
-    """
-    OCR ONLY inside bubble interior polygons.
-    Returns refined translations with improved accuracy.
-    """
-    np_img = np.array(full_img)
-    H, W = np_img.shape[:2]
-
-    translations = []
-
-    for idx, poly in enumerate(interior_polygons):
-        if not poly:
-            continue
-
-        # --- Create mask for this bubble ---
-        mask = polygon_to_mask((W, H), poly)
-
-        # Apply mask to original image (white outside)
-        bubble_img = np.where(mask[..., None] == 255, np_img, 255).astype(np.uint8)
-
-        # OCR on masked image
-        results = ocr_model.ocr(bubble_img)
-
-        if not results or not isinstance(results[0], dict):
-            continue
-
-        res = results[0]
-        polys = res.get("rec_polys", [])
-        texts = res.get("rec_texts", [])
-
-        for poly_coords, text in zip(polys, texts):
-            merged_text = text.strip()
-            if not merged_text:
-                continue
-
-            # Convert polygon coords to global coords
-            poly_global = [(int(x), int(y)) for x, y in poly_coords]
-
-            # Translate text
-            try:
-                translated = translate_text_azure(merged_text)
-            except:
-                translated = merged_text
-
-            translations.append({
-                "original": merged_text,
-                "translated": translated,
-                "polygon": poly_global,
-                "matched_bubble_idx": idx
-            })
-
-    return translations
-
 def match_translations_to_bubbles(translations, bubble_polygons, min_overlap=0.10):
     """
     Add matched_bubble_idx to each OCR translation.