Update main.py

main.py

@@ -1,22 +1,93 @@
 import streamlit as st
 import time
-from utils import tarot, prompts, llm, calculations, lang # <--- lang 추가
+from utils import tarot, prompts, llm, calculations, lang
 
 # --- [1. 앱 설정] ---
-st.set_page_config(page_title="Global AI Pantheon", page_icon="🔮", layout="wide")
+st.set_page_config(page_title="AI Pantheon", page_icon="🔮", layout="wide")
 
-# --- [2. 테마 스타일링] (기존 코드 유지 - 생략) ---
+# --- [2. 테마 스타일링 엔진 (CSS 완전체)] ---
 def apply_theme(mode):
-    # (이전 답변의 CSS 코드 그대로 사용하세요)
-    # 폰트 로드 부분만 조금 보강해서 다국어 지원하게
+    # 구글 폰트 (한/중/일/영 모두 커버)
     st.markdown("""
     <style>
-    @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');
+    @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&family=Noto+Sans+KR:wght@400;700&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', 'Noto Serif JP', serif;
+                color: #d4af37 !important;
+                text-shadow: 0 0 10px #d4af37;
+            }
+            .stButton>button {
+                background: linear-gradient(45deg, #4b0082, #8a2be2);
+                color: white;
+                border: 1px solid #d4af37;
+                border-radius: 8px;
+            }
+            .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', 'Noto Serif JP', 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;
+            }
+        """
+    }
     
-    # ... (나머지 CSS 테마 코드는 동일하므로 생략) ...
-    pass 
+    style = themes.get(mode, themes["tarot"])
+    st.markdown(f"<style>{style}</style>", unsafe_allow_html=True)
 
 # --- [3. 상태 초기화] ---
 if "messages" not in st.session_state: st.session_state.messages = []
@@ -32,13 +103,11 @@ current_lang = st.sidebar.selectbox(
     ["한국어", "English", "中文", "日本語"],
     label_visibility="collapsed"
 )
-# 선택된 언어 팩 가져오기
-TXT = lang.STRINGS[current_lang]
+TXT = lang.STRINGS[current_lang] # 언어 팩 로드
 
-# --- [4. 모드 선택 UI] ---
+# --- [4. 페르소나 모드 선택] ---
 st.sidebar.title(TXT["sidebar_title"])
 
-# 페르소나 이름도 언어별로 보여주기
 def get_persona_name(key):
     return TXT["personas"][key]
 
@@ -48,11 +117,11 @@ selected_mode = st.sidebar.radio(
     format_func=get_persona_name
 )
 
+# 테마 적용
 apply_theme(selected_mode)
 current_persona = prompts.get_persona(selected_mode)
 
 st.sidebar.markdown("---")
-# 설명도 번역된 걸로 표시
 st.sidebar.info(TXT["desc"][selected_mode])
 
 # --- [5. 메인 UI 로직] ---
@@ -60,28 +129,32 @@ st.title(f"{current_persona['icon']} {get_persona_name(selected_mode)}")
 st.markdown("---")
 
 # ==========================================
-# 🃏 모드 A: 타로
+# 🃏 모드 A: 타로 (Tarot)
 # ==========================================
 if selected_mode == "tarot":
+    # 1단계: 입력
     if st.session_state.tarot_stage == "input":
         with st.form("tarot_form"):
-            query = st.text_area(TXT["input_label"])
+            topic = st.selectbox(TXT["tarot_topic"], ["Love", "Money", "Career", "Health"])
+            query = st.text_area(TXT["input_label"], placeholder=TXT["tarot_placeholder"])
             submitted = st.form_submit_button(TXT["btn_submit"])
             
             if submitted and query:
-                st.session_state.user_query = query
+                st.session_state.user_query = f"[{topic}] {query}"
                 st.session_state.tarot_stage = "picking"
                 st.rerun()
 
