Update app.py

app.py

@@ -1,316 +1,274 @@
-import streamlit as st
+# app.py
+import os
+import json
 import random
 from datetime import datetime
 
+import streamlit as st
+
+# ========= Page Config =========
 st.set_page_config(
     page_title="SoulCompass - Find Your Inner North Star",
     page_icon="🧭",
     layout="wide"
 )
 
-# 初始化
-if 'daily_usage' not in st.session_state:
+# ========= Session Init =========
+if "daily_usage" not in st.session_state:
     st.session_state.daily_usage = 0
+if "journal" not in st.session_state:
+    st.session_state.journal = []
 
-# 簡化塔羅牌
-tarot_cards = [
-    "The Fool - New beginnings and fresh starts",
-    "The Magician - Power to manifest your dreams",
-    "The High Priestess - Trust your intuition",
-    "The Empress - Creativity and abundance",
-    "The Emperor - Structure and authority",
-    "The Lovers - Love and important choices",
-    "The Chariot - Victory through determination",
-    "Strength - Inner courage and compassion",
-    "The Hermit - Soul searching and wisdom",
-    "Wheel of Fortune - Life cycles and destiny",
-    "Justice - Balance and fairness",
-    "The Hanged Man - New perspective needed",
-    "Death - Transformation and renewal",
-    "Temperance - Patience and moderation",
-    "The Devil - Breaking free from limitations",
-    "The Tower - Sudden change and awakening",
-    "The Star - Hope and inspiration",
-    "The Moon - Dreams and subconscious",
-    "The Sun - Joy and success",
-    "Judgement - Spiritual awakening",
-    "The World - Completion and achievement"
-]
-
-# CSS
+# ========= CSS (simple) =========
 st.markdown("""
 <style>
-.header {
-    text-align: center;
-    background: linear-gradient(135deg, #6366F1, #C084FC);
-    color: white;
-    padding: 3rem;
-    border-radius: 20px;
-    margin-bottom: 2rem;
-}
-.card {
-    background: linear-gradient(135deg, #1E3A8A, #6366F1);
-    color: white;
-    padding: 2rem;
-    border-radius: 15px;
-    margin: 1rem;
-    text-align: center;
-    border: 2px solid #F59E0B;
-}
-.result {
-    background: linear-gradient(135deg, #6366F1, #C084FC);
-    color: white;
-    padding: 2rem;
-    border-radius: 15px;
-    margin: 2rem 0;
-}
-.quota {
-    background: linear-gradient(135deg, #F59E0B, #FB923C);
-    color: white;
-    padding: 1rem 2rem;
-    border-radius: 50px;
-    text-align: center;
-    font-weight: bold;
-    margin: 2rem auto;
-    max-width: 400px;
-}
+.header {text-align:center;background:linear-gradient(135deg,#6366F1,#C084FC);color:white;padding:2.2rem;border-radius:18px;margin-bottom:1.2rem}
+.badge {display:inline-block;background:rgba(255,255,255,.18);padding:.35rem .75rem;border-radius:999px;margin:.25rem}
+.card {background:linear-gradient(135deg,#1E3A8A,#6366F1);color:white;padding:1.1rem;border-radius:12px;border:2px solid #F59E0B}
+.result {background:linear-gradient(135deg,#6366F1,#C084FC);color:white;padding:1.1rem;border-radius:12px}
+.quota {background:linear-gradient(135deg,#F59E0B,#FB923C);color:white;padding:.5rem 1rem;border-radius:999px;
+        text-align:center;font-weight:700;margin:0 auto;max-width:360px}
+.small {font-size:.9rem;color:#6B7280}
+hr{border:none;height:1px;background:#E5E7EB;margin:1.2rem 0}
 </style>
 """, unsafe_allow_html=True)
 
