Update app.py

app.py

@@ -1,8 +1,7 @@
 import os, re, base64
 from langchain_core.documents import Document
 from langchain_chroma import Chroma
-from openai import OpenAI
-from langchain.embeddings.base import Embeddings
+from langchain_huggingface import HuggingFaceEmbeddings   # ✅ 雲端可直接使用
 from langchain_google_genai import ChatGoogleGenerativeAI
 import chromadb
 import gradio as gr
@@ -18,21 +17,9 @@ except ImportError:
 
 
 # =============================================
-# 1️⃣ 自訂 LM Studio Embedding 類別
+# 1️⃣ 使用 Hugging Face 雲端 embedding 模型
 # =============================================
-class LmStudioEmbeddings(Embeddings):
-    def __init__(self, model_name, url):
-        self.model_name = model_name
-        self.client = OpenAI(base_url=url, api_key="lm-studio")
-
-    def embed_query(self, text: str):
-        res = self.client.embeddings.create(input=text, model=self.model_name)
-        return res.data[0].embedding
-
-    def embed_documents(self, texts: list[str]):
-        res = self.client.embeddings.create(input=texts, model=self.model_name)
-        return [x.embedding for x in res.data]
-
+embedding = HuggingFaceEmbeddings(model_name="BAAI/bge-small-zh-v1.5")
 
 # =============================================
 # 2️⃣ 載入 QA 檔案並分類
@@ -65,13 +52,9 @@ for k, v in qa_docs.items():
 # =============================================
 # 3️⃣ 建立向量資料庫
 # =============================================
-embedding = LmStudioEmbeddings(
-    model_name="text-embedding-bge-large-zh-v1.5",
-    url="http://127.0.0.1:1234/v1"
-)
 client = chromadb.PersistentClient(path="./chroma_db")
-
 collection_names = {"證券": "stocks", "期貨": "futures", "複委託": "overseas"}
+
 vectordbs = {}
 for cat, docs in qa_docs.items():
     vectordbs[cat] = Chroma(
@@ -85,7 +68,7 @@ print("✅ 各類別向量資料庫建立完成")
 
 
 # =============================================
-# 4️⃣ 初始化 Gemini LLM
+# 4️⃣ 初始化 Gemini LLM（雲端）
 # =============================================
 API_KEY = os.getenv("GOOGLE_API_KEY")
 if not API_KEY:
@@ -132,7 +115,7 @@ def chat_fn(message, history):
 
 
 # =============================================
-# 6️⃣ 介面（含飛出動畫 + 純白footer）
+# 6️⃣ Gradio 介面（LINE風格 + 小輸入按鈕 + 純白footer）
 # =============================================
 logo_path = os.path.join(BASE_DIR, "mega.png")
 logo_base64 = ""
@@ -167,22 +150,16 @@ with gr.Blocks(
 
         /* ====== footer（純白背景） ====== */
         #footer {
-            position: fixed;
-            bottom: 40px;
-            left: 0;
-            width: 100%;
-            text-align: center;
-            font-size: 13px;
-            color: #aaa;                  /* ✅ 更淡的灰色 */
-            border-top: 1px solid #ddd;   /* ✅ 細灰分隔線 */
-            padding-top: 8px;
-            background-color: transparent; /* ✅ 移除反白 */
+            position: fixed; bottom: 40px; left: 0; width: 100%;
+            text-align: center; font-size: 13px; color: #aaa;
+            border-top: 1px solid #ddd; padding-top: 8px;
+            background-color: transparent;
         }
         @media (max-width: 768px) {
             #footer { position: relative; margin-top: 40px; }
         }
 
-        /* ====== LINE 風格輸入區 ====== */
+        /* ====== LINE風格輸入區 ====== */
         #input-row { display: flex; align-items: center; gap: 8px; margin-top: 10px; }
         #user-input {
             flex-grow: 1; border-radius: 20px; border: 1px solid #ccc;
@@ -190,56 +167,40 @@ with gr.Blocks(
             box-shadow: inset 0 0 1px rgba(0,0,0,0.05);
         }
 
-        /* 🟢 小巧箭頭按鈕（含飛出動畫） */
+        /* 🟢 小巧文字版「輸入」按鈕 */
         #send-btn {
-            background-color: #00b800; border: none; border-radius: 50%;
-            width: 28px; height: 28px; cursor: pointer;
-            display: flex; align-items: center; justify-content: center;
+            background-color: #00b800;
+            color: white;
+            border: none;
+            border-radius: 14px;
+            height: 26px;
+            padding: 0 10px;
+            font-size: 13px;
+            font-weight: 600;
+            cursor: pointer;
             transition: background-color 0.2s ease, transform 0.1s ease;
-            box-shadow: 0 1px 2px rgba(0,0,0,0.1); padding: 0; overflow: hidden;
-        }
-        #send-btn svg {
-            width: 12px; height: 12px; fill: white;
-            transition: transform 0.25s ease;
+            box-shadow: 0 1px 2px rgba(0,0,0,0.1);
         }
         #send-btn:hover { background-color: #00a000; }
