Update app.py

app.py

@@ -1,16 +1,13 @@
 """
 Square Theory Generator (10 best variations)
 ===========================================
-2025‑05‑28 v4 ● 한글 깨짐 완전 해결 + Top 10 결과
+2025‑05‑28 v5 ● 코드블록 JSON 파싱 오류 해결 + 한글 폰트 OK
 -------------------------------------------------
 변경 요약
 ---------
-1. **한글 폰트 강제 등록 로직 강화**
-   * 시스템에 ‘Malgun/Nanum/Apple’ 폰트가 없으면 → Google Fonts에서 **NanumGothic TTF**를 다운로드해 런타임 등록.
-2. **draw_square 안전성 강화**
-   * 각 코너 문자열을 `str()` 캐스팅, 널 값 방지.
-3. **기타**
-   * 코드 주석·정리.
+* **문제**: 일부 LLM 응답이 ```json … ``` 코드펜스로 감싸져 `json.loads()` 실패 → 예외.
+* **해결**: 코드펜스 제거 `clean_json_block()` 함수 추가.
+* 기타: 폰트 로직 그대로 유지.
 
 실행법
 ------
@@ -25,7 +22,6 @@ import os
 import json
 import tempfile
 import urllib.request
-import requests
 import gradio as gr
 import matplotlib.pyplot as plt
 from matplotlib import patches, font_manager, rcParams
@@ -48,19 +44,15 @@ def _set_korean_font():
             rcParams["font.family"] = cand
             break
     else:
-        # 폰트가 하나도 없으면 NanumGothic 다운로드 후 등록
         try:
             tmp_dir = tempfile.gettempdir()
             font_path = os.path.join(tmp_dir, "NanumGothic-Regular.ttf")
             if not os.path.exists(font_path):
-                print("[INFO] Downloading NanumGothic font…")
                 urllib.request.urlretrieve(NANUM_URL, font_path)
             font_manager.fontManager.addfont(font_path)
             rcParams["font.family"] = font_manager.FontProperties(fname=font_path).get_name()
-            print(f"[INFO] Registered fallback font: {rcParams['font.family']}")
         except Exception as e:
             print("[WARN] Font download failed, Korean text may break:", e)
-    # 그래프 음수 부호 깨짐 방지
     rcParams["axes.unicode_minus"] = False
 
 _set_korean_font()
@@ -104,6 +96,18 @@ SYSTEM_PROMPT = (
 )
 
 
+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",
@@ -114,13 +118,14 @@ def call_llm(seed: str):
         temperature=0.9,
         max_tokens=1024,
     )
-    raw = resp.choices[0].message.content.strip()
+    raw = resp.choices[0].message.content
+    cleaned = clean_json_block(raw)
     try:
-        data = json.loads(raw)
+        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원문: {raw[:300]} …")
+        raise ValueError(f"LLM JSON 파싱 실패: {exc}\n원문: {cleaned[:300]} …")
     return data
 
 # -------------------------------------------------