Update app.py

app.py

@@ -1,12 +1,12 @@
 """
 ======================================================
 📘 金融客服小智（Fintech Assistant）
-版本：v2 (重構示範 by Supervisor)
+版本：v2.1 (Hugging Face 部署版)
 改進重點：
-1. 模組化程式結構（易維護）
-2. 加入記憶體保存（多輪對話）
-3. 改善 Chroma 初始化與 QA 擷取
-4. 加強異常處理與容錯提示
+1. 改用記憶體型 Chroma，避免 PersistentClient 錯誤
+2. 路徑使用 os.getcwd() 以符合 HF Spaces
+3. 加入 QA 檔案容錯與模擬模式
+4. GOOGLE_API_KEY 以 Secrets 管理
 ======================================================
 """
 
@@ -33,26 +33,21 @@ except ImportError:
 # =============================================
 embedding = HuggingFaceEmbeddings(model_name="BAAI/bge-small-zh-v1.5")
 
-BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+BASE_DIR = os.getcwd()
 QA_PATH = os.path.join(BASE_DIR, "QA_v2.txt")
 LOGO_PATH = os.path.join(BASE_DIR, "mega.png")
 
-if not os.path.exists(QA_PATH):
-    raise FileNotFoundError("❌ 找不到 QA 檔案 QA_v2.txt，請確認路徑。")
-
 API_KEY = os.getenv("GOOGLE_API_KEY")
 if not API_KEY:
-    print("⚠️ 尚未設定 GOOGLE_API_KEY，系統將以模擬回覆運行。")
+    print("⚠️ 尚未設定 GOOGLE_API_KEY，將使用模擬模式。")
 
 
 # =============================================
-# 2️⃣ QA 載入與分類（改進版正規化）
+# 2️⃣ QA 載入與分類
 # =============================================
 def load_qa_documents(path: str):
     with open(path, "r", encoding="utf-8") as f:
         text = f.read()
-
-    # 改進版正規表達式，確保每筆 QA 含問題與答案
     pattern = r"(Ｑ[:：].*?Ａ[:：].*?)(?=Ｑ[:：]|$)"
     qas = re.findall(pattern, text, flags=re.S)
 
@@ -66,40 +61,44 @@ def load_qa_documents(path: str):
         elif "複委託" in qa:
             categories["複委託"].append(doc)
         else:
-            categories["證券"].append(doc)  # 預設分類
-
+            categories["證券"].append(doc)
     return categories
 
 
-qa_docs = load_qa_documents(QA_PATH)
-print("✅ 已成功載入 QA 檔案，共分為：", {k: len(v) for k, v in qa_docs.items()})
+if os.path.exists(QA_PATH):
+    qa_docs = load_qa_documents(QA_PATH)
+    print("✅ 已載入 QA 檔案，共分為：", {k: len(v) for k, v in qa_docs.items()})
+else:
+    print("⚠️ 未找到 QA_v2.txt，啟用空白知識庫模式。")
+    qa_docs = {"證券": [], "期貨": [], "複委託": []}
 
 
 # =============================================
-# 3️⃣ 向量資料庫初始化（避免重複寫入）
+# 3️⃣ 向量資料庫初始化（記憶體型）
 # =============================================
-client = chromadb.PersistentClient(path="./chroma_db")
+try:
+    client = chromadb.Client()
+except Exception:
+    import chromadb.api
+    client = chromadb.api.Client()
+
 collection_map = {"證券": "stocks", "期貨": "futures", "複委託": "overseas"}
 vectordbs = {}
 
 for cat, docs in qa_docs.items():
-    vectordb = Chroma(
-        client=client,
-        collection_name=collection_map[cat],
-        embedding_function=embedding
-    )
-    if vectordb._collection.count() == 0:
+    vectordb = Chroma(client=client, collection_name=collection_map[cat], embedding_function=embedding)
+    if hasattr(vectordb._collection, "count") and vectordb._collection.count() == 0 and docs:
         vectordb.add_documents(docs)
     vectordbs[cat] = vectordb
 
