Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import requests
 import google.generativeai as genai
 from flask import Flask, request, abort
 from linebot import LineBotApi, WebhookHandler
@@ -8,6 +9,8 @@ import json
 from PIL import Image
 from linebot.models import MessageEvent, TextMessage, TextSendMessage, QuickReply, QuickReplyButton, MessageAction, ImageMessage, FlexSendMessage, LocationMessage, LocationAction
 import urllib.parse
+import math
+import urllib.parse
 
 app = Flask(__name__)
 
@@ -232,31 +235,164 @@ def handle_image(event):
             )
 
     # ================= 【新增】專門處理使用者傳送位置的邏輯 =================
+# ================= 【升級版】全台通用的真實附近店家搜尋 =================
+# ================= 【終極版】客製化品牌 + 營業時間 + Flex 卡片 =================
 @handler.add(MessageEvent, message=LocationMessage)
 def handle_location(event):
-    # 這裡可以取得使用者真實的經緯度 (lat, lon)
-    # 未來如果妳想串接 Google Maps API 計算真實距離，就是用這兩個數字喔！
-    lat = event.message.latitude
-    lon = event.message.longitude
+    user_id = event.source.user_id
+    user_lat = event.message.latitude
+    user_lon = event.message.longitude
     
-    # 這是妳設計的完美回覆台詞
-    reply_text = """🧚‍♀️ 找到離你最近的藥妝店：
-
-1️⃣ 康是美 台北車站門市
-📍距離 350 公尺
-
-2️⃣ 屈臣氏 站前店
-📍距離 500 公尺
-
-3️⃣ 寶雅 台北站前店
-📍距離 800 公尺"""
-
-    # 直接回傳給使用者
-    line_bot_api.reply_message(
-        event.reply_token,
-        TextSendMessage(text=reply_text)
-    )
+    # 取出剛剛使用者選擇的品牌 (如果沒有紀錄，預設用全都要)
+    state = user_states.get(user_id, {})
+    brand_preference = state.get("search_brand", "全都要")
+    
+    google_api_key = os.environ.get('GOOGLE_MAPS_API_KEY')
+    if not google_api_key:
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="❌ 系統尚未設定 Google Maps API 金鑰。"))
+        return
 
+    # 計算距離的數學魔法
+    def calculate_distance(lat1, lon1, lat2, lon2):
+        R = 6371.0
+        lat1_rad, lon1_rad = math.radians(lat1), math.radians(lon1)
+        lat2_rad, lon2_rad = math.radians(lat2), math.radians(lon2)
+        dlon = lon2_rad - lon1_rad
+        dlat = lat2_rad - lat1_rad
+        a = math.sin(dlat / 2)**2 + math.cos(lat1_rad) * math.cos(lat2_rad) * math.sin(dlon / 2)**2
+        c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a))
+        return int(R * c * 1000)
+
+    # 決定呼叫 Google 的關鍵字
+    search_keyword = "藥妝店" if brand_preference == "全都要" else brand_preference
+    search_radius = 2000 # 稍微擴大到 2 公里，確保「全都要」時能找到足夠的店
+    
+    url = f"https://maps.googleapis.com/maps/api/place/textsearch/json?query={search_keyword}&location={user_lat},{user_lon}&radius={search_radius}&language=zh-TW&key={google_api_key}"
+    
+    try:
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="🧚‍♀️ 精靈正在為您繪製專屬地圖卡片，請稍候..."))
+        
+        response = requests.get(url)
+        data = response.json()
+        
+        if data['status'] != 'OK' or not data.get('results'):
+             line_bot_api.push_message(user_id, TextSendMessage(text="🥺 哎呀，精靈在您附近找不到符合的藥妝店呢..."))
+             if user_id in user_states: del user_states[user_id]
+             return
+             
+        # 整理資料並加上「營業時間」判斷
+        stores = []
+        for place in data['results']:
+            store_lat = place['geometry']['location']['lat']
+            store_lon = place['geometry']['location']['lng']
+            distance = calculate_distance(user_lat, user_lon, store_lat, store_lon)
+            
+            # 判斷營業狀態 (open_now)
+            open_now = place.get('opening_hours', {}).get('open_now')
+            if open_now is True:
+                hours_text = "🟢 營業中"
+            elif open_now is False:
+                hours_text = "🔴 休息中"
+            else:
+                hours_text = "🕒 營業時間需確認"
+
+            stores.append({
+                "name": place.get('name', '未知名稱'),
+                "rating": place.get('rating', '無評分'),
+                "lat": store_lat,
+                "lon": store_lon,
+                "distance": distance,
+                "address": place.get('formatted_address', ''),
+                "hours": hours_text # 將營業狀態存入
+            })
+            
+        # 依照距離由近到遠排序
+        sorted_stores = sorted(stores, key=lambda x: x['distance'])
+        
+        # 🌟 核心邏輯：依照使用者選擇來過濾店家！
+        nearest_stores = []
+        
+        if brand_preference == "全都要":
+            # 如果全都要，精靈會逐一尋找各品牌的第一名
+            found_w = found_c = found_p = False
+            for s in sorted_stores:
+                if "屈臣氏" in s['name'] and not found_w:
+                    nearest_stores.append(s)
+                    found_w = True
+                elif "康是美" in s['name'] and not found_c:
+                    nearest_stores.append(s)
+                    found_c = True
+                elif "寶雅" in s['name'] and not found_p:
+                    nearest_stores.append(s)
+                    found_p = True
+                # 如果三家都找到了就提早結束尋找
+                if found_w and found_c and found_p:
+                    break
+        else:
+            # 如果有指定品牌，就直接拿該品牌距離最近的前 3 名
+            nearest_stores = sorted_stores[:3]
+        
+        # 準備製作 Flex Message 的泡泡卡片
+        bubbles = []
+        for store in nearest_stores:
+            map_query = urllib.parse.quote(f"{store['name']} {store['address']}")
+            map_url = f"https://www.google.com/maps/search/?api=1&query={map_query}"
+            
+            # 組裝單張卡片 (加入營業時間欄位)
+            card = {
+                "type": "bubble",
+                "size": "kilo",
+                "body": {
+                    "type": "box",
+                    "layout": "vertical",
+                    "spacing": "sm",
+                    "contents": [
+                        {"type": "text", "text": f"🏪 {store['name']}", "weight": "bold", "size": "md", "wrap": True, "color": "#111111"},
+                        {"type": "text", "text": f"📍 距離 {store['distance']} 公尺", "size": "sm", "color": "#8c8c8c"},
+                        {"type": "text", "text": store['hours'], "size": "sm", "color": "#8c8c8c"}, # 🌟 這裡新增了營業時間！
+                        {"type": "text", "text": f"⭐ 評分: {store['rating']}", "size": "sm", "color": "#fbbc04", "weight": "bold"}
+                    ]
+                },
+                "footer": {
+                    "type": "box",
+                    "layout": "vertical",
+                    "spacing": "sm",
+                    "contents": [
+                        {
+                            "type": "button",
+                            "style": "primary",
+                            "color": "#00B900",
+                            "height": "sm",
+                            "action": {
+                                "type": "uri",
+                                "label": "地圖導航",
+                                "uri": map_url
+                            }
+                        }
+                    ]
+                }
+            }
+            bubbles.append(card)
+
+        flex_message = FlexSendMessage.new_from_json_dict({
+            "type": "flex",
+            "alt_text": "📍 附近的藥妝店來囉！",
+            "contents": {
+                "type": "carousel",
+                "contents": bubbles
+            }
+        })
+        
+        line_bot_api.push_message(user_id, flex_message)
+        
+        # 任務完成，清除狀態，避免卡住
+        if user_id in user_states:
+            del user_states[user_id]
+        
+    except Exception as e:
+        line_bot_api.push_message(user_id, TextSendMessage(text=f"❌ 搜尋過程發生錯誤：{e}"))
+        if user_id in user_states:
+            del user_states[user_id]
 
 # ================= 3. 多輪對話核心邏輯 =================
 @handler.add(MessageEvent, message=TextMessage)