+    # 2단계: 뽑기
     elif st.session_state.tarot_stage == "picking":
-        st.info(f"Your Worry: **{st.session_state.user_query}**")
-        if st.button("🎴 Draw 3 Cards", type="primary", use_container_width=True):
+        st.info(f"Query: **{st.session_state.user_query}**")
+        if st.button(TXT["tarot_pick_btn"], 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)
         for idx, card in enumerate(st.session_state.picked_cards):
@@ -90,37 +163,41 @@ if selected_mode == "tarot":
         
         if "ai_response" not in st.session_state:
             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,
-                    user_lang=current_lang 
+                    user_lang=current_lang
                 )
+        
         st.subheader(TXT["result_title"])
         st.markdown(st.session_state.ai_response)
         
-        if st.button("🔄 Retry"):
+        if st.button(TXT["retry_btn"]):
             st.session_state.tarot_stage = "input"
             del st.session_state["ai_response"]
             st.rerun()
 
 # ==========================================
-# 🏡 모드 B: 풍수지리
+# 🏡 모드 B: 풍수지리 (Feng Shui)
 # ==========================================
 elif selected_mode == "fengshui":
     with st.container():
+        st.caption(TXT["fengshui_info"])
         c1, c2 = st.columns(2)
         with c1:
-            birth_year = st.number_input("Year of Birth", 1950, 2025, 1990)
-            gender = st.radio("Gender", ["남성", "여성"]) # 계산식 때문에 값은 한글로 유지하되 표시는 바꿀 수 있음
+            birth_year = st.number_input(TXT["birth_year"], 1950, 2025, 1990)
+            # 성별은 계산 로직 때문에 내부값은 '남성/여성' 유지
+            gender_label = st.radio(TXT["gender"], ["Male", "Female"])
+            gender_val = "남성" if gender_label == "Male" else "여성"
         with c2:
-            direction = st.selectbox("Door Direction", ["북", "북동", "동", "남동", "남", "남서", "서", "북서"])
+            direction = st.selectbox(TXT["door_dir"], ["N", "NE", "E", "SE", "S", "SW", "W", "NW"])
     
     if prompt := st.chat_input(TXT["input_label"]):
-        fengshui_data = calculations.get_fengshui_info(birth_year, gender)
+        # 1. 계산
+        fengshui_data = calculations.get_fengshui_info(birth_year, gender_val)
         
-        # 시스템 프롬프트용 데이터 구성
+        # 2. AI 요청
         sys_msg = f"[Data]: {fengshui_data}, Current Door: {direction}"
         full_query = f"{sys_msg}\nUser Question: {prompt}"
 
@@ -131,19 +208,22 @@ elif selected_mode == "fengshui":
                 st.write(response)
 
 # ==========================================
-# 👹 모드 C: 신점
+# 👹 모드 C: 신점 (Shaman)
 # ==========================================
 elif selected_mode == "shaman":
     with st.container():
+        st.warning(TXT["shaman_warn"])
         c1, c2 = st.columns(2)
         with c1:
-            birth_year = st.number_input("Birth Year", 1960, 2025, 1995)
+            birth_year = st.number_input(TXT["birth_year"], 1960, 2025, 1995)
         with c2:
             prompt_input = st.text_input(TXT["input_label"])
 
     if st.button(TXT["btn_submit"]) and prompt_input:
+        # 1. 계산
         saju_data = calculations.get_ganji(birth_year)
         
+        # 2. AI 요청
         sys_msg = f"[Saju Data]: Year {saju_data['year_ganji']}, Element {saju_data['element']}"
         full_query = f"{sys_msg}\nUser Worry: {prompt_input}"
 
@@ -152,4 +232,7 @@ elif selected_mode == "shaman":
         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)
+                st.markdown(response)
+                
+            # 명식표 출력
+            st.info(f"📜 {TXT['saju_result']}: {saju_data['year_ganji']} ({saju_data['animal']})")