-# 主標題
+# ========= Header =========
 st.markdown("""
 <div class="header">
-    <h1>🧭 SoulCompass</h1>
-    <h2>Find Your Inner North Star</h2>
-    <p>🔮 Tarot Reading | 🔢 Numerology | 📖 Soul Journal | 🤖 AI Therapist</p>
-    <div style="margin-top: 20px;">
-        <span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 20px; margin: 10px;">
-            ⭐ 15,247 users
-        </span>
-        <span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 20px; margin: 10px;">
-            💯 4.9/5 rating
-        </span>
-    </div>
-</div>
-""", unsafe_allow_html=True)
-
-# 剩餘次數
-remaining = 5 - st.session_state.daily_usage
-st.markdown(f"""
-<div class="quota">
-    🎫 Daily Free Readings: {remaining}/5
+  <h1>🧭 SoulCompass</h1>
+  <h3>Find Your Inner North Star</h3>
+  <div>
+    <span class="badge">🔮 Tarot</span>
+    <span class="badge">🔢 Numerology</span>
+    <span class="badge">📖 Soul Journal</span>
+    <span class="badge">🤖 AI Guide</span>
+  </div>
 </div>
 """, unsafe_allow_html=True)
 
-# 主要功能區塊
-tab1, tab2, tab3, tab4 = st.tabs(["🔮 Tarot Reading", "🔢 Numerology", "📖 Soul Journal", "🤖 AI Therapist"])
-
-# 🔮 Tarot
+# ========= Sidebar (Model / RAG Controls) =========
+st.sidebar.header("⚙️ Settings")
+model_choice = st.sidebar.selectbox(
+    "LLM Model",
+    ["cerebras/btlm-3b-8k-base", "mistralai/Mistral-7B-Instruct-v0.2"],
+    help="BTLM 輕量穩定、適合免費空間；Mistral 口條更好、但較吃資源。"
+)
+rag_k = st.sidebar.slider("RAG Top-K", 1, 5, 3)
+temperature = st.sidebar.slider("Generation Temperature", 0.2, 1.2, 0.7, 0.1)
+
+st.sidebar.markdown("---")
+st.sidebar.caption("資料庫：`data/tarot_data_full.json`, `data/numerology_data_full.json`")
+st.sidebar.caption("檢索程式：`rag_utils.py`")
+
+# ========= RAG + LLM Loader =========
+@st.cache_resource
+def load_rag():
+    try:
+        import rag_utils
+        return rag_utils
+    except Exception as e:
+        st.warning("無法載入 rag_utils，請確認檔案存在於專案根目錄。")
+        return None
+
+@st.cache_resource
+def load_llm(model_id: str):
+    import torch
+    from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+    tok = AutoTokenizer.from_pretrained(model_id)
+    model = AutoModelForCausalLM.from_pretrained(
+        model_id,
+        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+        device_map="auto",
+    )
+    pipe = pipeline(
+        "text-generation",
+        model=model,
+        tokenizer=tok,
+        max_new_tokens=360,
+        temperature=temperature,
+        top_p=0.9,
+        repetition_penalty=1.08,
+        do_sample=True,
+    )
+    return pipe
+
+rag = load_rag()
+llm = load_llm(model_choice)
+
+def rag_answer(query: str, domain: str = "Tarot", k: int = 3):
+    """Search with FAISS and generate an answer using the LLM."""
+    if rag is None:
+        return "RAG 模組未載入，請檢查 rag_utils.py。", []
+    # retrieve
+    hits = rag.search_tarot(query, k) if domain == "Tarot" else rag.search_numerology(query, k)
+    context = "\n\n".join([h["text"] for h in hits]) if hits else "（無檢索結果）"
+    # prompt
+    prompt = f"""You are SoulCompass, a compassionate spiritual guide.
+Use ONLY the CONTEXT to answer the USER in Traditional Chinese with empathy and clarity.
+If info is missing, admit briefly and suggest a next step.
+
+CONTEXT:
+{context}
+
+USER QUESTION: {query}
+
+請用 5~8 句繁體中文回覆，語氣溫柔、務實，最後給出 1 句可行的小建議。"""
+    out = llm(prompt)[0]["generated_text"]
+    answer = out[len(prompt):].strip()
+    return answer, hits
+
+# ========= Data Helpers =========
+@st.cache_resource
+def load_tarot_names():
+    path = os.path.join("data", "tarot_data_full.json")
+    if not os.path.exists(path):
+        # tiny fallback
+        return ["The Fool", "The Magician", "The High Priestess", "The Empress", "The Emperor"]
+    with open(path, "r", encoding="utf-8") as f:
+        data = json.load(f)
+    return [c["name"] if "name" in c else c.get("card_name", "Card") for c in data]
+
+TAROT_NAMES = load_tarot_names()
+
+# ========= Tabs =========
+tab1, tab2, tab3, tab4 = st.tabs(["🔮 Tarot", "🔢 Numerology", "📖 Journal", "🤖 AI Guide"])
+
+# ---- TAB 1: Tarot ----
 with tab1:
