Update app.py

app.py

@@ -11,6 +11,7 @@ from dotenv import load_dotenv
 
 # --- 1. 環境變數 & Playwright 初始化 ---
 load_dotenv()
+# 強制設定 Playwright 瀏覽器存放路徑
 os.environ["PLAYWRIGHT_BROWSERS_PATH"] = "/home/user/.cache/ms-playwright"
 
 try:
@@ -27,18 +28,17 @@ from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
 from openai import OpenAI
 
-# 假設這是您組員開發的模組
+# 嘗試匯入組員的模組
 try:
     import emotion 
 except ImportError:
-    # 防止因缺少 emotion 模組而無法執行
-    print("警告：找不到 emotion 模組，請確保該檔案存在。")
+    print("警告：找不到 emotion 模組，請確保同級目錄下有 emotion.py。")
 
 # ==========================================
 # 0. 全域設定
 # ==========================================
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-GOOGLE_MAPS_API_KEY = os.getenv("GOOGLE_MAPS_API_KEY") # 請確保 .env 中有此項
+GOOGLE_MAPS_API_KEY = os.getenv("GOOGLE_MAPS_API_KEY")
 
 global_df = None
 global_mood_df = None
@@ -58,7 +58,7 @@ def init_static_data():
             })
 
 # ==========================================
-# 1. Google Maps 爬蟲同步功能 (僅爬取基本資料)
+# 1. Google Maps 爬蟲功能
 # ==========================================
 def sync_google_maps(url):
     global global_df, global_retriever, rag_initialized
@@ -74,11 +74,11 @@ def sync_google_maps(url):
             page = browser.new_page()
             yield "🌐 [2/4] 連線中..."
             page.goto(clean_url, wait_until="domcontentloaded", timeout=60000)
-            yield "⏳ [3/4] 等待加載資料..."
+            yield "⏳ [3/4] 等待列表加載..."
             time.sleep(10) 
             yield "📄 [4/4] 解析餐廳資訊..."
             
-            # 獲取標題與詳情
+            # 抓取餐廳名稱與基本資訊
             titles = page.locator('div.fontHeadlineSmall').all_inner_texts()
             details = page.locator('div.fontBodyMedium').all_inner_texts()
             
@@ -86,7 +86,6 @@ def sync_google_maps(url):
             for i, name in enumerate(titles):
                 name = name.strip()
                 if name:
