Update app.py

app.py

@@ -16,6 +16,20 @@ handler = WebhookHandler(os.environ.get('LINE_CHANNEL_SECRET'))
 genai.configure(api_key=os.environ.get('GEMINI_API_KEY'))
 model = genai.GenerativeModel('gemini-2.5-flash')
 
+# 建立我們的虛擬藥妝資料庫 (Mock Database)
+fake_database = {
+    "CeraVe適樂膚 長效潤澤修護霜 340g": [
+        {"store": "屈臣氏", "name": "CeraVe適樂膚 長效潤澤修護霜 340g", "price": "619","link": "https://www.watsons.com.tw/cerave-cerave%E9%81%A9%E6%A8%82%E8%86%9A%E9%95%B7%E6%95%88%E6%BD%A4%E6%BE%A4%E4%BF%AE%E8%AD%B7%E9%9C%9C-340g/p/BP_153671"},
+        {"store": "康是美", "name": "CeraVe適樂膚 長效潤澤修護霜 340g", "price": "619", "link": "https://shop.cosmed.com.tw/SalePage/Index/6414249"},
+        {"store": "寶雅", "name": "CeraVe適樂膚 長效潤澤修護霜 340g", "price": "619","link": "https://www.poyabuy.com.tw/SalePage/Index/10104261"}
+    ],
+    "Avene 雅漾舒護活泉水150ml": [
+        {"store": "屈臣氏", "name": "Avene 雅漾舒護活泉水150ml", "price": "299","discount": "🔥限時67折", "link": "https://www.watsons.com.tw/avene-%E9%9B%85%E6%BC%BE-%E9%9B%85%E6%BC%BE%E8%88%92%E8%AD%B7%E6%B4%BB%E6%B3%89%E6%B0%B4150ml/p/BP_291419"},
+        {"store": "康是美", "name": "Avene 雅漾舒護活泉水150ml", "price": "299","discount": "🔥限時67折", "link": "https://shop.cosmed.com.tw/SalePage/Index/5767522"},
+        {"store": "寶雅", "name": "Avene 雅漾舒護活泉水150ml", "price": "450", "link": "https://www.poyabuy.com.tw/SalePage/Index/9929185"}
+    ]
+}
+
 # ================= 2. 建立狀態儲存區（暫存每個用戶的進度） =================
 user_states = {}
 
