Update utils/bubble_utils.py

utils/bubble_utils.py

@@ -74,7 +74,7 @@ def visualize_all_debug(img, translations, bubble_polygons, step_name="debug", p
     out_path = f"{prefix}_{step_name}.png"
     debug.save(out_path)
     print(f"📌 Saved debug visualization → {out_path}")
-    return debug
+    return out_path
 
 
 # ===================== Main Bubble Translation Pipeline ===================
@@ -87,6 +87,7 @@ def bubble_pipeline_single(file_obj, num_chunks=1, polygon_strategy="hybrid", de
       - Bubble-aware polygon correction
       - Clean text rendering inside refined bubble interiors
     """
+    debug_files = []
 
     # -------------------------------------------------------
     # 1. Load image
@@ -101,10 +102,12 @@ def bubble_pipeline_single(file_obj, num_chunks=1, polygon_strategy="hybrid", de
     print(f"🔍 Found {len(bubble_polygons)} refined bubbles")
 
     if debug:
-        visualize_all_debug(
+        p = visualize_all_debug(
             full_img, [], bubble_polygons,
             step_name="bubbles_only", prefix="bubble_dbg"
         )
+        debug_files.append(p)
+
 
     # -------------------------------------------------------
     # 3. MASK-BASED OCR (replaces standard OCR)
@@ -132,10 +135,11 @@ def bubble_pipeline_single(file_obj, num_chunks=1, polygon_strategy="hybrid", de
         print("⚠️ No bubble polygons → skip polygon correction")
 
     if debug:
-        visualize_all_debug(
+        p = visualize_all_debug(
             full_img, translations, bubble_polygons,
             step_name="after_correction", prefix="bubble_dbg"
         )
+        debug_files.append(p)
 
     # -------------------------------------------------------
     # 5. Render translated text inside refined interior polygons
@@ -178,7 +182,7 @@ def bubble_pipeline_single(file_obj, num_chunks=1, polygon_strategy="hybrid", de
 
     table_data = [[t["original"], t["translated"]] for t in translations]
 
-    return filename, orig_html, trans_html, table_data, [translations]
+    return filename, orig_html, trans_html, table_data, [translations], debug_files
 
 def split_image_into_chunks(img, num_chunks):
     """