-    st.header("🔮 Tarot Reading")
+    st.subheader("🔮 Tarot Reading")
     st.write("*Let ancient wisdom guide your path*")
 
-    question = st.text_area(
-        "What would you like guidance on?",
-        placeholder="Example: I'm facing a difficult decision at work...",
-        height=100
-    )
-
-    if question:
+    colA, colB = st.columns([2, 1])
+    with colA:
+        question = st.text_area(
+            "你想得到什麼指引？",
+            placeholder="例如：我適合換工作嗎？和他未來的關係會如何？",
+            height=90
+        )
+    with colB:
         spread = st.selectbox(
-            "Choose your reading type",
-            ["Single Card - Quick Guidance", "Three Cards - Past Present Future", "Five Cards - Deep Insight"]
+            "抽牌陣列",
+            ["Single (1 card)", "Three (Past • Present • Future)", "Five (Deep Insight)"]
         )
 
-        if st.button("🔮 Draw Cards", type="primary"):
-            if st.session_state.daily_usage >= 5:
-                st.error("😔 Daily limit reached! Come back tomorrow")
-            else:
-                st.session_state.daily_usage += 1
-
-                with st.spinner("✨ Drawing your cards..."):
-                    import time
-                    time.sleep(2)
-
-                    num_cards = 1 if "Single" in spread else (3 if "Three" in spread else 5)
-                    drawn = random.sample(tarot_cards, num_cards)
-
-                    st.success("🌟 Your reading is complete!")
-
-                    for i, card in enumerate(drawn):
-                        st.markdown(f"""
-                        <div class="card">
-                            <h3>Card {i+1}</h3>
-                            <p>{card}</p>
-                        </div>
-                        """, unsafe_allow_html=True)
-
-                    # 最終解讀
-                    reading = f"""
-🔮 **Your SoulCompass Reading**
-
-**Question:** {question}
-
-**Cards Drawn:** {len(drawn)} cards
-
-**Interpretation:**
-The cards reveal important insights about your situation. The first card shows your current energy, while the others provide guidance for moving forward.
-
-**Key Message:** Trust your inner wisdom and stay open to new possibilities. The universe is supporting your journey.
-
-**Readings used today:** {st.session_state.daily_usage}/5
-"""
-
-                    st.markdown(f"""
-                    <div class="result">
-                        {reading.replace(chr(10), '<br>')}
-                    </div>
-                    """, unsafe_allow_html=True)
-
-# 🔢 Numerology
+    draw = st.button("🔮 抽牌")
+    if draw and not question:
+        st.warning("請先輸入你的問題")
+    if draw and question:
+        if st.session_state.daily_usage >= 5:
+            st.error("今日免費占卜次數已滿，明天再來吧！")
+        else:
+            st.session_state.daily_usage += 1
+            num_cards = 1 if "Single" in spread else (3 if "Three" in spread else 5)
+            drawn = random.sample(TAROT_NAMES, k=min(num_cards, len(TAROT_NAMES)))
+
+            st.success("🌟 抽牌完成")
+            cols = st.columns(len(drawn))
+            for i, name in enumerate(drawn):
+                with cols[i]:
+                    st.markdown(f'<div class="card"><h4>Card {i+1}</h4><p>{name}</p></div>', unsafe_allow_html=True)
+
+            # RAG + LLM 解讀：把「問題 + 牌名列表」一併送去檢索
+            query = f"問題：{question}；抽到的牌：{', '.join(drawn)}。請用塔羅牌義做整體解讀。"
+            with st.spinner("RAG 檢索中 & AI 正在生成解讀…"):
+                answer, refs = rag_answer(query, domain="Tarot", k=rag_k)
+
+            st.markdown(f'<div class="result">{answer.replace(chr(10), "<br>")}</div>', unsafe_allow_html=True)
+            st.caption(f"已使用：{st.session_state.daily_usage}/5 次")
+
+            with st.expander("查看參考段落（RAG 來源）"):
+                for i, r in enumerate(refs, 1):
+                    tag = r.get("card_name") or r.get("name") or "Card"
+                    st.markdown(f"**#{i} {tag}** · score: {r['score']:.3f}")
+                    st.write(r["text"])
+                    st.markdown("---")
+
+# ---- TAB 2: Numerology ----
 with tab2:
