Update add_text.py

add_text.py

@@ -1,54 +1,75 @@
-from PIL import Image, ImageDraw, ImageFont
-import numpy as np
 import cv2
 import textwrap
+from PIL import Image, ImageDraw, ImageFont
+import numpy as np
 
 
-def add_text(image, text, font_path, bubble_contour):
-    pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
-    draw = ImageDraw.Draw(pil_image)
+def add_text(
+    image,
+    text,
+    bbox,
+    font_path,
+    initial_font_size=42,
+    padding=10,
+    line_spacing=1.25
+):
+    """
+    Insere texto dentro de uma bolha respeitando largura, altura e estética tipográfica.
+    Prioriza quebra de linhas antes de reduzir drasticamente o tamanho da fonte.
+    """
 
-    x, y, w, h = cv2.boundingRect(bubble_contour)
+    x, y, w, h = bbox
+    img_pil = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    draw = ImageDraw.Draw(img_pil)
 
-    # Tamanho inicial baseado na bolha
-    font_size = max(12, int(h * 0.08))
-    wrapping_ratio = 0.09
+    font_size = initial_font_size
+    wrapping_ratio = 0.9  # começa permitindo múltiplas linhas
 
-    while True:
+    while font_size > 10:
         font = ImageFont.truetype(font_path, font_size)
-        line_height = int(font_size * 1.2)
 
-        wrapped_text = textwrap.fill(
-            text,
-            width=max(1, int(w * wrapping_ratio)),
-            break_long_words=True
-        )
+        max_chars_per_line = max(1, int((w - 2 * padding) / (font_size * wrapping_ratio)))
+        lines = textwrap.wrap(text, width=max_chars_per_line)
 
-        lines = wrapped_text.split("\n")
-        total_height = len(lines) * line_height
+        # --- REGRA CRÍTICA: NÃO PERMITIR LINHA ÚNICA SE HOUVER ALTURA ---
+        min_lines = 2 if h > font_size * 2.5 else 1
+        if len(lines) < min_lines:
+            wrapping_ratio -= 0.03
+            continue
 
-        # 🔍 Verificação REAL de largura em pixels
-        max_line_width = max(
-            draw.textlength(line, font=font) for line in lines
-        )
+        # Medições reais
+        line_heights = []
+        max_line_width = 0
 
-        if total_height <= h and max_line_width <= w:
-            break
+        for line in lines:
+            bbox_text = draw.textbbox((0, 0), line, font=font)
+            lw = bbox_text[2] - bbox_text[0]
+            lh = bbox_text[3] - bbox_text[1]
+            max_line_width = max(max_line_width, lw)
+            line_heights.append(lh)
 
-        # Ajustes progressivos
-        font_size -= 1
-        wrapping_ratio += 0.015
+        total_height = int(sum(line_heights) * line_spacing)
 
-        if font_size <= 10:
+        # --- VERIFICA ENCAIXE ---
+        if total_height <= (h - 2 * padding) and max_line_width <= (w - 2 * padding):
             break
 
-    # Centralização vertical
-    text_y = y + (h - total_height) // 2
+        # --- PRIORIDADE CORRETA ---
+        if max_line_width > (w - 2 * padding):
+            wrapping_ratio -= 0.02  # quebra mais linhas
+        else:
+            font_size -= 1          # só reduz fonte se necessário
+
+    # --- DESENHO FINAL ---
+    current_y = y + padding + (h - total_height) // 2
 
     for line in lines:
-        line_width = draw.textlength(line, font=font)
-        text_x = x + (w - line_width) // 2
-        draw.text((text_x, text_y), line, fill=(0, 0, 0), font=font)
-        text_y += line_height
+        bbox_text = draw.textbbox((0, 0), line, font=font)
+        lw = bbox_text[2] - bbox_text[0]
+
+        text_x = x + (w - lw) // 2
+        draw.text((text_x, current_y), line, fill=(0, 0, 0), font=font)
+
+        current_y += int((bbox_text[3] - bbox_text[1]) * line_spacing)
 
-    return cv2.cvtColor(np.array(pil_image), cv2.COLOR_RGB2BGR)
+    return cv2.cvtColor(np.array(img_pil), cv2.COLOR_RGB2BGR)