Update app.py

app.py

@@ -1,42 +1,67 @@
 """
-Square Theory Generator (Gradio demo)
-------------------------------------
-이 스크립트는 Adam Aaronson의 “Square Theory” 개념을 한국어 카피라이팅·브랜드 네이밍에
-실험적으로 적용하기 위한 간단한 웹 앱입니다.
-
-🧩 **무엇을 하나요?**
-1. 사용자가 네 개의 단어를 입력하면 (윗변 A·B, 아랫변 A'·B')
-2. 네 변이 완전한 사각형을 이루는지 시각적으로 보여 줍니다.
-3. 동시에 두 줄 슬로건, 브랜드 네임 후보를 자동 제안합니다.
-
-✍️ **구현 포인트**
-- **Gradio Blocks**를 사용해 입력·출력 레이아웃을 자유롭게 구성합니다. (cf. gradio docs)  
-- **Matplotlib**로 간단한 사각형 도식을 그려서 시각적 이해를 돕습니다.
-- 단어 조합 로직은 데모 목적의 심플 버전입니다. 필요하면 WordNet·꼬꼬마 등으로 확장 가능.
-
-Author: ChatGPT (OpenAI)
+Square Theory Generator (Korean‑friendly, LLM‑powered)
+-----------------------------------------------------
+변경 사항
+0) Matplotlib 한글 폰트 자동 설정으로 깨짐 해결
+1) 입력 프롬프트를 하나만 두고, 나머지 3단어는 LLM이 생성
+2) OpenAI 공식 Python SDK (`openai`) 사용 – 사용자가 제공한 `client.responses.create` 예시 반영
+3) LLM이 사각형 단어 + 두 줄 카피·브랜드 네임까지 JSON으로 반환 → 자동 출력
+
+⚠️ **OPENAI_API_KEY** 환경 변수를 미리 설정해 두세요.
 """
 
+import os
+import json
 import gradio as gr
 import matplotlib.pyplot as plt
-from matplotlib import patches
-
-
-def draw_square(a: str, b: str, a_alt: str, b_alt: str):
-    """Return a matplotlib Figure that visualizes the Square Theory diagram."""
-    # Create figure & axis
+from matplotlib import patches, font_manager, rcParams
+from openai import OpenAI
+
+# -------------------------------------------------
+# 0. 한글 폰트 자동 탐색 & 설정 (Malgun, Nanum 등)
+# -------------------------------------------------
+
+def _set_korean_font():
+    candidates = [
+        "Malgun Gothic",  # Windows
+        "NanumGothic",    # Linux (apt install fonts-nanum)
+        "AppleGothic",    # macOS
+        "DejaVu Sans",    # fallback (기본 폰트지만 한글 일부 지원)
+    ]
+    available = {f.name for f in font_manager.fontManager.ttflist}
+    for cand in candidates:
+        if cand in available:
+            rcParams["font.family"] = cand
+            break
+    rcParams["axes.unicode_minus"] = False  # 음수 부호 깨짐 방지
+
+_set_korean_font()
+
+# -------------------------------------------------
+# 1. OpenAI 클라이언트 초기화
+# -------------------------------------------------
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+if not OPENAI_API_KEY:
+    raise EnvironmentError("OPENAI_API_KEY 환경 변수를 설정하세요.")
+
+client = OpenAI()
+
+# -------------------------------------------------
+# 2. Square Theory 도식 그리기
+# -------------------------------------------------
+
+def draw_square(words):
+    """words = dict with keys tl, tr, br, bl"""
     fig, ax = plt.subplots(figsize=(4, 4))
-    # Draw square border
+    # 사각형 프레임
     square = patches.Rectangle((0, 0), 1, 1, fill=False, linewidth=2)
     ax.add_patch(square)
 
-    # Corner labels
-    ax.text(-0.05, 1.05, a, ha="right", va="bottom", fontsize=14, fontweight="bold")
-    ax.text(1.05, 1.05, b, ha="left", va="bottom", fontsize=14, fontweight="bold")
-    ax.text(-0.05, -0.05, a_alt, ha="right", va="top", fontsize=14, fontweight="bold")
-    ax.text(1.05, -0.05, b_alt, ha="left", va="top", fontsize=14, fontweight="bold")
+    ax.text(-0.05, 1.05, words["tl"], ha="right", va="bottom", fontsize=14, fontweight="bold")
+    ax.text(1.05, 1.05, words["tr"], ha="left", va="bottom", fontsize=14, fontweight="bold")
+    ax.text(1.05, -0.05, words["br"], ha="left", va="top", fontsize=14, fontweight="bold")
+    ax.text(-0.05, -0.05, words["bl"], ha="right", va="top", fontsize=14, fontweight="bold")
 
-    # Hide axes
     ax.set_xticks([])
     ax.set_yticks([])
     ax.set_xlim(-0.2, 1.2)
@@ -44,51 +69,71 @@ def draw_square(a: str, b: str, a_alt: str, b_alt: str):
     ax.set_aspect("equal")
     return fig
 
+# -------------------------------------------------
+# 3. LLM 프롬프트 & 응답 파싱
+# -------------------------------------------------
 