-    st.header("🔢 Numerology")
+    st.subheader("🔢 Numerology")
     st.write("*Discover your numbers*")
 
     col1, col2 = st.columns(2)
-
     with col1:
-        st.subheader("Life Path Number")
-        birth_date = st.date_input("Your birth date")
-
-        if birth_date and st.button("Calculate"):
-            date_str = birth_date.strftime('%Y%m%d')
-            total = sum(int(d) for d in date_str)
-            while total > 9:
-                total = sum(int(d) for d in str(total))
-
-            meanings = {
-                1: "Leader - Independent and pioneering",
-                2: "Peacemaker - Cooperative and diplomatic",
-                3: "Creative - Expressive and optimistic",
-                4: "Builder - Practical and hardworking",
-                5: "Explorer - Freedom-loving and adventurous",
-                6: "Nurturer - Caring and responsible",
-                7: "Seeker - Analytical and spiritual",
-                8: "Achiever - Ambitious and organized",
-                9: "Humanitarian - Compassionate and wise"
-            }
-
-            st.success(f"Your Life Path Number: **{total}**")
-            st.info(meanings.get(total, "Special number with unique meaning"))
+        birth_date = st.date_input("你的生日")
+        if st.button("計算生命靈數"):
+            if birth_date:
+                s = birth_date.strftime("%Y%m%d")
+                total = sum(int(d) for d in s)
+                while total > 9 and total not in (11, 22, 33):
+                    total = sum(int(d) for d in str(total))
+                st.success(f"你的生命靈數是：{total}")
+                with st.spinner("查詢靈數意義…"):
+                    q = f"生命靈數 {total} 的核心特質與建議是什麼？"
+                    ans, refs = rag_answer(q, domain="Numerology", k=rag_k)
+                st.markdown(f'<div class="result">{ans.replace(chr(10), "<br>")}</div>', unsafe_allow_html=True)
+                with st.expander("參考段落"):
+                    for i, r in enumerate(refs, 1):
+                        tag = r.get("number") or "N/A"
+                        st.markdown(f"**#{i} Number {tag}** · score: {r['score']:.3f}")
+                        st.write(r["text"])
+                        st.markdown("---")
+            else:
+                st.warning("請先選擇生日")
 
     with col2:
-        st.subheader("Name Number")
-        name = st.text_input("Your full name")
-
-        if name and st.button("Analyze Name"):
-            name_num = len(name) % 9 + 1
-            st.success(f"Your Name Number: **{name_num}**")
-            st.info("This number influences how others see you")
+        name = st.text_input("你的全名（可選）")
+        if st.button("簡易名字數字"):
+            if name.strip():
+                n = (len(name.replace(" ", "")) % 9) + 1
+                st.info(f"名字數字（簡式）：{n}")
+                q = f"名字數字 {n} 的人在人際與自我表達上的建議？"
+                ans, _ = rag_answer(q, domain="Numerology", k=rag_k)
+                st.write(ans)
+            else:
+                st.warning("請輸入名字")
 
-# 📖 Soul Journal
+# ---- TAB 3: Journal ----
 with tab3:
-    st.header("📖 Soul Journal")
-    st.write("*Record your inner journey*")
-
-    mood = st.slider("Today's mood (1-10)", 1, 10, 7)
-    emotion = st.selectbox("Main emotion", ["😊 Happy", "😔 Sad", "😰 Worried", "😌 Calm", "🤗 Grateful"])
-
-    entry = st.text_area(
-        "Write your thoughts",
-        placeholder="How are you feeling today? What's on your mind?",
-        height=120
-    )
-
-    if st.button("💾 Save Entry"):
-        if entry:
-            st.success("✅ Your journal entry has been saved!")
-            st.balloons()
+    st.subheader("📖 Soul Journal")
+    mood = st.slider("今天的心情（1-10）", 1, 10, 7)
+    emotion = st.selectbox("主要情緒", ["😊 開心", "😔 失落", "😰 焦慮", "😌 平靜", "🤗 感恩"])
+    entry = st.text_area("寫下今天的想法", height=140,
+                         placeholder="你最近在思考什麼？今天最感謝的一件事是？")
+
+    if st.button("💾 儲存日記"):
+        if entry.strip():
+            st.session_state.journal.append({
+                "time": datetime.now().strftime("%Y-%m-%d %H:%M"),
+                "mood": mood,
+                "emotion": emotion,
+                "text": entry.strip()
+            })
+            st.success("已儲存 🌟")
         else:
-            st.warning("Please write something first")
-
-# 🤖 AI Therapist
-with tab4:
-    st.header("🤖 AI Therapist")
-    st.write("*Your 24/7 companion*")
-
-    st.markdown("""
-    <div style="background: #F3F4F6; padding: 20px; border-radius: 15px; margin: 20px 0;">
-        <div style="display: flex; align-items: center;">
-            <div style="font-size: 3rem; margin-right: 15px;">🤖</div>
-            <div>
-                <h4 style="margin: 0; color: #6366F1;">SoulCompass AI Guide</h4>
-                <p style="margin: 5px 0; color: #6B7280;">Here to listen and support you</p>
-            </div>
-        </div>
-    </div>
-    """, unsafe_allow_html=True)
-
-    topic = st.selectbox(
-        "What would you like to talk about?",
-        ["General Support", "Stress & Anxiety", "Relationships", "Life Goals", "Personal Growth"]
-    )
+            st.warning("請先寫點內容")
 
-    message = st.text_area(
-        "Share your thoughts",
-        placeholder="Tell me what's on your mind...",
-        height=100
-    )
+    if st.session_state.journal:
+        st.markdown("#### 最近的紀錄")
+        for j in reversed(st.session_state.journal[-10:]):
+            st.markdown(f"**{j['time']} · {j['emotion']} · 心情 {j['mood']}**")
+            st.markdown(f"<div class='card' style='margin:.4rem 0'>{j['text']}</div>", unsafe_allow_html=True)
 