@@ -264,21 +400,48 @@ def handle_message(event):
     user_id = event.source.user_id
     user_msg = event.message.text.strip()
 
+    # ================= 【升級】附近店家 - 步驟 1：選擇品牌 =================
     if user_msg == "附近店家":
-        # 建立一個要求使用者分享位置的按鈕
-        location_button = QuickReply(items=[
-            QuickReplyButton(action=LocationAction(label="📍 點我分享位置"))
+        user_states[user_id] = {"step": "ASK_BRAND"} # 記錄用戶進入找店流程
+        
+        brand_buttons = QuickReply(items=[
+            QuickReplyButton(action=MessageAction(label="屈臣氏", text="屈臣氏")),
+            QuickReplyButton(action=MessageAction(label="康是美", text="康是美")),
+            QuickReplyButton(action=MessageAction(label="寶雅", text="寶雅")),
+            QuickReplyButton(action=MessageAction(label="全都要", text="全都要"))
         ])
         
         line_bot_api.reply_message(
             event.reply_token,
             TextSendMessage(
-                text="📍請分享您的位置，我來幫您尋找附近的藥妝店！", 
-                quick_reply=location_button
+                text="請問你想尋找哪一間藥妝店呢？", 
+                quick_reply=brand_buttons
             )
         )
         return
 
+    # ================= 【升級】附近店家 - 步驟 2：要求定位 =================
+    state = user_states.get(user_id)
+    if state and state.get("step") == "ASK_BRAND":
+        if user_msg in ["屈臣氏", "康是美", "寶雅", "全都要"]:
+            state["search_brand"] = user_msg # 把使用者選的品牌存起來
+            state["step"] = "ASK_LOCATION"   # 推進到下一步
+            
+            location_button = QuickReply(items=[
+                QuickReplyButton(action=LocationAction(label="📍 點我分享位置"))
+            ])
+            
+            line_bot_api.reply_message(
+                event.reply_token,
+                TextSendMessage(
+                    text=f"沒問題！請分享您的位置，精靈來幫您找最近的【{user_msg}】！", 
+                    quick_reply=location_button
+                )
+            )
+        else:
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="請點擊下方的按鈕選擇品牌喔！"))
+        return
+
     # 任何時候輸入「精靈顧問」或「重新開始」，皆重置該用戶的狀態
     if user_msg == "精靈顧問" or user_msg == "重新開始":
         user_states[user_id] = {"step": "ASK_SKIN"}