Update main.py

main.py

@@ -1,240 +1,155 @@
 import streamlit as st
 import time
-from utils import tarot, prompts, llm, calculations
+from utils import tarot, prompts, llm, calculations, lang # <--- lang 추가
 
 # --- [1. 앱 설정] ---
-st.set_page_config(page_title="AI 만신전", page_icon="🔮", layout="wide")
+st.set_page_config(page_title="Global AI Pantheon", page_icon="🔮", layout="wide")
 
-# --- [2. 테마 스타일링 엔진] ---
+# --- [2. 테마 스타일링] (기존 코드 유지 - 생략) ---
 def apply_theme(mode):
-    # 구글 폰트 로드
+    # (이전 답변의 CSS 코드 그대로 사용하세요)
+    # 폰트 로드 부분만 조금 보강해서 다국어 지원하게
     st.markdown("""
     <style>
-    @import url('https://fonts.googleapis.com/css2?family=Nanum+Myeongjo:wght@700&family=Nanum+Pen+Script&family=Noto+Sans+KR:wght@400;700&display=swap');
+    @import url('https://fonts.googleapis.com/css2?family=Nanum+Myeongjo:wght@700&family=Noto+Serif+SC:wght@600&family=Noto+Serif+JP:wght@600&family=Nanum+Pen+Script&display=swap');
     </style>
     """, unsafe_allow_html=True)
-
-    themes = {
-        "tarot": """
-            /* 🔮 타로 테마 */
-            .stApp {
-                background: linear-gradient(135deg, #1a0b2e 0%, #2d1b4e 100%);
-                color: #e0d4fc;
-            }
-            h1, h2, h3 {
-                font-family: 'Nanum Myeongjo', serif;
-                color: #d4af37 !important;
-                text-shadow: 0 0 10px #d4af37;
-            }
-            .stButton>button {
-                background: linear-gradient(45deg, #4b0082, #8a2be2);
-                color: white;
-                border: 1px solid #d4af37;
-            }
-            .stTextInput>div>div>input, .stTextArea>div>div>textarea {
-                background-color: #2d1b4e;
-                color: white;
-                border: 1px solid #d4af37;
-            }
-        """,
-        "fengshui": """
-            /* 🏡 풍수 테마 */
-            .stApp {
-                background-color: #f5f5f0;
-                background-image: url("https://www.transparenttextures.com/patterns/rice-paper.png");
-                color: #3e2723;
-            }
-            h1, h2, h3 {
-                font-family: 'Nanum Myeongjo', serif;
-                color: #2e7d32 !important;
-            }
-            .stButton>button {
-                background-color: #5d4037;
-                color: #fff8e1;
-                border-radius: 0px;
-            }
-            .stChatMessage {
-                background-color: #ffffff;
-                border: 1px solid #d7ccc8;
-            }
-        """,
-        "shaman": """
-            /* 👹 점집 테마 */
-            .stApp {
-                background-color: #1a0505;
-                color: #ffcccc;
-            }
-            h1, h2, h3 {
-                font-family: 'Nanum Pen Script', cursive;
-                font-size: 3.5rem !important;
-                color: #ff0000 !important;
-                text-shadow: 2px 2px 0px #ffff00;
-            }
-            .stButton>button {
-                background-color: #b71c1c;
-                color: #ffff00;
-                border: 2px solid #ffff00;
-                font-family: 'Nanum Pen Script', cursive;
-                font-size: 1.5rem;
-            }
-            .stTextInput>div>div>input {
-                background-color: #330000;
-                color: yellow;
-                border: 2px solid red;
-            }
-        """
-    }
     
-    style = themes.get(mode, themes["tarot"])
-    st.markdown(f"<style>{style}</style>", unsafe_allow_html=True)
+    # ... (나머지 CSS 테마 코드는 동일하므로 생략) ...
+    pass 
 
 # --- [3. 상태 초기화] ---
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-if "tarot_stage" not in st.session_state:
-    st.session_state.tarot_stage = "input"
-if "picked_cards" not in st.session_state:
-    st.session_state.picked_cards = []
-
-# --- [4. 사이드바 UI] ---
-st.sidebar.title("⛩️ 만신전 입장")
+if "messages" not in st.session_state: st.session_state.messages = []
+if "tarot_stage" not in st.session_state: st.session_state.tarot_stage = "input"
+if "picked_cards" not in st.session_state: st.session_state.picked_cards = []
+
+# ==========================================
+# 🌐 언어 선택 (사이드바 최상단)
+# ==========================================
+st.sidebar.markdown("### 🌐 Language")
+current_lang = st.sidebar.selectbox(
+    "Select Language", 
+    ["한국어", "English", "中文", "日本語"],
+    label_visibility="collapsed"
+)
+# 선택된 언어 팩 가져오기
+TXT = lang.STRINGS[current_lang]
+
+# --- [4. 모드 선택 UI] ---
+st.sidebar.title(TXT["sidebar_title"])
+
+# 페르소나 이름도 언어별로 보여주기
+def get_persona_name(key):
+    return TXT["personas"][key]
+
 selected_mode = st.sidebar.radio(
-    "누구를 찾으셨습니까?",
+    TXT["select_persona"],
     ("tarot", "fengshui", "shaman"),
-    format_func=lambda x: prompts.get_persona(x)["name"]
+    format_func=get_persona_name
 )
 