-    if message and st.button("💫 Get Support"):
-        with st.spinner("🤖 Thinking..."):
-            import time
-            time.sleep(1)
-
-            responses = [
-                f"Thank you for sharing about {topic.lower()}. I hear that you're saying: '{message}'. Your feelings are completely valid, and it's brave of you to reach out.",
-                f"I understand you're dealing with {topic.lower()}. What you've shared - '{message}' - shows real self-awareness. You're taking positive steps by talking about it.",
-                f"Regarding {topic.lower()}, I want you to know that '{message}' resonates with me. Remember, you have inner strength and wisdom to navigate this."
-            ]
-
-            response = random.choice(responses)
-
-            st.markdown(f"""
-            <div style="margin: 20px 0;">
-                <div style="background: #E5E7EB; padding: 15px; border-radius: 15px; margin-bottom: 10px;">
-                    <strong>You:</strong> {message}
-                </div>
-                <div style="background: #6366F1; color: white; padding: 15px; border-radius: 15px;">
-                    <strong>🤖 AI Guide:</strong> {response}
-                    <br><br>
-                    <small>💙 I'm here to support you. For serious concerns, please reach out to a professional.</small>
-                </div>
-            </div>
-            """, unsafe_allow_html=True)
-
-# 頁腳
-st.markdown("---")
-st.markdown("""
-<div style="text-align: center; color: #6B7280; padding: 2rem;">
-    <p>🧭 <strong>SoulCompass</strong> - Find Your Inner North Star</p>
-    <p>Made with ❤️ for spiritual seekers worldwide</p>
-    <p>💎 <a href="#" style="color: #6366F1;">Upgrade to Pro</a> | 📧 [email protected] | 🌐 soulcompass.ai</p>
-</div>
-""", unsafe_allow_html=True)
-# app.py 內適當位置（例如新增一個 Tab 或在抽牌後呼叫）
-import rag_utils
-
-with st.expander("🔎 Tarot / Numerology 知識查詢（RAG）"):
-    q = st.text_input("輸入關鍵字（例如：new beginnings, career change, love advice ...）")
-    target = st.radio("資料庫", ["Tarot", "Numerology"], horizontal=True)
-    k = st.slider("Top K", 1, 5, 3)
-
-    if q:
-        if target == "Tarot":
-            hits = rag_utils.search_tarot(q, k=k)
+# ---- TAB 4: AI Guide ----
+with tab4:
+    st.subheader("🤖 AI Guide")
+    st.write("*Your 24/7 companion (not a substitute for professional help)*")
+    topic = st.selectbox("主題", ["General Support", "Stress & Anxiety", "Relationships", "Life Goals", "Personal Growth"])
+    msg = st.text_area("想說什麼？", height=110, placeholder="把心裡話告訴我…")
+    if st.button("💫 取得支持"):
+        if not msg.strip():
+            st.warning("請先輸入訊息")
         else:
-            hits = rag_utils.search_numerology(q, k=k)
-
-        for h in hits:
-            st.markdown(f"**#{h['rank']} • score {h['score']:.3f}**")
-            if 'card_name' in h:
-                st.markdown(f"🃏 **{h['card_name']}**")
-            if 'number' in h:
-                st.markdown(f"🔢 **Number {h['number']}**")
-            st.write(h['text'])
-            st.markdown("---")
+            seed_q = f"主題：{topic}；訊息：{msg}。請以溫柔、務實、具行動建議的方式回應。"
+            with st.spinner("AI 思考中…"):
+                ans, _ = rag_answer(seed_q, domain="Tarot", k=2)  # 混搭少量塔羅語料，提升靈性語氣
+            st.markdown(f'<div class="result">{ans.replace(chr(10), "<br>")}</div>', unsafe_allow_html=True)
+            st.caption("若遇到嚴重困擾，請尋求專業協助。")
+
+# ========= Footer =========
+st.markdown("<hr>", unsafe_allow_html=True)
+st.markdown(
+    "<div style='text-align:center' class='small'>🧭 <b>SoulCompass</b> · Made with ❤️ · RAG + LLM powered</div>",
+    unsafe_allow_html=True
+)