-                    # 擷取地址或評價資訊
                     addr = details[i].strip() if i < len(details) else ""
                     restaurant_list.append({
                         "Name": name,
@@ -97,7 +96,7 @@ def sync_google_maps(url):
             browser.close()
 
             if not restaurant_list:
-                yield "⚠️ 找不到餐廳。請確認連結已設為「共用」且為列表格式。"
+                yield "⚠️ 找不到餐廳。請確認連結格式正確且已公開。"
                 return
 
             global_df = pd.DataFrame(restaurant_list)
@@ -110,15 +109,14 @@ def sync_google_maps(url):
         yield f"❌ 系統錯誤：{str(e)}"
 
 # ==========================================
-# 2. 推薦邏輯與 Embed API 生成
+# 2. 推薦邏輯與 Embed API 嵌入
 # ==========================================
 def get_restaurant_data(mood_score_str, food_choice):
     init_static_data()
     if global_df is None or global_df.empty: 
         return None, True, "資料庫未載入", "尚未同步地圖資料", ""
     
-    all_restaurant_names = "、".join(global_df['Name'].tolist())
-    
+    all_names = "、".join(global_df['Name'].tolist())
     try: score = int(str(mood_score_str).split(' ')[0])
     except: score = 3
     
@@ -137,62 +135,55 @@ def get_restaurant_data(mood_score_str, food_choice):
             is_match = False
     
     result = candidates.sample(1).iloc[0]
-    return result, is_match, "個人清單", mood_reason, all_restaurant_names
+    return result, is_match, "個人清單", mood_reason, all_names
 
 def generate_content_with_groq(restaurant_name, restaurant_detail, user_diary, mood_score, mood_guide_reason):
-    if not GROQ_API_KEY: return "⚠️ 請設定 GROQ_API_KEY"
+    if not GROQ_API_KEY: return "⚠️ 請在 .env 設定 GROQ_API_KEY"
     client = OpenAI(api_key=GROQ_API_KEY, base_url="https://api.groq.com/openai/v1")
-    system_prompt = "你是一個幽默、懂吃且善解人意的 AI 朋友，會根據對方的心情推薦合適的餐廳。請用溫暖且具備食記風格的口吻來描述。"
-    user_msg = f"【狀態】心情分數：{mood_score}\n【日記】{user_diary}\n【建議原因】{mood_guide_reason}\n【推薦餐廳】{restaurant_name}\n餐廳資料：{restaurant_detail}"
+    system_prompt = "你是一個幽默且暖心的美食達人，會根據心情日記推薦餐廳。"
+    user_msg = f"心情分數：{mood_score}\n日記：{user_diary}\n餐廳：{restaurant_name}\n資訊：{restaurant_detail}"
     try:
         response = client.chat.completions.create(model="llama-3.3-70b-versatile", messages=[{"role": "system", "content": system_prompt}, {"role": "user", "content": user_msg}])
         return response.choices[0].message.content
-    except Exception as e: return f"Groq Error: {str(e)}"
+    except Exception as e: return f"AI 生成失敗: {str(e)}"
 
-def mood_agent_logic(score_input, food_input, diary_input, debug_mode):
+def mood_agent_logic(score_input, food_input, diary_input):
     if not rag_initialized:
-        yield "⚠️ 請先點擊上方按鈕同步地圖清單！", "", ""
+        yield "⚠️ 請先同步地圖清單！", "", ""
         return
 
     restaurant, is_match, _, mood_reason, all_names = get_restaurant_data(score_input, food_input)
     
     prefix = ""
     if not is_match and food_input != "隨便":
-        prefix = f"> 💡 **溫馨提示**：清單中暫無『{food_input}』，已從名單「{all_names}」中挑選最適合您今日心情的一間！\n\n"
+        prefix = f"> 💡 **溫馨提示**：清單中暫無『{food_input}』，已從現有名單挑選最適合的店！\n\n"
 
     name = restaurant['Name']
     addr = restaurant.get('Address', '')
     url = restaurant.get('URL', '#')
-    rag_info = str(restaurant.get('RAG_Content', ''))
     
-    # 產生 AI 文案
-    ai_text = generate_content_with_groq(name, rag_info, diary_input, score_input, mood_reason)
+    ai_text = generate_content_with_groq(name, restaurant['RAG_Content'], diary_input, score_input, mood_reason)
     final_markdown = f"{prefix}### 🍽️ 今日推薦：{name}\n\n{ai_text}"
     
-    # --- 關鍵：使用 Google Maps Embed API 替代圖片 ---
-    # query 參數包含 名稱 與 地址 以提高準確度
+    # Google Maps Embed API HTML
     query_str = urllib.parse.quote(f"{name} {addr}")
     if GOOGLE_MAPS_API_KEY:
         embed_html = f"""
-        <div style="border-radius:15px; overflow:hidden; border:1px solid #ddd;">
-            <iframe
-              width="100%"
-              height="350"
-              frameborder="0" style="border:0"
+        <div style="border-radius:12px; overflow:hidden; border:1px solid #ddd;">
+            <iframe width="100%" height="350" frameborder="0" style="border:0"
               src="https://www.google.com/maps/embed/v1/place?key={GOOGLE_MAPS_API_KEY}&q={query_str}"
-              allowfullscreen>
-            </iframe>
+              allowfullscreen></iframe>
         </div>
         """
     else:
-        embed_html = "<p style='color:red;'>⚠️ 尚未設定 GOOGLE_MAPS_API_KEY，無法顯示地圖預覽。</p>"
+        embed_html = "<p style='color:red;'>⚠️ 缺少 GOOGLE_MAPS_API_KEY</p>"
 
-    nav_button = f'<div style="text-align:center; margin-top:10px;"><a href="{url}" target="_blank" style="background:#4CAF50;color:white;padding:10px 20px;border-radius:25px;text-decoration:none;font-weight:bold;">📍 前往 Google Maps 導航</a></div>'
+    nav_button = f'<div style="text-align:center; margin-top:10px;"><a href="{url}" target="_blank" style="background:#4CAF50;color:white;padding:10px 20px;border-radius:25px;text-decoration:none;font-weight:bold;">📍 Google Maps 導航</a></div>'
     
     yield final_markdown, embed_html, nav_button
 
 # ==========================================
-# 3. 橋接情緒辨識邏輯 (保持原有功能)
+# 3. 橋接情緒辨識 (依賴 emotion.py)
 # ==========================================
 def _score_to_radio_value(score):
     mapping = {1: "1 (心情差)", 2: "2 (不太好)", 3: "3 (普通)", 4: "4 (不錯)", 5: "5 (超棒)"}
@@ -228,48 +219,49 @@ def bridge_predict_upload(img, st):
     return out_result, score_update, tabs_update, out_st
 
 # ==========================================
-# 4. Gradio 介面
+# 4. Gradio UI 介面
 # ==========================================
 with gr.Blocks(title="AI 心情食堂", css=getattr(emotion, 'css', '')) as demo:
     st_state = gr.State(emotion.AppState())
 
     with gr.Tabs() as tabs:
-        # Tab 1: 情緒辨識
+        # Step 1
         with gr.TabItem("😊 第一步：測測心情", id=0):
             with gr.Column():
-                gr.Markdown("### 📸 透過鏡頭或照片辨識您的當下情緒")
+                gr.Markdown("### 📸 臉部情緒辨識")
                 with gr.Row():
                     btn_start = gr.Button("開啟攝影機", variant="primary")
                     btn_stop  = gr.Button("停止", variant="secondary", visible=False)
                 cam = gr.Image(sources=["webcam"], streaming=True, type="numpy", label="攝影機畫面", visible=False)
-                result_markdown = gr.Markdown("請按上方按鈕開始辨識")
-                upload_img = gr.Image(sources=["upload"], type="numpy", label="或是上傳照片辨識")
+                result_markdown = gr.Markdown("請按上方按鈕開始")
+                upload_img = gr.Image(sources=["upload"], type="numpy", label="上傳照片辨識")
 
-        # Tab 2: 主功能區
-        with gr.TabItem("🍽️ 第二步：心情食堂推薦", id=1):
+        # Step 2
+        with gr.TabItem("🍽️ 第二步：心情食堂", id=1):
             with gr.Column():
                 gr.Markdown("### 🔗 同步 Google Maps 分享清單")
                 with gr.Row():
-                    map_url = gr.Textbox(label="清單連結", placeholder="https://maps.app.goo.gl/...")
-                    sync_btn = gr.Button("🔄 同步我的口袋名單", variant="secondary")
-                sync_msg = gr.Markdown("ℹ️ 尚未載入資料庫")
+                    map_url = gr.Textbox(label="清單連結", placeholder="貼上 Google Maps 分享連結")
+                    sync_btn = gr.Button("🔄 同步清單", variant="secondary")
+                sync_msg = gr.Markdown("ℹ️ 尚未同步")
                 
-                gr.Divider()
+                # 修正處：gr.Divider() 改為 gr.Separator()
+                gr.Separator()
                 
                 with gr.Row():
                     with gr.Column(scale=1):
                         score_input = gr.Radio(
                             ["1 (心情差)", "2 (不太好)", "3 (普通)", "4 (不錯)", "5 (超棒)"], 
-                            label="心情分數 (辨識後自動填寫)", 
+                            label="心情分數", 
                             value="3 (普通)"
                         )
                         food_input = gr.Radio(["吃飯", "吃麵", "隨便"], label="想吃什麼？", value="隨便")
-                        diary_input = gr.Textbox(lines=3, label="心情日記 (選填)", placeholder="寫下現在的想法...")
-                        submit_btn = gr.Button("🍱 推薦今日驚喜", variant="primary")
+                        diary_input = gr.Textbox(lines=3, label="心情日記", placeholder="今天發生了什麼事？")
+                        submit_btn = gr.Button("🍱 推薦美食", variant="primary")
                     
                     with gr.Column(scale=1):
-                        agent_output = gr.Markdown(label="AI 推薦語")
-                        map_embed_output = gr.HTML(label="地圖實景預覽") 
+                        agent_output = gr.Markdown(label="AI 建議")
+                        map_embed_output = gr.HTML(label="地圖預覽") 
                         map_nav_output = gr.HTML()
 
     # 事件綁定
@@ -278,7 +270,7 @@ with gr.Blocks(title="AI 心情食堂", css=getattr(emotion, 'css', '')) as demo
     btn_stop.click(fn=bridge_stop_click, inputs=[st_state], outputs=[result_markdown, cam, btn_stop, btn_start, st_state])
     cam.stream(fn=bridge_predict_frame, inputs=[cam, st_state], outputs=[cam, result_markdown, st_state, btn_stop, btn_start, score_input, tabs])
     upload_img.change(fn=bridge_predict_upload, inputs=[upload_img, st_state], outputs=[result_markdown, score_input, tabs, st_state])
-    submit_btn.click(fn=mood_agent_logic, inputs=[score_input, food_input, diary_input, gr.Checkbox(value=False, visible=False)], outputs=[agent_output, map_embed_output, map_nav_output])
+    submit_btn.click(fn=mood_agent_logic, inputs=[score_input, food_input, diary_input], outputs=[agent_output, map_embed_output, map_nav_output])
 
 if __name__ == "__main__":
     demo.launch()