app.py

# app.py
import os
from PIL import Image, ImageDraw, ImageFont
import gradio as gr
from transformers import pipeline

# 1) 이미지 캡셔닝 파이프라인
captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")

# 2) 밈 + 자막 합성 함수
def make_meme_with_subtitle(image_path, top_text, bottom_text):
    # 원본 이미지 로드
    img = Image.open(image_path).convert("RGB")
    w, h = img.size

    # ① 자동 캡션 생성
    out = captioner(img, max_length=40, num_beams=5)
    raw = out[0].get("generated_text") or out[0].get("text") or ""
    subtitle = raw.strip().capitalize()

    # ② 자막용 검은 띠
    bar_h = int(h * 0.15)
    bar = Image.new("RGB", (w, bar_h), color="black")
    combined_h = h + bar_h
    combined = Image.new("RGB", (w, combined_h))
    combined.paste(img, (0, 0))
    combined.paste(bar, (0, h))

    draw = ImageDraw.Draw(combined)
    # 사용할 폰트 (시스템에 없으면 기본 폰트)
    try:
        font_path = "arialbd.ttf"
        base_font_size = int(bar_h * 0.5)
        font = ImageFont.truetype(font_path, size=base_font_size)
    except:
        font = ImageFont.load_default()

    # ③ 자막 그리기 (이미지 하단 중앙)
    tw, th = draw.textsize(subtitle, font=font)
    tx = (w - tw) // 2
    ty = h + (bar_h - th) // 2
    draw.text((tx, ty), subtitle, font=font, fill="white")

    # ④ 밈 텍스트 (상단·하단)
    meme_font_size = int(h * 0.07)
    try:
        meme_font = ImageFont.truetype(font_path, size=meme_font_size)
    except:
        meme_font = ImageFont.load_default()

    # 상단 텍스트
    if top_text:
        text = top_text.upper()
        tw, th = draw.textsize(text, font=meme_font)
        draw.text(((w - tw)//2, 10), text, font=meme_font, fill="white", stroke_width=2, stroke_fill="black")

    # 하단 텍스트 (이미지 영역 안쪽)
    if bottom_text:
        text = bottom_text.upper()
        tw, th = draw.textsize(text, font=meme_font)
        draw.text(((w - tw)//2, h - th - 10), text, font=meme_font, fill="white", stroke_width=2, stroke_fill="black")

    return combined

# 3) Gradio UI
with gr.Blocks() as demo:
    gr.Markdown("## 📸 밈 생성 + 자동 영어 자막 합성")
    gr.Markdown("이미지를 업로드하고, 상단·하단 밈 텍스트를 입력하면\n자동 생성된 영어 자막과 함께 밈 스타일 이미지를 반환합니다.")
    img_in = gr.Image(type="filepath", label="Upload Image")
    top_txt = gr.Textbox(label="Top Text (optional)", placeholder="e.g. WHEN YOU REALIZE...", lines=1)
    bottom_txt = gr.Textbox(label="Bottom Text (optional)", placeholder="e.g. ...IT'S MONDAY AGAIN", lines=1)
    btn = gr.Button("Generate Meme")
    out_img = gr.Image(label="Meme with Subtitle")
    btn.click(fn=make_meme_with_subtitle, inputs=[img_in, top_txt, bottom_txt], outputs=out_img)

# 4) 앱 실행
if __name__ == "__main__":
    demo.launch(
        server_name="0.0.0.0",
        server_port=int(os.environ.get("PORT", 7860))
    )