Update main.py

main.py

@@ -2,7 +2,7 @@ from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import List, Dict, Any
-# utils 폴더에 해당 파일들이 실제로 있어야 합니다!
+# utils 폴더 확인 필수
 from utils import tarot, calculations, prompts, llm 
 import os
 import traceback
@@ -17,7 +17,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# --- 1. 요청 모델 (형님 코드 그대로 유지) ---
+# --- 1. 요청 모델 ---
 class TarotRequest(BaseModel):
     cards: List[str]
     topic: str
@@ -27,8 +27,8 @@ class TarotRequest(BaseModel):
 class FengShuiRequest(BaseModel):
     year: int
     gender: str
-    door_dir: str  # 앱에서 door_dir로 보내고 있음
-    head_dir: str  # 앱에서 head_dir로 보내고 있음
+    door_dir: str  
+    head_dir: str  
     query: str
     lang: str = "한국어"
 
@@ -38,14 +38,18 @@ class SajuRequest(BaseModel):
     day: int
     hour: int
     minute: int
-    calendar_type: str # 앱에서 calendar_type으로 보내고 있음
+    calendar_type: str 
     query: str
     lang: str = "한국어"
 
-# --- 2. 페르소나 생성기 (🔥 여기가 수정된 부분!) ---
+# --- 2. 페르소나 생성기 (🔥 다국어 완벽 지원 수정!) ---
 def get_dynamic_persona(role_type: str, lang: str) -> Dict[str, str]:
-    # 한국어 감지 (Korean, ko, 한국어 등이 포함되면 한국어 모드)
-    is_korean = any(k in lang for k in ["한국", "Korean", "ko"])
+    # 언어값이 비어있으면 기본값 영어
+    target_lang = lang if lang else "English"
+    
+    # 한국어 특화 처리 (한국어는 뉘앙스가 중요해서 따로 뺌)
+    is_korean = any(k in target_lang for k in ["한국", "Korean", "ko"])
+    
     prompt_text = ""
     
     if role_type == "tarot":
@@ -56,11 +60,11 @@ def get_dynamic_persona(role_type: str, lang: str) -> Dict[str, str]:
             말투: 부드럽고 신비로운 존댓말 (해요체).
             """
         else:
-            # 영어 모드 강력 고정
-            prompt_text = """
+            # 🔥 [수정] 영어가 아니라 {target_lang}으로 답변하라고 지시
+            prompt_text = f"""
             You are 'Emily', a mysterious Tarot Reader.
-            CRITICAL INSTRUCTION: Even if the user input is in Korean, you must ANSWER STRICTLY IN ENGLISH.
-            Do NOT say "I will translate". Just provide the reading directly.
+            CRITICAL INSTRUCTION: You must ANSWER STRICTLY IN {target_lang}.
+            Even if the user input is in another language, TRANSLATE your reading to {target_lang}.
             Tone: Mystical, soft, empathetic.
             """
             
@@ -71,11 +75,11 @@ def get_dynamic_persona(role_type: str, lang: str) -> Dict[str, str]:
             논리적이고 전문적으로 한국어로 답변하세요.
             """
         else:
-            # 영어 모드 강력 고정
-            prompt_text = """
+            # 🔥 [수정] {target_lang} 사용
+            prompt_text = f"""
             You are a Feng Shui Master.
-            CRITICAL INSTRUCTION: Even if the user input is in Korean, you must ANSWER STRICTLY IN ENGLISH.
-            Do NOT mention translation. Speak with authority as a Master.
+            CRITICAL INSTRUCTION: You must ANSWER STRICTLY IN {target_lang}.
+            Do NOT mention translation. Speak with authority as a Master in {target_lang}.
             Tone: Professional, logical, wise.
             """
 
@@ -86,19 +90,19 @@ def get_dynamic_persona(role_type: str, lang: str) -> Dict[str, str]:
             거친 반말(~해라, ~구나)을 쓰며 한국어로 답변하세요.
             """
         else:
-            # 영어 모드 강력 고정
-            prompt_text = """
+            # 🔥 [수정] {target_lang} 사용
+            prompt_text = f"""
             You are 'Shaman Cheon-Myeong'.
-            CRITICAL INSTRUCTION: Even if the user asks in Korean, you must ANSWER STRICTLY IN ENGLISH.
-            Do NOT say "I will translate". Speak directly to the user's soul.
+            CRITICAL INSTRUCTION: You must ANSWER STRICTLY IN {target_lang}.
+            Speak directly to the user's soul in {target_lang}.
             Tone: Mystical, Charismatic, slightly archaic style.
             """
     else:
-        prompt_text = f"Answer in {lang}."
+        prompt_text = f"Answer strictly in {target_lang}."
 
     return {"system_prompt": prompt_text}
 
-# --- 3. 엔드포인트 (형님 코드 주소 그대로 복구) ---
+# --- 3. 엔드포인트 ---
 
 @app.get("/")
 def read_root():
@@ -111,15 +115,15 @@ def get_tarot_deck():
 @app.post("/tarot/read")
 def read_tarot(req: TarotRequest):
     try:
-        # DB 매칭
         selected_cards = [card for card in tarot.MAJOR_ARCANA if card["name"] in req.cards]
-        
         persona = get_dynamic_persona("tarot", req.lang)
         
-        # 카드 설명 (단순화)
         card_infos = []
+        # 카드 설명은 한국어 아니면 일단 영어로 나감 (추후 베트남어 DB 있으면 여기도 분기 가능)
+        is_kor = "한국" in req.lang or "Korean" in req.lang
         for c in selected_cards:
-            desc = c.get('desc_eng') or c.get('desc') or c.get('desc_kor') or "No description"
+            desc = c.get('desc_kor') if is_kor else c.get('desc_eng') or c.get('desc')
+            if not desc: desc = "No description"
             card_infos.append(f"- {c['name']}: {desc}")
             
         card_str = "\n".join(card_infos)
@@ -138,12 +142,10 @@ def read_tarot(req: TarotRequest):
         print(traceback.format_exc())
         return {"result": f"Server Error: {str(e)}"}
 
-# 🔥 [복구] 주소: /fengshui/analyze (앱 요청 주소)
 @app.post("/fengshui/analyze")
 def analyze_fengshui(req: FengShuiRequest):
     try:
         gender_kor = "남성" if req.gender in ["Male", "남성"] else "여성"
-        # 형님 코드 변수명 door_dir, head_dir 그대로 사용
         fs_data = calculations.get_fengshui_advanced(req.year, gender_kor, req.door_dir, req.head_dir)
         
         persona = get_dynamic_persona("fengshui", req.lang)
@@ -154,14 +156,12 @@ def analyze_fengshui(req: FengShuiRequest):
     except Exception as e:
         return {"result": f"Error: {str(e)}"}
 
-# 🔥 [복구] 주소: /shaman/read (앱 요청 주소 확인 필요: 형님 코드엔 shaman/read로 되어있었음)
 @app.post("/shaman/read")
 def read_saju(req: SajuRequest):
     try:
         import datetime
         b_date = datetime.date(req.year, req.month, req.day)
         b_time = datetime.time(req.hour, req.minute)
-        # 형님 코드 변수명 calendar_type 그대로 사용
         saju_data = calculations.get_ganji_full(b_date, b_time, req.calendar_type)
         
         persona = get_dynamic_persona("shaman", req.lang)