Update utils/polygon_utils.py

utils/polygon_utils.py

@@ -7,11 +7,20 @@ import numpy as np
 from PIL import Image, ImageDraw, ImageFont
 import textwrap
 
-def shrink_polygon(polygon, shrink_ratio=0.9):
+def shrink_or_expand_polygon(polygon, shrink_ratio=0.9):
+    """
+    Resize a polygon around its centroid.
+    ratio < 1 → shrink
+    ratio > 1 → expand
+    """
     cx = sum(x for x, _ in polygon) / len(polygon)
     cy = sum(y for _, y in polygon) / len(polygon)
-    shrunk = [((x - cx) * shrink_ratio + cx, (y - cy) * shrink_ratio + cy) for x, y in polygon]
-    return [(int(x), int(y)) for x, y in shrunk]
+
+    new_poly = [
+        ((x - cx) * ratio + cx, (y - cy) * ratio + cy)
+        for x, y in polygon
+    ]
+    return [(int(x), int(y)) for x, y in new_poly]
 
 import numpy as np
 import cv2
@@ -61,7 +70,7 @@ def render_translated_chunk(img: Image.Image, translations, font_path="NotoSansS
 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)
+    render_polygon = shrink_or_expand_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)