Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
-from datetime import datetime, timedelta
 import os
 import requests
 from supabase import create_client, Client
+from datetime import datetime, timedelta, timezone
+
+# 設定台北時區
+TAIPEI_TZ = timezone(timedelta(hours=8))
 
 # --- 1. 連線設定 ---
 # 請確認 HF Secrets 裡都有設定這些變數
@@ -20,7 +23,8 @@ else:
 # --- 2. 輔助函式 (日期與時間) ---
 def get_date_options():
     options = []
-    today = datetime.now()
+    # 這裡一定要加時區，不然會抓到 UTC 時間
+    today = datetime.now(TAIPEI_TZ)
     weekdays = ["(一)", "(二)", "(三)", "(四)", "(五)", "(六)", "(日)"]
     for i in range(30): 
         current_date = today + timedelta(days=i)
@@ -56,7 +60,7 @@ def send_line_notify(data):
     if not LINE_ACCESS_TOKEN or not LINE_ADMIN_ID: 
         return
     
-    # 簡單的文字通知 (您也可以換回之前的 Flex Message)
+    # 簡單的文字通知
     message = (
         f"🔥 新訂位通知 🔥\n"
         f"姓名：{data['name']}\n"
@@ -87,6 +91,23 @@ def handle_booking(name, tel, email, date_str, time, pax, remarks):
     if not name or not tel or not date_str or not time:
         return "⚠️ 請完整填寫必填欄位 (姓名、電話、日期、時間)"
     
+    # --- 1. 防呆機制：檢查是否重複提交 ---
+    try:
+        # 搜尋該電話在該時段的訂單
+        existing = supabase.table("bookings").select("id")\
+            .eq("tel", tel)\
+            .eq("date", date_str)\
+            .eq("time", time)\
+            .neq("status", "顧客已取消")\
+            .execute()
+            
+        if existing.data and len(existing.data) > 0:
+            return "⚠️ 系統偵測到您已預約過此時段，請勿重複提交。"
+    except Exception as e:
+        print(f"Check duplicate error: {e}")
+        pass
+
+    # --- 準備資料 ---
     data = {
         "name": name, 
         "tel": tel, 
@@ -99,17 +120,21 @@ def handle_booking(name, tel, email, date_str, time, pax, remarks):
     }
     
     try:
-        # 1. 寫入資料庫
+        # 2. 寫入資料庫
         supabase.table("bookings").insert(data).execute()
         
-        # 2. 發送 LINE
+        # 3. 發送 LINE 通知
         send_line_notify(data)
         
+        # 4. 回傳成功訊息
         return """
         <div style='text-align: center; color: #fff; padding: 20px; border: 1px solid #d4af37; border-radius: 8px; background: #222;'>
             <h2 style='color: #d4af37; margin: 0;'>Request Received</h2>
             <p style='margin: 10px 0;'>🥂 預約申請已提交</p>
-            <p style='font-size: 0.9em; color: #aaa;'>請留意您的 Email 確認信。</p>
+            <p style='font-size: 0.9em; color: #aaa;'>系統將發送確認信至您的 Email。</p>
+            <p style='font-size: 0.85em; color: #ff5252; margin-top: 5px;'>
+                (若未收到，請檢查您的<b>垃圾信件匣</b>)
+            </p>
         </div>
         """
     except Exception as e:
@@ -117,21 +142,22 @@ def handle_booking(name, tel, email, date_str, time, pax, remarks):
         return f"❌ 系統錯誤: {str(e)}"
 
 # --- 5. 核心邏輯：處理確認連結 (Webhook) ---
+# ⚠️ 修正重點：這裡是 check_confirmation，不是 send_confirmation_email
 def check_confirmation(request: gr.Request):
     """
-    當網址帶有 ?id=xx&action=confirm 時觸發
+    當網址帶有 ?id=xx&action=confirm 或 ?id=xx&action=cancel 時觸發
     """
     if not request: return ""
     params = request.query_params
     action = params.get('action')
     bid = params.get('id')
     
+    # --- 情況 A: 確認訂位 ---
     if action == 'confirm' and bid:
         try:
             # 更新資料庫狀態
             supabase.table("bookings").update({"status": "顧客已確認"}).eq("id", bid).execute()
             
-            # 回傳給前端顯示的確認訊息
             return f"""
             <script>
                 document.addEventListener('DOMContentLoaded', function() {{
@@ -145,6 +171,27 @@ def check_confirmation(request: gr.Request):
         except Exception as e:
             print(f"Confirm Error: {e}")
             return ""
+
+    # --- 情況 B: 取消訂位 ---
+    elif action == 'cancel' and bid:
+        try:
+            # 更新資料庫狀態
+            supabase.table("bookings").update({"status": "顧客已取消"}).eq("id", bid).execute()
+            
+            return f"""
+            <script>
+                document.addEventListener('DOMContentLoaded', function() {{
+                    alert('已為您取消訂位 (編號 {bid})');
+                }});
+            </script>
+            <div style='padding:20px; background:#333; border: 1px solid #ff5252; color:#ff5252; text-align:center; margin-bottom:20px; border-radius:8px;'>
+                🚫 訂位已取消。<br>期待您下次有機會再來訪。
+            </div>
+            """
+        except Exception as e:
+            print(f"Cancel Error: {e}")
+            return ""
+            
     return ""
 
 # --- 6. 介面設定 (Theme & CSS) ---
@@ -197,14 +244,12 @@ with gr.Blocks(theme=theme, css=custom_css, title="Booking") as demo:
         with gr.Column():
             gr.Markdown("### 📅 預約資訊 Booking Info")
             
-            # --- 之前漏掉的 Date & Pax ---
             date_options = get_date_options()
             booking_date = gr.Dropdown(choices=date_options, label="選擇日期 Select Date", interactive=True)
             pax_count = gr.Slider(minimum=1, maximum=10, value=2, step=1, label="用餐人數 Guest Count")
 
         with gr.Column():
              gr.Markdown("### 🕰️ 選擇時段 Time Slot")
-             # --- 之前漏掉的 Status & Time ---
              status_box = gr.Markdown("請先選擇日期...", visible=True)
              time_slot = gr.Dropdown(choices=[], label="可用時段 Available Time", interactive=True)
 
@@ -216,9 +261,8 @@ with gr.Blocks(theme=theme, css=custom_css, title="Booking") as demo:
             cust_name = gr.Textbox(label="訂位姓名 Name *", placeholder="ex. 王小明")
             cust_tel = gr.Textbox(label="手機號碼 Phone *", placeholder="ex. 0912-xxx-xxx")
         
-        # --- 新增的 Email 與 備註 ---
         with gr.Row():
-            cust_email = gr.Textbox(label="電子信箱 E-mail (接收確認 E-Mail用)", placeholder="example@gmail.com")
+            cust_email = gr.Textbox(label="電子信箱 E-mail (接收確認信用，請記得檢查垃圾信件匣。)", placeholder="example@gmail.com")
         with gr.Row():
             cust_remarks = gr.Textbox(label="備註 Remarks (過敏/慶生/特殊需求)", lines=2)