Initial commit

2026-04-13 18:03:33 -07:00
commit 73d16c9b69
26 changed files with 1660 additions and 0 deletions
@@ -0,0 +1,115 @@
+import textwrap
+from PIL import Image, ImageDraw, ImageFont
+import os
+
+MAX_BUBBLES = 3
+MAX_LINE_WIDTH = 22
+BUBBLE_PADDING = 14
+FONT_SIZE = 22
+TAIL_SIZE = 14
+
+def get_font(size=FONT_SIZE):
+    candidates = [
+        "/System/Library/Fonts/Helvetica.ttc",
+        "/System/Library/Fonts/Arial.ttf",
+        "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
+        "/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf",
+    ]
+    for path in candidates:
+        if os.path.exists(path):
+            try:
+                return ImageFont.truetype(path, size)
+            except Exception:
+                pass
+    return ImageFont.load_default()
+
+def wrap_text(text, max_width=MAX_LINE_WIDTH):
+    return textwrap.fill(text, max_width)
+
+def draw_bubble(draw, x, y, w, h, tail_side="bottom"):
+    r = 16
+    draw.rounded_rectangle([x, y, x+w, y+h], radius=r, fill="white", outline="black", width=3)
+
+    if tail_side == "bottom":
+        tx = x + w // 2
+        ty = y + h
+        draw.polygon([
+            (tx - TAIL_SIZE, ty - 4),
+            (tx + TAIL_SIZE, ty - 4),
+            (tx, ty + TAIL_SIZE),
+        ], fill="white")
+        draw.line([(tx - TAIL_SIZE, ty - 4), (tx, ty + TAIL_SIZE)], fill="black", width=3)
+        draw.line([(tx + TAIL_SIZE, ty - 4), (tx, ty + TAIL_SIZE)], fill="black", width=3)
+    elif tail_side == "top":
+        tx = x + w // 2
+        ty = y
+        draw.polygon([
+            (tx - TAIL_SIZE, ty + 4),
+            (tx + TAIL_SIZE, ty + 4),
+            (tx, ty - TAIL_SIZE),
+        ], fill="white")
+        draw.line([(tx - TAIL_SIZE, ty + 4), (tx, ty - TAIL_SIZE)], fill="black", width=3)
+        draw.line([(tx + TAIL_SIZE, ty + 4), (tx, ty - TAIL_SIZE)], fill="black", width=3)
+
+def add_speech_bubbles(img_path, dialogue):
+    if not dialogue:
+        return
+
+    entries = [d for d in dialogue if d.get("line", "").strip()][:MAX_BUBBLES]
+    if not entries:
+        return
+
+    img = Image.open(img_path).convert("RGB")
+    draw = ImageDraw.Draw(img)
+    font = get_font(FONT_SIZE)
+    small_font = get_font(FONT_SIZE - 6)
+
+    iw, ih = img.size
+    margin = 18
+
+    zones = [
+        ih // 8,
+        ih * 5 // 8,
+        ih // 4,
+    ]
+
+    for i, entry in enumerate(entries):
+        speaker = entry.get("speaker", "").strip()
+        line = entry.get("line", "").strip()
+        if not line:
+            continue
+
+        wrapped = wrap_text(line)
+
+        bbox = draw.textbbox((0, 0), wrapped, font=font)
+        text_w = bbox[2] - bbox[0]
+        text_h = bbox[3] - bbox[1]
+
+        if speaker and speaker != "?":
+            spk_bbox = draw.textbbox((0, 0), speaker, font=small_font)
+            spk_h = spk_bbox[3] - spk_bbox[1] + 4
+        else:
+            spk_h = 0
+
+        bw = min(text_w + BUBBLE_PADDING * 2, iw - margin * 2)
+        bh = text_h + spk_h + BUBBLE_PADDING * 2
+
+        x_offset = margin if i % 2 == 0 else iw - bw - margin
+        by = zones[i % len(zones)]
+
+        by = max(margin, min(by, ih - bh - margin))
+
+        tail = "bottom" if by < ih // 2 else "top"
+        draw_bubble(draw, x_offset, by, bw, bh, tail_side=tail)
+
+        if speaker and speaker != "?":
+            draw.text((x_offset + BUBBLE_PADDING, by + BUBBLE_PADDING), speaker, font=small_font, fill="#444444")
+
+        draw.text(
+            (x_offset + BUBBLE_PADDING, by + BUBBLE_PADDING + spk_h),
+            wrapped,
+            font=font,
+            fill="black",
+        )
+
+    img.save(img_path)