@@ -51,12 +65,16 @@ def handle_image(event):
             
             # 2. 設定 Prompt，要求 Gemini 直接分析這張圖片
             vision_prompt = """
+            請辨識這張圖片中的保養品。
+            如果是 CeraVe 的 PM 乳液，請直接回答：CeraVe長效清爽保濕乳PM
+            如果是 理膚寶水 B5，請直接回答：理膚寶水B5全面修復霜
+            如果都不是
             你現在是一位專業的台灣藥妝購物精靈。請看這張圖片中的產品：
             1. 辨識它是什麼真實的品牌與品名。
             2. 估算它在台灣藥妝店(如屈臣氏、康是美)的合理售價。
             3. 指出哪個商家買得到。
             
-            請直接輸出以下格式的純文字（不需要包含其他廢話）：
+            並請直接輸出以下格式的純文字（不需要包含其他廢話）：
             品名：[產品完整名稱]
             售價：$[估計價格]
             商家：[推薦藥妝店名稱]
@@ -64,21 +82,133 @@ def handle_image(event):
             
             # 3. 把圖片和 Prompt 一起丟給 Gemini (支援多模態)
             response = model.generate_content([vision_prompt, pil_image])
-            ai_reply = response.text
+            product_name = response.text.strip()
             
-            # 4. 將結果推播給使用者（目前先用文字，後續可升級成圖卡）
-            line_bot_api.push_message(
-                user_id,
-                TextSendMessage(text=f"🧚‍♀️ 辨識成功！\n\n{ai_reply}")
-            )
+            # --- 第 3 步：雙軌制 (假資料庫 vs AI估價) ---
+            if product_name in fake_database:
+                # 🌟 路線 A：命中資料庫，使用妳設定的真實數據！
+                data_list = fake_database[product_name]
+                price_tag = "資料庫真實價"  # 讓圖卡顯示這是真實數據
+            else:
+                # 🤖 路線 B：資料庫沒有，呼叫 AI 進行現場估價！
+                line_bot_api.push_message(user_id, TextSendMessage(text=f"🔍 辨識出「{product_name}」，資料庫無建檔，啟動 AI 現場估價..."))
+                
+                estimate_prompt = f"""
+                請估算「{product_name}」這款產品在台灣屈臣氏、康是美、寶雅的合理售價。
+                請務必「只回傳」一個標準的 JSON 陣列，不要有任何額外的說明文字。格式嚴格如下：
+                [
+                  {{"store": "屈臣氏", "name": "{product_name}", "price": "估計數字"}},
+                  {{"store": "康是美", "name": "{product_name}", "price": "估計數字"}},
+                  {{"store": "寶雅", "name": "{product_name}", "price": "估計數字"}}
+                ]
+                """
+                est_response = model.generate_content(estimate_prompt)
+                clean_text = est_response.text.replace("```json", "").replace("```", "").strip()
+                data_list = json.loads(clean_text)
+                price_tag = "AI 估計價"  # 讓圖卡顯示這是 AI 猜的
+            
+            # --- 第 4 步：將資料塞進橫向圖卡 (不管哪條路線，最後都會來到這裡) ---
+            # --- 第 4 步：將資料塞進橫向圖卡 ---
+            bubbles = []
+            for item in data_list:
+                
+                # --- 這裡用來判斷要不要顯示折扣標籤 ---
+                discount_box = []
+                # 只有當資料庫裡的 discount 不是空字串時，才生出這塊紅色的標籤
+                if item.get("discount"): 
+                    discount_box = [
+                        {
+                            "type": "box",
+                            "layout": "baseline",
+                            "spacing": "sm",
+                            "contents": [
+                                {
+                                    "type": "text",
+                                    "text": item["discount"],
+                                    "color": "#ffffff",
+                                    "size": "xs",
+                                    "weight": "bold",
+                                    "margin": "sm"
+                                }
+                            ],
+                            "backgroundColor": "#ff3b30", # 紅色底
+                            "cornerRadius": "md",
+                            "paddingAll": "2px",
+                            "paddingStart": "8px",
+                            "paddingEnd": "8px",
+                            "margin": "sm"
+                        }
+                    ]
+
+                # 這是原本卡片的身體內容，我們把剛剛可能生出來的折扣標籤塞進去
+                card_contents = [
+                    {"type": "text", "text": item["store"], "weight": "bold", "color": "#1DB446", "size": "md"},
+                    {"type": "text", "text": item["name"], "weight": "bold", "size": "lg", "margin": "md", "wrap": True},
+                ]
+                
+                # 把折扣標籤接在商品名稱後面
+                card_contents.extend(discount_box) 
+                
+                # 補上最後的價格線
+                card_contents.extend([
+                    {"type": "separator", "margin": "lg"},
+                    {
+                        "type": "box",
+                        "layout": "horizontal",
+                        "margin": "md",
+                        "contents": [
+                            {"type": "text", "text": price_tag, "size": "sm", "color": "#555555"},
+                            {"type": "text", "text": f"${item['price']}", "size": "lg", "color": "#111111", "align": "end", "weight": "bold"}
+                        ]
+                    }
+                ])
+
+                # 組裝最終的卡片
+                card = {
+                  "type": "bubble",
+                  "body": {
+                    "type": "box",
+                    "layout": "vertical",
+                    "contents": card_contents
+                  },
+                  "footer": {
+                    "type": "box",
+                    "layout": "vertical",
+                    "contents": [
+                      {
+                        "type": "button",
+                        "style": "primary",
+                        "color": "#1DB446",
+                        "height": "sm",
+                        "action": {
+                          "type": "uri",
+                          "label": "前往商家購買",
+                          # 這裡改用資料庫裡設定的專屬網址！(如果沒設定，就預設用 Google 搜尋)
+                          "uri": item.get("link", f"https://www.google.com/search?q={item['store']}+{item['name']}")
+                        }
+                      }
+                    ]
+                  }
+                }
+                bubbles.append(card)
+            
+            # --- 第 5 步：打包發送圖卡 ---
+            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)
             del user_states[user_id]
             
         except Exception as e:
             line_bot_api.push_message(
                 user_id,
-                TextSendMessage(text=f"❌ 哎呀！精靈的眼睛剛剛眨了一下，沒看清楚照片。錯誤原因：\n{e}")
+                TextSendMessage(text=f"❌ 哎呀！精靈遇到了一點問題。錯誤原因：\n{e}")
             )
 
 # ================= 3. 多輪對話核心邏輯 =================