-def generate_square(a: str, b: str, a_alt: str, b_alt: str):
-    """Core callback: returns diagram + text outputs."""
-    # Basic phrases
-    top_phrase = f"{a} {b}"
-    bottom_phrase = f"{a_alt} {b_alt}"
-
-    # Simple 2‑line slogan
-    slogan = f"{top_phrase}!\n{bottom_phrase}!"
-
-    # Brand suggestions (kor+eng)
-    brand_korean = f"{a_alt}{b_alt}"
-    brand_english = f"{a.capitalize()}{b.capitalize()}"
-    brand_combo = f"{brand_korean} / {brand_english}"
-
-    # Figure
-    fig = draw_square(a, b, a_alt, b_alt)
+SYSTEM_PROMPT = """너는 한국어 카피·브랜드 네이밍 전문가이자 Square Theory(사각형 이론) 도��미다.\n사용자가 제시한 하나의 단어(TL)를 기반으로, 다음 JSON 형식으로 사각형을 완성하라.\n- "tl": 입력 단어 그대로\n- "tr": TL과 의미·음이 대응되는 다른 단어\n- "br": TR과 의미·음이 대응되는 다른 단어\n- "bl": BR과 의미·음이 대응되는 다른 단어로 TL과도 연결이 있어야 한다\n또한, "top_phrase"(tl+tr), "bottom_phrase"(bl+br) 두 표현을 자연스러운 한국어 관용구로 제시하고,\n"slogan"(두 줄 카피)와 "brand"(브랜드 네임 후보) 필드를 함께 포함하라.\n결과는 반드시 **한글 UTF‑8**로 인코딩된 JSON만 반환한다."""
 
-    return fig, top_phrase, bottom_phrase, slogan, brand_combo
 
+def call_llm(seed):
+    """Returns parsed dict from LLM JSON output."""
+    response = client.responses.create(
+        model="gpt-4.1-mini",
+        input=[
+            {"role": "system", "content": [{"type": "input_text", "text": SYSTEM_PROMPT}]},
+            {"role": "user", "content": [{"type": "input_text", "text": seed}]},
+        ],
+        text={"format": {"type": "text"}},
+        temperature=0.9,
+        max_output_tokens=512,
+    )
+    raw = response.choices[0].message.content[0]["text"]
+    try:
+        data = json.loads(raw)
+    except json.JSONDecodeError as e:
+        raise ValueError(f"LLM 응답이 JSON 형식이 아닙니다: {e}\n원본: {raw}")
+    return data
+
+# -------------------------------------------------
+# 4. Gradio 콜백
+# -------------------------------------------------
+
+def generate(seed_word):
+    data = call_llm(seed_word)
+
+    words = {
+        "tl": data["tl"],
+        "tr": data["tr"],
+        "br": data["br"],
+        "bl": data["bl"],
+    }
+    fig = draw_square(words)
+    return (
+        fig,
+        data.get("top_phrase", ""),
+        data.get("bottom_phrase", ""),
+        data.get("slogan", ""),
+        data.get("brand", ""),
+    )
 
-with gr.Blocks(title="Square Theory Generator 🇰🇷") as demo:
-    gr.Markdown("""# ✨ Square Theory Generator\n네 개의 단어로 사각형을 완성해 카피·브랜드 아이디어를 만들어 보세요.""")
-
-    with gr.Row():
-        a = gr.Textbox(label="단어 1 (왼쪽 위)")
-        b = gr.Textbox(label="단어 2 (오른쪽 위)")
-    with gr.Row():
-        a_alt = gr.Textbox(label="대응 단어 1 (왼쪽 아래)")
-        b_alt = gr.Textbox(label="대응 단어 2 (오른쪽 아래)")
+# -------------------------------------------------
+# 5. Gradio UI
+# -------------------------------------------------
+with gr.Blocks(title="Square Theory Generator – LLM Powered 🇰🇷") as demo:
+    gr.Markdown("""# 🟧 Square Theory Generator\n단어 하나만 입력하면, LLM이 사각형 구조·카피·브랜드 네임을 자동으로 제안합니다.""")
 
-    btn = gr.Button("사각형 생성")
+    seed = gr.Textbox(label="시드 단어(TL)", placeholder="예: 골든")
+    run_btn = gr.Button("사각형 생성")
 
     fig_out = gr.Plot(label="사각형 다이어그램")
     top_out = gr.Textbox(label="윗변 표현")
     bottom_out = gr.Textbox(label="아랫변 표현")
-    slogan_out = gr.Textbox(label="2행 슬로건 제안")
+    slogan_out = gr.Textbox(label="두 줄 슬로건")
     brand_out = gr.Textbox(label="브랜드 네임 제안")
 
-    btn.click(
-        fn=generate_square,
-        inputs=[a, b, a_alt, b_alt],
-        outputs=[fig_out, top_out, bottom_out, slogan_out, brand_out],
-    )
+    run_btn.click(fn=generate, inputs=seed, outputs=[fig_out, top_out, bottom_out, slogan_out, brand_out])
 
 if __name__ == "__main__":
-    # For local testing; in production, set share=True or host on Spaces.
+    # launch(): share=True 하면 외부에서 접속 가능
     demo.launch()