-print("✅ 向量資料庫已建立完成。")
+print("✅ 向量資料庫初始化完成。")
 
 
 # =============================================
-# 4️⃣ 初始化 LLM 與對話記憶
+# 4️⃣ 初始化 LLM 與記憶體
 # =============================================
 if API_KEY:
-    llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", google_api_key=API_KEY)
+    llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=API_KEY)
 else:
     llm = None  # 模擬模式
 
@@ -110,19 +109,16 @@ memory = ConversationBufferMemory(memory_key="chat_history", return_messages=Tru
 # 5️⃣ 對話邏輯
 # =============================================
 def auto_detect_category(text: str):
-    """根據關鍵詞自動偵測使用者詢問的業務類別"""
     if any(k in text for k in ["股票", "證券", "開戶", "下單", "交割"]):
         return "證券"
     elif any(k in text for k in ["期貨", "選擇權", "保證金"]):
         return "期貨"
     elif any(k in text for k in ["複委託", "海外", "美股", "港股"]):
         return "複委託"
-    else:
-        return "證券"
+    return "證券"
 
 
 def chat_fn(message, history):
-    """核心對話函式"""
     category = auto_detect_category(message)
     vectordb = vectordbs[category]
     docs = vectordb.similarity_search(message, k=2)
@@ -141,18 +137,17 @@ def chat_fn(message, history):
             response = llm.invoke(prompt)
             reply = getattr(response, "content", None) or getattr(response, "text", "⚠️ 無回覆")
         else:
-            reply = "（模擬模式���這是示範回覆：請確認是否已設定 GOOGLE_API_KEY。"
+            reply = "（模擬模式）這是示範回覆，請確認已設定 GOOGLE_API_KEY。"
     except Exception as e:
         reply = f"⚠️ 生成錯誤：{e}"
 
-    # 保存對話記憶
     memory.save_context({"role": "user", "content": message},
                         {"role": "assistant", "content": reply})
     return reply
 
 
 # =============================================
-# 6️⃣ Gradio 介面（重構版）
+# 6️⃣ Gradio 介面
 # =============================================
 logo_base64 = ""
 if os.path.exists(LOGO_PATH):
@@ -169,7 +164,6 @@ with gr.Blocks(
             pointer-events: none;
         }
         #logo-top img { width: 120px; height: auto; display: block; }
-
         #footer { text-align:center; font-size:13px; color:#aaa; margin-top: 20px; }
     """
 ) as demo:
@@ -182,10 +176,8 @@ with gr.Blocks(
     with gr.Row():
         with gr.Column(scale=4):
             chatbot = gr.Chatbot(label="💬 對話紀錄", type="messages", height=500)
-
-            with gr.Row():
-                user_input = gr.Textbox(placeholder="請輸入問題...", show_label=False)
-                send_btn = gr.Button("送出", variant="primary")
+            user_input = gr.Textbox(placeholder="請輸入問題...", show_label=False)
+            send_btn = gr.Button("送出", variant="primary")
 
             def handle_input(message, history):
                 if not message.strip():
@@ -198,10 +190,7 @@ with gr.Blocks(
             user_input.submit(handle_input, [user_input, chatbot], [chatbot, user_input])
             send_btn.click(handle_input, [user_input, chatbot], [chatbot, user_input])
 
-            def clear_all():
-                memory.clear()
-                return [], gr.update(value="")
-            gr.Button("🧹 清除對話").click(clear_all, outputs=[chatbot, user_input])
+            gr.Button("🧹 清除對話").click(lambda: ([], gr.update(value="")), outputs=[chatbot, user_input])
 
         with gr.Column(scale=1):
             gr.Markdown("### 🔍 常見問題")
@@ -218,4 +207,4 @@ with gr.Blocks(
 
     gr.HTML("<div id='footer'>© Fintech Assistant — 僅業務使用，非官方授權</div>")
 
-demo.launch()
+demo.launch()