-        #send-btn:hover svg { transform: rotate(10deg) scale(1.15); }
-        #send-btn:active { transform: scale(0.9); }
-        #send-btn:active svg { animation: send-fly 0.5s ease-out; }
-
-        @keyframes send-fly {
-            0% { transform: translateX(0) scale(1); opacity: 1; }
-            50% { transform: translateX(8px) scale(1.2); opacity: 0; }
-            100% { transform: translateX(0) scale(1); opacity: 1; }
-        }
+        #send-btn:active { transform: scale(0.95); }
     """
 ) as demo:
-    # 左上角 logo
     if logo_base64:
-        gr.HTML(f"""
-            <div id="logo-top"><img src="data:image/png;base64,{logo_base64}" alt="logo"></div>
-        """)
+        gr.HTML(f"<div id='logo-top'><img src='data:image/png;base64,{logo_base64}' alt='logo'></div>")
 
-    # 標題區
     gr.HTML("""
-        <div id="main-title-wrapper" style="text-align:center; margin-top:20px;">
+        <div id='main-title-wrapper' style='text-align:center; margin-top:20px;'>
             <h1 id='main-title'><span>👨‍💼 我是小智&nbsp;&nbsp;您的金融好幫手 🫰</span></h1>
             <p id='sub-title' style='margin-top:10px; font-size:14px; color:#666;'>Powered by Gemini & LangChain</p>
         </div>
     """)
 
-    # 聊天介面
     with gr.Row():
         with gr.Column(scale=4):
             chatbox = gr.Chatbot(label="💬 對話紀錄", type="messages")
-
             with gr.Row(elem_id="input-row"):
                 user_input = gr.Textbox(elem_id="user-input", show_label=False, placeholder="輸入訊息...", scale=8)
-                send_btn = gr.Button(
-                    value="""
-                    <svg viewBox="0 0 24 24"><path d="M3 12l18-9-6 9 6 9z"/></svg>
-                    """, elem_id="send-btn", scale=1
-                )
+                send_btn = gr.Button("輸入", elem_id="send-btn", scale=1)
 
             def handle_input(message, history):
                 if not message.strip():
@@ -272,10 +233,8 @@ with gr.Blocks(
                 return [], gr.update(value="", placeholder="輸入訊息...")
             gr.Button("🧹 整理畫面").click(clear_memory, outputs=[chatbox, user_input])
 
-    # 底部 footer（純白）
     gr.HTML("<div id='footer'>© Fintech Assistant — 僅業務使用，非官方授權</div>")
 
-    # 手機鍵盤彈出自動捲動
     demo.load(None, None, None, js="""
         window.addEventListener('focusin', () => {
             document.querySelector('textarea')?.scrollIntoView({ behavior: 'smooth', block: 'center' });