Update app.py

app.py

@@ -1,74 +1,33 @@
 # app.py
-import os
-import json
-import random
+import os, json, uuid, 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"
-)
-
-# ========= 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 = []
+# ========== 基本設定 ==========
+st.set_page_config(page_title="SoulCompass", page_icon="🧭", layout="wide")
 
-# ========= CSS (simple) =========
-st.markdown("""
-<style>
-.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)
+# 使用者暫時 UID（未登入先用本機隨機 ID）
+if "uid" not in st.session_state:
+    st.session_state.uid = str(uuid.uuid4())[:8]
+UID = st.session_state.uid
+USER_DIR = os.path.join("data", "users", UID)
+os.makedirs(USER_DIR, exist_ok=True)
+JOURNAL_PATH = os.path.join(USER_DIR, "journal.jsonl")
 
-# ========= Header =========
 st.markdown("""
-<div class="header">
-  <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 style="text-align:center;background:linear-gradient(135deg,#6366F1,#C084FC);color:white;padding:1.4rem;border-radius:14px;margin-bottom:12px;">
+  <h2 style="margin:0;">🧭 SoulCompass — Tarot • Numerology • Journal • Guide</h2>
 </div>
 """, unsafe_allow_html=True)
 
-# ========= 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`")
-# ========= Data Loaders (Tarot / Numerology) =========
+# ========== 讀取 Tarot / Numerology 資料 ==========
 @st.cache_resource
 def load_tarot_names():
-    """從 data/tarot_data_full.json 讀牌名（支援 name 或 card_name）。"""
     path = os.path.join("data", "tarot_data_full.json")
     with open(path, "r", encoding="utf-8") as f:
         data = json.load(f)
     names = []
-    for i, c in enumerate(data):
+    for c in data:
         nm = c.get("name") or c.get("card_name")
         if nm:
             names.append(nm)
@@ -78,36 +37,47 @@ def load_tarot_names():
 
 @st.cache_resource
 def load_numerology_data():
-    """讀取 data/numerology_data_full.json，回傳 (numbers_list, by_number_dict)。"""
     path = os.path.join("data", "numerology_data_full.json")
     with open(path, "r", encoding="utf-8") as f:
         data = json.load(f)
-    numbers, by_num = [], {}
-    for row in data:
-        num = row.get("number")
-        if num is None: 
+    by_num, nums = {}, []
+    for r in data:
+        if "number" not in r: 
             continue
-        numbers.append(int(num))
-        by_num[int(num)] = {
-            "name": row.get("name", ""),
-            "keywords": row.get("keywords", []),
-            "description": row.get("description", ""),
-            "advice": row.get("advice", "")
+        n = int(r["number"])
+        nums.append(n)
+        by_num[n] = {
+            "name": r.get("name",""),
+            "keywords": r.get("keywords",[]),
+            "description": r.get("description",""),
+            "advice": r.get("advice","")
         }
-    numbers = sorted(set(numbers))
-    if not numbers:
+    nums = sorted(set(nums))
+    if not nums:
         raise ValueError("No numerology numbers found in data/numerology_data_full.json")
-    return numbers, by_num
+    return nums, by_num
 
 def life_path_number_from_date(dt):
-    """把生日轉成生命靈數（保留 11/22/33）。"""
     s = dt.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))
     return total
 
-# ========= RAG + LLM Loader =========
+TAROT_NAMES = load_tarot_names()
+NUMBERS, NUM_MAP = load_numerology_data()
+
+# ========== Sidebar：模型與 RAG 參數 ==========
+st.sidebar.header("⚙️ Settings")
+model_choice = st.sidebar.selectbox(
+    "LLM Model",
+    ["cerebras/btlm-3b-8k-base", "mistralai/Mistral-7B-Instruct-v0.2"],
+    help="BTLM 輕量、CPU 可跑；Mistral 口條更好但較吃資源。"
+)
+rag_k = st.sidebar.slider("RAG Top-K", 1, 5, 3)
+temperature = st.sidebar.slider("Temperature", 0.2, 1.2, 0.7, 0.1)
+
+# ========== RAG + LLM Loader ==========
 @st.cache_resource
 def load_rag():
     try:
@@ -118,42 +88,37 @@ def load_rag():
         return None
 
 @st.cache_resource
-def load_llm(model_id: str = "cerebras/btlm-3b-8k-base"):
+def load_llm(model_id: str):
     import torch
     from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-
     tok = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
     model = AutoModelForCausalLM.from_pretrained(
         model_id,
         trust_remote_code=True,
-        torch_dtype=torch.float32,        # 免費 CPU 用 float32 最穩
+        torch_dtype=torch.float32,     # 免費 CPU 最穩
         device_map="auto",
         low_cpu_mem_usage=True,
-        offload_folder="/tmp",            # ★ 提供 offload 路徑
+        offload_folder="/tmp",
     )
-    pipe = pipeline(
+    return pipeline(
         "text-generation",
         model=model,
         tokenizer=tok,
-        max_new_tokens=320,
-        temperature=0.7,
+        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.
@@ -168,157 +133,223 @@ USER QUESTION: {query}
     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")
+# ========== Journal 長期保存（JSONL）==========
+def load_journal(path=JOURNAL_PATH):
+    entries = []
+    if os.path.exists(path):
+        with open(path, "r", encoding="utf-8") as f:
+            for line in f:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    entries.append(json.loads(line))
+                except Exception:
+                    pass
+    return entries
+
+def append_journal(entry: dict, path=JOURNAL_PATH):
+    os.makedirs(os.path.dirname(path), exist_ok=True)
+    with open(path, "a", encoding="utf-8") as f:
+        f.write(json.dumps(entry, ensure_ascii=False) + "\n")
+
+def save_all_journal(entries, path=JOURNAL_PATH):
+    os.makedirs(os.path.dirname(path), exist_ok=True)
+    with open(path, "w", encoding="utf-8") as f:
+        for e in entries:
+            f.write(json.dumps(e, ensure_ascii=False) + "\n")
+
+def export_journal_csv(path=JOURNAL_PATH):
     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]
+        return None
+    rows = load_journal(path)
+    out_csv = os.path.join(USER_DIR, "journal_export.csv")
+    import csv
+    with open(out_csv, "w", encoding="utf-8", newline="") as f:
+        w = csv.writer(f)
+        w.writerow(["time", "mood", "emotion", "text"])
+        for r in rows:
+            w.writerow([r.get("time",""), r.get("mood",""), r.get("emotion",""), r.get("text","")])
+    return out_csv
 
-TAROT_NAMES = load_tarot_names()
+if "journal" not in st.session_state:
+    st.session_state.journal = load_journal()
+
+# ========== AI 心靈導師（安全護欄＋生成）==========
+CRISIS_KEYWORDS = [
+    "自殺","想死","結束生命","活不下去","傷害自己","割腕",
+    "殺了我","想不開","無法活","輕生","自殘","不如死了"
+]
+
+def is_crisis(text: str) -> bool:
+    if not text:
+        return False
+    # 中文大小寫無差；直接檢索原文關鍵字
+    return any(k in text for k in CRISIS_KEYWORDS)
+
+def guide_answer(user_msg: str, topic: str = "General"):
+    if is_crisis(user_msg):
+        crisis_reply = (
+            "我聽見你現在非常難受，謝謝你願意說出來。你的感受很重要，也值得被好好照顧。\n\n"
+            "目前這些想法可能代表你需要更即時、面對面的支持。請立刻聯絡你所在地的緊急服務，"
+            "或尋求可信任的人陪伴（家人、朋友、學校輔導/公司 EAP）。\n\n"
+            "你也可以尋找在地的心理諮商與危機支援資源（醫院身心科、社福單位、24 小時關懷專線）。\n\n"
+            "現在先做三件小事：1) 找一個安全的空間，2) 打給一位你信任的人，3) 緩慢做 5 次深呼吸（吸4、停2、吐6）。\n\n"
+            "我在這裡，願意繼續聽你說。"
+        )
+        return crisis_reply, []
+
+    # 可選：混入少量 RAG 提示（用 Numerology 當中性語料）
+    try:
+        ctx_hits = rag.search_numerology("自我照顧 情緒 調節 建議", k=2) if rag else []
+        context = "\n\n".join([h["text"] for h in ctx_hits]) if ctx_hits else "（無檢索）"
+    except Exception:
+        ctx_hits, context = [], "（無檢索）"
+
+    prompt = f"""You are SoulCompass, a supportive, non-clinical companion.
+- Respond in Traditional Chinese.
+- Be empathetic, specific, and action-oriented.
+- Do NOT diagnose conditions or offer medical advice.
+- Offer 2~4 small, doable next steps, tailored to the user's message.
+- Keep it under 10 sentences.
+
+OPTIONAL CONTEXT (self-care ideas):
+{context}
 
-# ========= Tabs =========
-tab1, tab2, tab3, tab4 = st.tabs(["🔮 Tarot", "🔢 Numerology", "📖 Journal", "🤖 AI Guide"])
+TOPIC: {topic}
+USER: {user_msg}
 
-# ---- TAB 1: Tarot ----
+請用溫柔、務實、去評判的語氣回覆，先同理、再聚焦、最後給 2~4 個可行的小步驟。"""
+    out = llm(prompt)[0]["generated_text"]
+    return out[len(prompt):].strip(), ctx_hits
+
+# ========== Tabs ==========
+tab1, tab2, tab3, tab4 = st.tabs(["🔮 Tarot", "🔢 Numerology", "📖 Journal", "🤖 AI 心靈導師"])
+
+# ---- Tarot ----
 with tab1:
     st.subheader("🔮 Tarot Reading")
-    st.write("*Let ancient wisdom guide your path*")
-
-    colA, colB = st.columns([2, 1])
-    with colA:
-        question = st.text_area(
-            "你想得到什麼指引？",
-            placeholder="例如：我適合換工作嗎？和他未來的關係會如何？",
-            height=90
-        )
-    with colB:
-        spread = st.selectbox(
-            "抽牌陣列",
-            ["Single (1 card)", "Three (Past • Present • Future)", "Five (Deep Insight)"]
-        )
-
-    draw = st.button("🔮 抽牌")
-    if draw and not question:
-        st.warning("請先輸入你的問題")
-    if draw and question:
-        if st.session_state.daily_usage >= 5:
-            st.error("今日免費占卜次數已滿，明天再來吧！")
+    q = st.text_input("想詢問什麼？（例：我適合換工作嗎？）", key="tarot_q")
+    spread = st.radio("抽牌陣列", ["Single (1)", "Three (3)", "Five (5)"], horizontal=True, key="tarot_spread")
+    if st.button("抽牌並解讀", key="tarot_go"):
+        if not q.strip():
+            st.warning("請先輸入問題")
         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.subheader("🔢 Numerology")
-    st.write("*Discover your numbers*")
-
-    col1, col2 = st.columns(2)
-    with col1:
-        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("參考段落"):
+            n = 1 if "1" in spread else (3 if "3" in spread else 5)
+            cards = random.sample(TAROT_NAMES, k=min(n, len(TAROT_NAMES)))
+            st.success("你抽到：" + ", ".join(cards))
+            query = f"問題：{q}；抽到的牌：{', '.join(cards)}。請用塔羅牌義做整體解讀。"
+            with st.spinner("RAG 檢索 + 生成中…"):
+                ans, refs = rag_answer(query, domain="Tarot", k=rag_k)
+            st.markdown(ans)
+            if refs:
+                with st.expander("參考段落（RAG 來源）"):
                     for i, r in enumerate(refs, 1):
-                        tag = r.get("number") or "N/A"
-                        st.markdown(f"**#{i} Number {tag}** · score: {r['score']:.3f}")
+                        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("---")
-            else:
-                st.warning("請先選擇生日")
 
+# ---- Numerology ----
+with tab2:
+    st.subheader("🔢 Numerology")
+    b = st.date_input("你的生日", key="dob")
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button("計算生命靈數", key="lp_go"):
+            if b:
+                lp = life_path_number_from_date(b)
+                st.success(f"你的生命靈數：{lp}（{NUM_MAP.get(lp,{}).get('name','')}）")
+                qn = f"生命靈數 {lp} 的核心特質、盲點與行動建議？"
+                with st.spinner("RAG 檢索 + 生成中…"):
+                    ans, refs = rag_answer(qn, domain="Numerology", k=rag_k)
+                st.markdown(ans)
     with col2:
-        name = st.text_input("你的全名（可選）")
-        if st.button("簡易名字數字"):
+        name = st.text_input("你的姓名（可選）", key="name_num")
+        if st.button("名字數字", key="name_go"):
             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.info(f"名字數字：{n}（{NUM_MAP.get(n,{}).get('name','')}）")
+                qn = f"名字數字 {n} 的人在人際與自我表達上的建議？"
+                ans, _ = rag_answer(qn, domain="Numerology", k=rag_k)
                 st.write(ans)
-            else:
-                st.warning("請輸入名字")
 
-# ---- TAB 3: Journal ----
+# ---- Journal ----
 with tab3:
     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("請先寫點內容")
+    st.caption(f"本機使用者 ID：{UID}")
+    mood = st.slider("今天的心情（1-10）", 1, 10, 7, key="mood")
+    emotion = st.selectbox("主要情緒", ["😊 開心","😔 失落","😰 焦慮","😌 平靜","🤗 感恩"], key="emotion")
+    entry = st.text_area("寫下今天的想法", height=140, placeholder="你最近在思考什麼？今天最感謝的一件事是？", key="entry")
+
+    c1, c2, c3 = st.columns(3)
+    with c1:
+        if st.button("💾 儲存日記", key="save_journal"):
+            if entry.strip():
+                item = {
+                    "time": datetime.now().strftime("%Y-%m-%d %H:%M"),
+                    "mood": st.session_state.mood,
+                    "emotion": st.session_state.emotion,
+                    "text": entry.strip()
+                }
+                st.session_state.journal.append(item)
+                append_journal(item)
+                st.success("已儲存 🌟")
+            else:
+                st.warning("請先寫點內容")
+
+    with c2:
+        if st.button("⬇️ 匯出 CSV", key="export_journal"):
+            out_csv = export_journal_csv()
+            if out_csv and os.path.exists(out_csv):
+                with open(out_csv, "rb") as f:
+                    st.download_button("下載日記 CSV", f, file_name="journal_export.csv")
+            else:
+                st.warning("目前沒有日記可匯出")
 
-    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)
+    with c3:
+        if st.button("🗑️ 清空全部日記", key="clear_journal"):
+            save_all_journal([])
+            st.session_state.journal = []
+            st.success("全部日記已清空")
 
-# ---- TAB 4: AI Guide ----
+    if st.session_state.journal:
+        st.markdown("---")
+        st.markdown("**最近紀錄（新→舊）**")
+        for j in reversed(st.session_state.journal[-50:]):
+            st.markdown(f"- **{j['time']} · {j['emotion']} · 心情 {j['mood']}**  \n{j['text']}")
+
+        if st.button("🧠 生成 7 日回顧建議", key="review7"):
+            recent = "\n".join([j["text"] for j in st.session_state.journal[-20:]])
+            qj = f"這是我最近的日記重點：\n{recent}\n請用溫柔務實的語氣，總結 3 點觀察，並給 3 個可行的小建議。"
+            ans, _ = rag_answer(qj, domain="Numerology", k=2)  # 用較中性的語料
+            st.markdown(ans)
+
+# ---- AI 心靈導師 ----
 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("請先輸入訊息")
+    st.subheader("🤖 AI 心靈導師")
+    st.caption("此功能提供一般性支持與自我照顧建議，非醫療或心理治療之替代。若有危機，請立即尋求在地緊急支援。")
+    topic = st.selectbox(
+        "主題",
+        ["General", "Stress & Anxiety", "Relationships", "Work & Burnout", "Self-esteem", "Grief & Loss"],
+        index=0,
+        key="guide_topic"
+    )
+    user_msg = st.text_area("想跟我說點什麼？", height=140, placeholder="把現在的感受、困擾或情境說給我聽…", key="guide_msg")
+    if st.button("💬 取得支持", key="guide_go"):
+        if not user_msg.strip():
+            st.warning("請先輸入一段想說的話")
         else:
-            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("若遇到嚴重困擾，請尋求專業協助。")
+            with st.spinner("AI 正在整理回應…"):
+                reply, refs = guide_answer(user_msg, topic)
+            st.markdown(reply)
+            if refs:
+                with st.expander("參考段落（RAG 來源）"):
+                    for i, r in enumerate(refs, 1):
+                        st.markdown(f"**#{i}** · score {r['score']:.3f}")
+                        st.write(r["text"])
+                        st.markdown("---")
 
-# ========= Footer =========
+# ========== 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
-)
+st.caption("🧭 SoulCompass · demo MVP · RAG + LLM powered")