Update app.py

app.py

@@ -86,27 +86,53 @@ def sync_google_maps(url):
             page = browser.new_page()
             yield "🌐 [2/4] 連線中..."
             page.goto(clean_url, wait_until="domcontentloaded", timeout=60000)
-            yield "⏳ [3/4] 等待加載 (10秒)..."
+            yield "⏳ [3/4] 等待列表完整加載 (10秒)..."
             time.sleep(10) 
-            yield "📄 [4/4] 解析中..."
-            titles = page.locator('div.fontHeadlineSmall').all_inner_texts()
-            details = page.locator('div.fontBodyMedium').all_inner_texts()
+            
+            yield "📄 [4/4] 解析餐廳資訊與圖片..."
+            
+            # 改用容器式爬取，確保資料對齊
+            # Google Maps 列表項目通常在 [role="article"] 或特定的 div 內
+            # 這裡使用更通用的定位方式找尋包含名稱的區塊
+            containers = page.locator('div.m67Hec').all() 
             
             restaurant_list = []
-            for i, name in enumerate(titles):
-                name = name.strip()
-                if name:
-                    addr = details[i].strip() if i < len(details) else "詳情見地圖"
+            for item in containers:
+                try:
+                    # 抓取名稱
+                    name_el = item.locator('div.fontHeadlineSmall')
+                    if name_el.count() == 0: continue
+                    name = name_el.inner_text().strip()
+                    
+                    # 抓取地址/描述
+                    addr_el = item.locator('div.fontBodyMedium')
+                    addr = addr_el.inner_text().strip() if addr_el.count() > 0 else "詳情見地圖"
+                    
+                    # 抓取圖片連結 (抓取 img 標籤的 src)
+                    img_el = item.locator('img')
+                    img_url = None
+                    if img_el.count() > 0:
+                        img_url = img_el.get_attribute('src')
+                    
+                    # 如果沒抓到圖片，給一個預設的餐廳占位圖
+                    if not img_url or "http" not in img_url:
+                        img_url = "https://images.unsplash.com/photo-1517248135467-4c7edcad34c4?w=500"
+
                     restaurant_list.append({
                         "Name": name,
                         "Address": addr,
+                        "Image_URL": img_url, # 儲存圖片網址
                         "RAG_Content": f"餐廳：{name}，地址：{addr}",
                         "URL": f"https://www.google.com/search?q={urllib.parse.quote(name)}"
                     })
+                except Exception as e:
+                    print(f"解析單筆資料出錯: {e}")
+                    continue
+
             browser.close()
 
             if not restaurant_list:
-                yield "⚠️ 找不到餐廳。請確認連結已設為「共用」。"
+                yield "⚠️ 找不到餐廳。請確認連結已設為「共用」且頁面上有顯示列表。"
                 return
 
             global_df = pd.DataFrame(restaurant_list)
@@ -114,7 +140,7 @@ def sync_google_maps(url):
             vectorstore = FAISS.from_texts(global_df['RAG_Content'].tolist(), embeddings, metadatas=global_df.to_dict('records'))
             global_retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
             rag_initialized = True
-            yield f"✅ 同步成功！共有 {len(restaurant_list)} 間餐廳。"
+            yield f"✅ 同步成功！共有 {len(restaurant_list)} 間餐廳（已含圖片）。"
     except Exception as e:
         yield f"❌ 系統錯誤：{str(e)}"
 
@@ -175,24 +201,28 @@ def mood_agent_logic(score_input, food_input, diary_input, debug_mode):
         yield "⚠️ 請先同步地圖清單！", None, ""
         return
 
+    # 從資料庫抽樣一間餐廳
     restaurant, is_match, rec_categories, mood_reason, all_names = get_restaurant_data(score_input, food_input)
     
-    # 增加告知訊息
     prefix = ""
     status_hint = ""
     if not is_match and food_input != "隨便":
         prefix = f"> 💡 **溫馨提示**：您的清單目前只有：**{all_names}**，沒找到符合『{food_input}』的店。我先幫您從現有名單挑一間最讚的！\n\n"
         status_hint = f"\n(注意：清單中無 '{food_input}'，請幽默解釋為何先推薦這間。)"
 
-    name = restaurant['Name']; url = restaurant.get('URL', '#')
+    name = restaurant['Name']
+    url = restaurant.get('URL', '#')
+    img_url = restaurant.get('Image_URL') # 取得我們剛抓到的圖片網址
     rag_info = str(restaurant.get('RAG_Content', '')) + status_hint
     
+    # 呼叫 Groq 生成文字
     ai_text = generate_content_with_groq(name, rag_info, diary_input, score_input, mood_reason)
     
     final_response = f"{prefix}### 🍽️ 推薦：{name}\n\n{ai_text}"
     map_html = f'<div style="text-align:center"><a href="{url}" target="_blank" style="background:#4CAF50;color:white;padding:8px 16px;border-radius:20px;text-decoration:none">🗺️ 導航</a></div>'
     
-    yield final_response, None, map_html
+    # 修改點：這裡的第二個回傳值改為 img_url，Gradio 會自動加載該網址圖片
+    yield final_response, img_url, map_html
 
 # ==========================================
 # 3. 橋接邏輯 (保留原本情緒辨識部分)