Update app.py

app.py

@@ -4,67 +4,119 @@ import os
 import requests
 from supabase import create_client, Client
 
-# --- 連線設定 ---
+# --- 1. 連線設定 ---
+# 請確認 HF Secrets 裡都有設定這些變數
 LINE_ACCESS_TOKEN = os.getenv("LINE_ACCESS_TOKEN")
 LINE_ADMIN_ID = os.getenv("LINE_ADMIN_ID")
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_KEY")
 
-# 🔍【除錯用】印出變數狀態 (不要印出完整金鑰，印出是否讀取到即可)
-print(f"DEBUG: SUPABASE_URL is {'Set' if SUPABASE_URL else 'NONE/EMPTY'}")
-print(f"DEBUG: SUPABASE_KEY is {'Set' if SUPABASE_KEY else 'NONE/EMPTY'}")
+# 建立 Supabase 連線
+if SUPABASE_URL and SUPABASE_KEY:
+    supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+else:
+    print("⚠️ 警告：SUPABASE 環境變數未設定")
 
-# --- 如果沒抓到，先不要執行 create_client 以免報錯 ---
-if not SUPABASE_URL or not SUPABASE_KEY:
-    raise ValueError("❌ 錯誤：找不到 Supabase 環境變數，請檢查 Space A 的 Secrets 設定！")
+# --- 2. 輔助函式 (日期與時間) ---
+def get_date_options():
+    options = []
+    today = datetime.now()
+    weekdays = ["(一)", "(二)", "(三)", "(四)", "(五)", "(六)", "(日)"]
+    for i in range(30): 
+        current_date = today + timedelta(days=i)
+        date_str = f"{current_date.strftime('%Y-%m-%d')} {weekdays[current_date.weekday()]}"
+        options.append(date_str)
+    return options
 
-supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+def update_time_slots(date_str):
+    if not date_str:
+        return gr.update(choices=[]), "請先選擇日期"
+    try:
+        clean_date_str = date_str.split(" ")[0] 
+        date_obj = datetime.strptime(clean_date_str, "%Y-%m-%d")
+        weekday = date_obj.weekday() 
+    except:
+        return gr.update(choices=[]), "日期格式錯誤"
 
-# --- LINE 通知函式 (更新版：包含備註) ---
-def send_line_notify(data):
-    if not LINE_ACCESS_TOKEN or not LINE_ADMIN_ID: return
+    slots = ["18:00", "18:30", "19:00", "19:30", "20:00", "20:30", 
+             "21:00", "21:30", "22:00", "22:30", "23:00", "23:30", 
+             "00:00", "00:30", "01:00"]
     
-    msg_body = [
-        {"type": "text", "text": "姓名", "color": "#aaaaaa", "size": "sm", "flex": 2},
-        {"type": "text", "text": data['name'], "wrap": True, "color": "#666666", "size": "sm", "flex": 5},
-        {"type": "text", "text": "電話", "color": "#aaaaaa", "size": "sm", "flex": 2},
-        {"type": "text", "text": data['tel'], "wrap": True, "color": "#666666", "size": "sm", "flex": 5},
-        {"type": "text", "text": "Email", "color": "#aaaaaa", "size": "sm", "flex": 2},
-        {"type": "text", "text": data.get('email', '-'), "wrap": True, "color": "#666666", "size": "sm", "flex": 5},
-        {"type": "text", "text": "備註", "color": "#aaaaaa", "size": "sm", "flex": 2},
-        {"type": "text", "text": data.get('remarks', '無'), "wrap": True, "color": "#666666", "size": "sm", "flex": 5}
-    ]
+    if weekday == 4 or weekday == 5: 
+        slots.extend(["01:30", "02:00", "02:30"])
+        status_msg = f"✨ 已選擇 {date_str} (週末營業至 03:00)"
+    else:
+        slots.extend(["01:30"])
+        status_msg = f"🌙 已選擇 {date_str} (平日營業至 02:00)"
+        
+    return gr.update(choices=slots, value=slots[0] if slots else None), status_msg
+
+# --- 3. LINE 通知函式 ---
+def send_line_notify(data):
+    if not LINE_ACCESS_TOKEN or not LINE_ADMIN_ID: 
+        return
     
