app.py

"""
Square Theory Generator (10 best variations)
===========================================
2025‑05‑28 v5 ● 코드블록 JSON 파싱 오류 해결 + 한글 폰트 OK
-------------------------------------------------
변경 요약
---------
* **문제**: 일부 LLM 응답이 ```json … ``` 코드펜스로 감싸져 `json.loads()` 실패 → 예외.
* **해결**: 코드펜스 제거 `clean_json_block()` 함수 추가.
* 기타: 폰트 로직 그대로 유지.

실행법
------
```bash
pip install --upgrade gradio matplotlib openai requests
export OPENAI_API_KEY="sk-..."
python square_theory_gradio.py
```
"""

import os
import json
import tempfile
import urllib.request
import gradio as gr
import matplotlib.pyplot as plt
from matplotlib import patches, font_manager, rcParams
from openai import OpenAI

# -------------------------------------------------
# 0. 한글 폰트 설정 (다운로드 fallback 포함)
# -------------------------------------------------

PREFERRED_FONTS = ["Malgun Gothic", "NanumGothic", "AppleGothic", "DejaVu Sans"]
NANUM_URL = (
    "https://github.com/google/fonts/raw/main/ofl/nanumgothic/"
    "NanumGothic-Regular.ttf"
)

def _set_korean_font():
    available = {f.name for f in font_manager.fontManager.ttflist}
    for cand in PREFERRED_FONTS:
        if cand in available:
            rcParams["font.family"] = cand
            break
    else:
        try:
            tmp_dir = tempfile.gettempdir()
            font_path = os.path.join(tmp_dir, "NanumGothic-Regular.ttf")
            if not os.path.exists(font_path):
                urllib.request.urlretrieve(NANUM_URL, font_path)
            font_manager.fontManager.addfont(font_path)
            rcParams["font.family"] = font_manager.FontProperties(fname=font_path).get_name()
        except Exception as e:
            print("[WARN] Font download failed, Korean text may break:", e)
    rcParams["axes.unicode_minus"] = False

_set_korean_font()

# -------------------------------------------------
# 1. OpenAI 클라이언트
# -------------------------------------------------
if not os.getenv("OPENAI_API_KEY"):
    raise EnvironmentError("OPENAI_API_KEY 환경 변수를 설정하세요.")

client = OpenAI()

# -------------------------------------------------
# 2. Square Diagram
# -------------------------------------------------

def draw_square(words):
    fig, ax = plt.subplots(figsize=(4, 4))
    ax.add_patch(patches.Rectangle((0, 0), 1, 1, fill=False, linewidth=2))
    ax.text(-0.05, 1.05, str(words.get("tl", "")), ha="right", va="bottom", fontsize=14, fontweight="bold")
    ax.text(1.05, 1.05, str(words.get("tr", "")), ha="left", va="bottom", fontsize=14, fontweight="bold")
    ax.text(1.05, -0.05, str(words.get("br", "")), ha="left", va="top", fontsize=14, fontweight="bold")
    ax.text(-0.05, -0.05, str(words.get("bl", "")), ha="right", va="top", fontsize=14, fontweight="bold")
    ax.set_xticks([])
    ax.set_yticks([])
    ax.set_xlim(-0.2, 1.2)
    ax.set_ylim(-0.2, 1.2)
    ax.set_aspect("equal")
    return fig

# -------------------------------------------------
# 3. LLM Prompt & Call
# -------------------------------------------------
SYSTEM_PROMPT = (
    "너는 한국어 카피·브랜드 네이밍 전문가이자 Square Theory 도우미다. "
    "사용자가 준 하나의 단어(tl)를 기반으로 품질이 가장 뛰어난 것부터 10개의 제안을 JSON 배열로 반환해라. "
    "각 배열 원소는 tl, tr, br, bl, top_phrase, bottom_phrase, slogan, brand 필드를 가지며, "
    "사각형 네 꼭짓점(tl>tr>br>bl)과 두 표현·슬로건·브랜드 네임이 자연스럽게 연결돼야 한다. "
    "배열은 최고의 제안이 index 0, 그다음이 index 1 … 9 순서여야 한다. "
    "결과는 JSON 외 다른 텍스트를 포함하면 안 된다."
)


def clean_json_block(text: str) -> str:
    """Remove ```json ... ``` or ``` ... ``` fences."""
    text = text.strip()
    if text.startswith("```"):
        # remove first ``` line
        text = text.split("\n", 1)[1] if "\n" in text else text[3:]
        # strip trailing fence
        if text.endswith("```"):
            text = text[:-3]
    return text.strip()


def call_llm(seed: str):
    resp = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "system", "content": SYSTEM_PROMPT},
            {"role": "user", "content": seed},
        ],
        temperature=0.9,
        max_tokens=1024,
    )
    raw = resp.choices[0].message.content
    cleaned = clean_json_block(raw)
    try:
        data = json.loads(cleaned)
        if not isinstance(data, list) or len(data) != 10:
            raise ValueError("JSON 배열 길이가 10이 아님")
    except Exception as exc:
        raise ValueError(f"LLM JSON 파싱 실패: {exc}\n원문: {cleaned[:300]} …")
    return data

# -------------------------------------------------
# 4. Gradio callback
# -------------------------------------------------

def generate(seed_word: str):
    results = call_llm(seed_word)
    fig = draw_square({k: results[0][k] for k in ("tl", "tr", "br", "bl")})
    md_lines = []
    for idx, item in enumerate(results, 1):
        md_lines.append(
            f"### {idx}. {item['top_phrase']} / {item['bottom_phrase']}\n"
            f"- **슬로건**: {item['slogan']}\n"
            f"- **브랜드 네임**: {item['brand']}\n"
            f"- (tl={item['tl']}, tr={item['tr']}, br={item['br']}, bl={item['bl']})\n"
        )
    return fig, "\n".join(md_lines)

# -------------------------------------------------
# 5. UI
# -------------------------------------------------
with gr.Blocks(title="Square Theory – Top 10 🇰🇷") as demo:
    gr.Markdown("""# 🔟 Square Theory 제안 Top 10\n단어 1개 → LLM이 평가·정렬한 10개 사각형/카피/브랜드 네임""")
    seed = gr.Textbox(label="시드 단어(TL)", placeholder="예: 골든")
    run = gr.Button("생성")
    fig_out = gr.Plot(label="1위 사각형")
    md_out = gr.Markdown(label="Top 10 제안")

    run.click(generate, inputs=seed, outputs=[fig_out, md_out])

if __name__ == "__main__":
    demo.launch()