Update utils/polygon_utils.py

utils/polygon_utils.py

@@ -40,80 +40,54 @@ def inpaint_polygon(img: Image.Image, polygon, mode="auto", fallback_color=(255,
     draw.polygon(polygon, fill=fallback_color)
     return img_copy
 
-def draw_wrapped_text(img, polygon, text, font_path, polygon_for_size=None):
+def merge_polygons_to_convex_hull(polygons):
+    points = [pt for poly in polygons for pt in poly]
+    hull = MultiPoint(points).convex_hull
+    return list(hull.exterior.coords)
+
+def render_translated_chunk(img: Image.Image, translations, font_path="NotoSansSC-Regular.ttf", font_scale=1.0):
+    from PIL import ImageDraw
+    img_copy = img.copy()
+
+    for entry in translations:
+        polygon = entry.get("polygon") or entry.get("polygons")
+        text = entry.get("translated", "")
+
+        if polygon and text:
+            img_copy = draw_translated_text_convex(img_copy, polygon, text, font_path, font_scale)
+
+    return img_copy
+
+def draw_translated_text_convex(img, polygon_coords, text, font_path="NotoSansSC-Regular.ttf", font_scale=1.0):
+    from PIL import ImageDraw
+    font_polygon = polygon_coords
+    render_polygon = shrink_polygon(polygon_coords, shrink_ratio=0.9)
+    img = inpaint_polygon(img, render_polygon, mode="auto", fallback_color=(255, 255, 255))
+    debug_color = (180, 255, 180)
+    draw = ImageDraw.Draw(img)
+    draw.line(render_polygon + [render_polygon[0]], fill=debug_color, width=1)
+    draw_wrapped_text(img, render_polygon, text, font_path, polygon_for_size=font_polygon, font_scale=font_scale)
+    return img
+
+def draw_wrapped_text(img, polygon, text, font_path, polygon_for_size=None, font_scale=1.0):
     from PIL import ImageDraw, ImageFont
     import textwrap
-
     polygon_for_size = polygon_for_size or polygon
     draw = ImageDraw.Draw(img)
-
-    # Bounding box from polygon for size estimation
     xs, ys = zip(*polygon_for_size)
     x_min, x_max = min(xs), max(xs)
     y_min, y_max = min(ys), max(ys)
     box_width = x_max - x_min
     box_height = y_max - y_min
-
-    # Estimate font size from area and text length
-    avg_char_width = 0.4  # Assumed ratio
+    avg_char_width = 0.4
     estimated_size = int(min(box_height / 1.2, box_width / (len(text) * avg_char_width)))
-    font_size = max(10, estimated_size)
-
+    estimated_size = max(6, estimated_size)
+    font_size = int(estimated_size * font_scale)
     font = ImageFont.truetype(font_path, font_size)
-
-    # Wrap text
     max_chars = max(1, int(box_width / (font.getbbox("A")[2] + 1)))
     wrapped = textwrap.fill(text, width=max_chars)
-
-    # Compute actual text size for centering
     bbox = draw.textbbox((0, 0), wrapped, font=font)
     text_w, text_h = bbox[2] - bbox[0], bbox[3] - bbox[1]
-
     x = x_min + (box_width - text_w) / 2
     y = y_min + (box_height - text_h) / 2
-
-    draw.text((x, y), wrapped, font=font, fill="black", align="center")
-
-def draw_translated_text_convex(img, polygon_coords, text, font_path="NotoSansSC-Regular.ttf"):
-    """
-    Inpaints original text area, draws a light polygon boundary for debug,
-    and renders wrapped translated text using a font size fit to the original polygon.
-    """
-    # Use original polygon to compute font size
-    font_polygon = polygon_coords
-
-    # Shrink polygon for rendering to avoid text overflow
-    render_polygon = shrink_polygon(polygon_coords, shrink_ratio=0.9)
-
-    # Inpaint using shrunk region
-    img = inpaint_polygon(img, render_polygon, mode="auto", fallback_color=(255, 255, 255))
-
-    # Optional: Light debug polygon (pastel green or light gray)
-    debug_color = (180, 255, 180)  # light green
-    draw = ImageDraw.Draw(img)
-    draw.line(render_polygon + [render_polygon[0]], fill=debug_color, width=1)
-
-    # Render text inside the shrunk polygon but compute font size from original size
-    draw_wrapped_text(img, render_polygon, text, font_path, polygon_for_size=font_polygon)
-
-    return img
-
-def merge_polygons_to_convex_hull(polygons):
-    points = [pt for poly in polygons for pt in poly]
-    hull = MultiPoint(points).convex_hull
-    return list(hull.exterior.coords)
-
-def render_translated_chunk(img: Image.Image, translations, font_path="NotoSansSC-Regular.ttf"):
-    """
-    Draws all translated text on a copy of the image using inpainting and polygon wrapping.
-    """
-    img_copy = img.copy()
-
-    for entry in translations:
-        polygon = entry.get("polygon") or entry.get("polygons")  # accept either key
-        text = entry.get("translated", "")
-
-        if polygon and text:
-            img_copy = draw_translated_text_convex(img_copy, polygon, text, font_path)
-
-    return img_copy
+    draw.text((x, y), wrapped, font=font, fill="black", align="center")