-    # ... (此處省略部分重複的 Flex Message 結構，重點是把上面的 msg_body 塞進去) ...
-    # 為了節省篇幅，這裡直接發送簡單通知，您可以用之前的 Flex Message 結構替換
-    requests.post(
-        "https://api.line.me/v2/bot/message/push",
-        headers={"Authorization": f"Bearer {LINE_ACCESS_TOKEN}", "Content-Type": "application/json"},
-        json={"to": LINE_ADMIN_ID, "messages": [{"type": "text", "text": f"🔥 新訂位：{data['name']} ({data['pax']}人)\n時間：{data['date']} {data['time']}\n備註：{data.get('remarks', '無')}"}]}
+    # 簡單的文字通知 (您也可以換回之前的 Flex Message)
+    message = (
+        f"🔥 新訂位通知 🔥\n"
+        f"姓名：{data['name']}\n"
+        f"電話：{data['tel']}\n"
+        f"日期：{data['date']} {data['time']}\n"
+        f"人數：{data['pax']} 位\n"
+        f"Email：{data.get('email', '-')}\n"
+        f"備註：{data.get('remarks', '無')}"
     )
+    
+    try:
+        requests.post(
+            "https://api.line.me/v2/bot/message/push",
+            headers={
+                "Authorization": f"Bearer {LINE_ACCESS_TOKEN}", 
+                "Content-Type": "application/json"
+            },
+            json={
+                "to": LINE_ADMIN_ID, 
+                "messages": [{"type": "text", "text": message}]
+            }
+        )
+    except Exception as e:
+        print(f"LINE 發送失敗: {e}")
 
-# --- 核心邏輯 ---
+# --- 4. 核心邏輯：處理訂位 ---
 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 "⚠️ 請完整填寫必填欄位"
+        return "⚠️ 請完整填寫必填欄位 (姓名、電話、日期、時間)"
     
     data = {
-        "name": name, "tel": tel, "email": email, 
-        "date": date_str, "time": time, "pax": pax, 
-        "remarks": remarks, "status": "待處理"
+        "name": name, 
+        "tel": tel, 
+        "email": email, 
+        "date": date_str, 
+        "time": time, 
+        "pax": pax, 
+        "remarks": remarks, 
+        "status": "待處理"
     }
     
     try:
-        # 寫入資料庫
+        # 1. 寫入資料庫
         supabase.table("bookings").insert(data).execute()
-        # 發送 LINE
+        
+        # 2. 發送 LINE
         send_line_notify(data)
-        return """<div style='text-align: center; color: #fff; padding: 20px; border: 1px solid #d4af37; border-radius: 8px; background: #222;'>
-            <h2 style='color: #d4af37;'>Request Received</h2><p>🥂 預約申請已提交，請留意確認信。</p></div>"""
+        
+        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>
+        </div>
+        """
     except Exception as e:
+        print(f"Error: {e}")
         return f"❌ 系統錯誤: {str(e)}"
 
-# --- 處理確認連結 (Webhook) ---
+# --- 5. 核心邏輯：處理確認連結 (Webhook) ---
 def check_confirmation(request: gr.Request):
     """
     當網址帶有 ?id=xx&action=confirm 時觸發
@@ -78,50 +130,122 @@ def check_confirmation(request: gr.Request):
         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:#d4af37; color:black; text-align:center; margin-bottom:20px; border-radius:8px;'>
-                🎉 感謝您的確認！我們期待您的光臨。
+            <div style='padding:20px; background:#d4af37; color:black; text-align:center; margin-bottom:20px; border-radius:8px; font-weight:bold;'>
+                🎉 感謝您的確認！我們期待您的光臨。 (訂單編號: {bid})
             </div>
             """
-        except:
+        except Exception as e:
+            print(f"Confirm Error: {e}")
             return ""
     return ""
 
-# --- 介面佈局 ---
-# 主題與 CSS 設定 (沿用之前的) ... 
+# --- 6. 介面設定 (Theme & CSS) ---
+theme = gr.themes.Soft(
+    primary_hue="amber",
+    neutral_hue="zinc",
+    font=[gr.themes.GoogleFont("Playfair Display"), "ui-sans-serif", "sans-serif"],
+).set(
+    body_background_fill="#0F0F0F",
+    block_background_fill="#1a1a1a",
+    block_border_width="1px",
+    block_border_color="#333",
+    input_background_fill="#262626",
+    input_border_color="#444",
+    body_text_color="#E0E0E0",
+    block_title_text_color="#d4af37",
+    button_primary_background_fill="#d4af37",
+    button_primary_text_color="#000000",
+)
 
-with gr.Blocks(theme=gr.themes.Soft(), title="Booking") as demo:
-    # 隱藏的 HTML 區塊，用來接收確認訊息
+custom_css = """
+footer {display: none !important;}
+.gradio-container, .block, .row, .column { overflow: visible !important; }
+.options, .wrap .options {
+    background-color: #262626 !important;
+    border: 1px solid #d4af37 !important;
+    z-index: 10000 !important;
+    box-shadow: 0 5px 15px rgba(0,0,0,0.5);
+}
+.item, .options .item { color: #E0E0E0 !important; padding: 8px 12px !important; }
+.item:hover, .item.selected, .options .item:hover { background-color: #d4af37 !important; color: black !important; }
+input:focus, .dropdown-trigger:focus-within { border-color: #d4af37 !important; box-shadow: 0 0 8px rgba(212, 175, 55, 0.4) !important; }
+h3 { border-bottom: 1px solid #444; padding-bottom: 5px; margin-bottom: 10px; }
+.legal-footer {
+    text-align: center; margin-top: 15px; padding-top: 15px; border-top: 1px solid #333; color: #666; font-size: 0.75rem; line-height: 1.5; font-family: sans-serif;
+}
+.legal-footer strong { color: #888; }
+"""
+
+# --- 7. 介面佈局 (完整版) ---
+with gr.Blocks(theme=theme, css=custom_css, title="Booking") as demo:
+    
+    # [隱藏] 用來接收 URL 確認參數的區塊
     confirm_msg_box = gr.HTML()
     
-    # 頁面載入時檢查網址參數
+    # 頁面載入時，執行 check_confirmation
     demo.load(check_confirmation, inputs=None, outputs=confirm_msg_box)
 
-    # ... (Header 與 日期選擇區塊 保持不變) ...
+    with gr.Row():
+        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)
+
+    gr.HTML("<div style='height: 10px'></div>")
 
     gr.Markdown("### 👤 聯絡人資料 Contact")
     with gr.Group():
         with gr.Row():
-            cust_name = gr.Textbox(label="訂位姓名 *", placeholder="ex. 王小明")
-            cust_tel = gr.Textbox(label="手機號碼 *", placeholder="ex. 0912-xxx-xxx")
+            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="電子信箱 (接收確認信用)", placeholder="example@gmail.com")
+            cust_email = gr.Textbox(label="電子信箱 Email (接收確認信用)", placeholder="example@gmail.com")
         with gr.Row():
-            cust_remarks = gr.Textbox(label="備註 (過敏/慶生/特殊需求)", lines=2)
+            cust_remarks = gr.Textbox(label="備註 Remarks (過敏/慶生/特殊需求)", lines=2)
 
-    submit_btn = gr.Button("確認預約 Request Booking", variant="primary")
+    gr.HTML("<div style='height: 15px'></div>")
+    
+    submit_btn = gr.Button("確認預約 Request Booking", size="lg", variant="primary")
     output_msg = gr.HTML()
     
-    # ... (Event Listeners) ...
+    # 頁尾警語
+    gr.HTML("""
+    <div class="legal-footer">
+        <p style="margin-bottom: 5px;">© 2026 CIE CIE TAIPEI. All Rights Reserved.</p>
+        <p>內容涉及酒類產品訊息，請勿轉發分享給未達法定購買年齡者；未滿十八歲請勿飲酒。<br>
+        <strong>喝酒不開車，開車不喝酒。</strong></p>
+    </div>
+    """)
+
+    # --- 互動事件綁定 ---
+    # 1. 日期改變 -> 更新時段
+    booking_date.change(update_time_slots, inputs=booking_date, outputs=[time_slot, status_box])
+    
+    # 2. 送出按鈕 -> 處理訂位
     submit_btn.click(
         handle_booking, 
         inputs=[cust_name, cust_tel, cust_email, booking_date, time_slot, pax_count, cust_remarks], 
         outputs=output_msg
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()