Create main.py

main.py

@@ -0,0 +1,110 @@
+import streamlit as st
+from utils import tarot, prompts, llm
+
+# --- [1. 앱 설정] ---
+st.set_page_config(page_title="AI 만신전", page_icon="🔮", layout="wide")
+
+# 세션 상태 초기화
+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 = []
+
+# --- [2. 사이드바: 모드 선택] ---
+st.sidebar.title("⛩️ 모드 선택")
+selected_mode = st.sidebar.radio(
+    "누구를 만나시겠습니까?",
+    ("tarot", "fengshui", "shaman"),
+    format_func=lambda x: prompts.get_persona(x)["name"]
+)
+
+# 현재 모드의 페르소나 로드
+current_persona = prompts.get_persona(selected_mode)
+
+st.sidebar.markdown("---")
+st.sidebar.info(current_persona["description"])
+st.sidebar.markdown(f"**현재 모드:** {current_persona['icon']} {current_persona['name']}")
+
+
+# --- [3. 메인 화면 로직] ---
+st.title(f"{current_persona['icon']} {current_persona['name']}")
+st.caption("AI가 당신의 운명을 시뮬레이션합니다.")
+st.divider()
+
+# ==========================================
+#  🃏 모드 A: 타로 마스터
+# ==========================================
+if selected_mode == "tarot":
+    # 1단계: 고민 입력
+    if st.session_state.tarot_stage == "input":
+        st.subheader("무엇이 궁금하신가요?")
+        query = st.text_input("고민을 입력하세요", key="tarot_query")
+        
+        if st.button("운명 확인하기") and query:
+            st.session_state.user_query = query
+            st.session_state.tarot_stage = "picking"
+            st.rerun()
+
+    # 2단계: 카드 뽑기
+    elif st.session_state.tarot_stage == "picking":
+        st.subheader(f"'{st.session_state.user_query}'에 대한 답을 찾습니다.")
+        st.info("아래 버튼을 눌러 카드를 뽑으세요.")
+        
+        if st.button("🎴 카드 3장 뽑기", type="primary", use_container_width=True):
+            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):
+            with cols[idx]:
+                st.image(tarot.get_card_image(card["name"]), caption=card["name"])
+                st.write(f"**{card['name']}**: {card['desc']}")
+        
+        st.divider()
+        
+        # AI 해석 생성
+        if "ai_response" not in st.session_state:
+            with st.spinner("운명을 해석하는 중입니다..."):
+                response = llm.get_ai_response(
+                    st.session_state.user_query, 
+                    current_persona, 
+                    context_data=st.session_state.picked_cards
+                )
+                st.session_state.ai_response = response
+        
+        st.write(st.session_state.ai_response)
+        
+        if st.button("🔄 다시 상담하기"):
+            st.session_state.tarot_stage = "input"
+            del st.session_state["ai_response"] # 이전 응답 삭제
+            st.rerun()
+
+# ==========================================
+#  🏡 & 👹 모드 B/C: 풍수 & 점쟁이 (채팅형)
+# ==========================================
+else:
+    # 채팅 기록 표시
+    for msg in st.session_state.messages:
+        with st.chat_message(msg["role"]):
+            st.write(msg["content"])
+
+    # 입력창
+    if prompt := st.chat_input("고민을 털어놓으세요..."):
+        # 사용자 메시지
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.write(prompt)
+
+        # AI 응답
+        with st.chat_message("assistant"):
+            with st.spinner(f"{current_persona['name']}가 접신 중..."):
+                response = llm.get_ai_response(prompt, current_persona)
+                st.write(response)
+        
+        st.session_state.messages.append({"role": "assistant", "content": response})