-# 테마 적용 및 페르소나 로드
 apply_theme(selected_mode)
 current_persona = prompts.get_persona(selected_mode)
 
 st.sidebar.markdown("---")
-st.sidebar.success(f"현재 접속: **{current_persona['name']}**")
-st.sidebar.info(current_persona["description"])
+# 설명도 번역된 걸로 표시
+st.sidebar.info(TXT["desc"][selected_mode])
 
 # --- [5. 메인 UI 로직] ---
-st.title(f"{current_persona['icon']} {current_persona['name']}")
+st.title(f"{current_persona['icon']} {get_persona_name(selected_mode)}")
 st.markdown("---")
 
 # ==========================================
-# 🃏 모드 A: 타로 (Tarot)
+# 🃏 모드 A: 타로
 # ==========================================
 if selected_mode == "tarot":
-    # 1단계: 입력
     if st.session_state.tarot_stage == "input":
         with st.form("tarot_form"):
-            st.subheader("무엇이 당신을 번뇌하게 합니까?")
-            topic = st.selectbox("고민 주제", ["💘 연애/사랑", "💰 금전/사업", "🎓 학업/진로", "🫂 인간관계"])
-            query = st.text_area("구체적인 고민을 적어주세요", placeholder="예: 썸타는 사람이 있는데 연락이 안 와요.")
-            submitted = st.form_submit_button("운명의 카드 뽑으러 가기")
+            query = st.text_area(TXT["input_label"])
+            submitted = st.form_submit_button(TXT["btn_submit"])
             
             if submitted and query:
-                st.session_state.user_query = f"[{topic}] {query}"
+                st.session_state.user_query = query
                 st.session_state.tarot_stage = "picking"
                 st.rerun()
 
-    # 2단계: 뽑기
     elif st.session_state.tarot_stage == "picking":
-        st.info(f"당신의 고민: **{st.session_state.user_query}**")
-        st.write("▼ 아래 버튼을 눌러 운명을 확인하세요 ▼")
-        if st.button("🎴 신중하게 카드 3장 뽑기", type="primary", use_container_width=True):
-            with st.spinner("운명을 점지하는 중..."):
+        st.info(f"Your Worry: **{st.session_state.user_query}**")
+        if st.button("🎴 Draw 3 Cards", type="primary", use_container_width=True):
+            with st.spinner(TXT["loading"]):
                 time.sleep(1.5)
                 st.session_state.picked_cards = tarot.draw_tarot_cards(3)
                 st.session_state.tarot_stage = "result"
                 st.rerun()
 
-    # 3단계: 결과
     elif st.session_state.tarot_stage == "result":
         cols = st.columns(3)
-        positions = ["과거", "현재", "미래"]
         for idx, card in enumerate(st.session_state.picked_cards):
             with cols[idx]:
-                st.caption(positions[idx])
                 st.image(tarot.get_card_image(card["name"]), caption=card["name"])
         
         if "ai_response" not in st.session_state:
-            with st.spinner("해석 중..."):
+            with st.spinner(TXT["analyzing"]):
+                # 언어 정보(current_lang)를 함께 넘김
                 st.session_state.ai_response = llm.get_ai_response(
-                    st.session_state.user_query, current_persona, st.session_state.picked_cards
+                    st.session_state.user_query, 
+                    current_persona, 
+                    st.session_state.picked_cards,
+                    user_lang=current_lang 
                 )
+        st.subheader(TXT["result_title"])
         st.markdown(st.session_state.ai_response)
         
-        if st.button("🔄 다시 하기"):
+        if st.button("🔄 Retry"):
             st.session_state.tarot_stage = "input"
             del st.session_state["ai_response"]
             st.rerun()
 
 # ==========================================
-# 🏡 모드 B: 풍수지리 (Feng Shui)
+# 🏡 모드 B: 풍수지리
 # ==========================================
 elif selected_mode == "fengshui":
     with st.container():
-        st.info("정통 본명궁법(Life Gua)에 의해 귀하의 길흉 방위를 계산합니다.")
-        
         c1, c2 = st.columns(2)
         with c1:
-            birth_year = st.number_input("가장(집주인) 출생연도", 1950, 2025, 1990)
-            gender = st.radio("성별", ["남성", "여성"])
+            birth_year = st.number_input("Year of Birth", 1950, 2025, 1990)
+            gender = st.radio("Gender", ["남성", "여성"]) # 계산식 때문에 값은 한글로 유지하되 표시는 바꿀 수 있음
         with c2:
-            direction = st.selectbox("현재 현관 방향", ["북", "북동", "동", "남동", "남", "남서", "서", "북서"])
-            room_type = st.selectbox("진단할 공간", ["침실", "거실", "서재"])
-
-    if prompt := st.chat_input("구조에 대해 물어보세요 (예: 현관에 거울이 있어)"):
-        # 1. 계산 실행
+            direction = st.selectbox("Door Direction", ["북", "북동", "동", "남동", "남", "남서", "서", "북서"])
+    
+    if prompt := st.chat_input(TXT["input_label"]):
         fengshui_data = calculations.get_fengshui_info(birth_year, gender)
         
-        # 2. AI에게 넘길 정보 구성
-        sys_msg = f"""
-        [데이터 분석]
-        - 본명궁: {fengshui_data['gua_number']} ({fengshui_data['group']})
-        - 길(Good) 방위: {fengshui_data['good_direction']}
-        - 흉(Bad) 방위: {fengshui_data['bad_direction']}
-        - 현재 현관: {direction}쪽
-        """
-        full_query = f"{sys_msg}\n사용자 질문: {prompt}\n\n위 데이터를 근거로 논리적으로 조언해."
+        # 시스템 프롬프트용 데이터 구성
+        sys_msg = f"[Data]: {fengshui_data}, Current Door: {direction}"
+        full_query = f"{sys_msg}\nUser Question: {prompt}"
 
         with st.chat_message("user"): st.write(prompt)
         with st.chat_message("assistant"):
-            with st.spinner("방위를 계산하는 중..."):
-                response = llm.get_ai_response(full_query, current_persona)
+            with st.spinner(TXT["analyzing"]):
+                response = llm.get_ai_response(full_query, current_persona, user_lang=current_lang)
                 st.write(response)
-            
-            # 분석표 출력
-            with st.expander("📊 정밀 분석표 보기", expanded=True):
-                st.success(f"✅ 추천 방향: {fengshui_data['good_direction']}")
-                st.error(f"🚫 피할 방향: {fengshui_data['bad_direction']}")
 
 # ==========================================
-# 👹 모드 C: 신점 (Shaman)
+# 👹 모드 C: 신점
 # ==========================================
 elif selected_mode == "shaman":
     with st.container():
-        st.warning("거짓으로 적으면 점괘 안 나온다! 똑바로 적어라.")
-        
         c1, c2 = st.columns(2)
         with c1:
-            birth_year = st.number_input("태어난 연도", 1960, 2025, 1995)
+            birth_year = st.number_input("Birth Year", 1960, 2025, 1995)
         with c2:
-            prompt_input = st.text_input("뭐가 궁금하냐?", placeholder="올해 결혼운 어때?")
+            prompt_input = st.text_input(TXT["input_label"])
 
-    if st.button("점괘 확인") and prompt_input:
-        # 1. 사주 계산 실행
+    if st.button(TXT["btn_submit"]) and prompt_input:
         saju_data = calculations.get_ganji(birth_year)
         
-        # 2. AI 프롬프트 구성
-        sys_msg = f"""
-        [내담자 사주]
-        - {saju_data['year_ganji']} ({saju_data['animal']}띠)
-        - 타고난 기운: {saju_data['element']} ({saju_data['color']}색)
-        """
-        full_query = f"{sys_msg}\n고민: {prompt_input}\n\n내담자는 '{saju_data['element']}' 기운이다. 이에 빗대어 호통치며 조언해라."
+        sys_msg = f"[Saju Data]: Year {saju_data['year_ganji']}, Element {saju_data['element']}"
+        full_query = f"{sys_msg}\nUser Worry: {prompt_input}"
 
         st.divider()
-        st.write(f"🗣️ **나:** {prompt_input}")
-        
-        with st.spinner("방울 흔드는 중... 🔔"):
-            response = llm.get_ai_response(full_query, current_persona)
-            st.markdown(f"**👹 무당:** {response}")
-            
-            st.info(f"📜 명식: **{saju_data['year_ganji']} ({saju_data['animal']})** - 너는 **{saju_data['element']}**의 기운이야!")
+        with st.chat_message("user"): st.write(prompt_input)
+        with st.chat_message("assistant"):
+            with st.spinner(TXT["loading"]):
+                response = llm.get_ai_response(full_query, current_persona, user_lang=current_lang)
+                st.markdown(response)