import gradio as gr
import os
import requests
from supabase import create_client, Client
from datetime import datetime, timedelta, timezone
# --- 1. 設定與初始化 ---
TAIPEI_TZ = timezone(timedelta(hours=8))
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")
if SUPABASE_URL and SUPABASE_KEY:
supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
# --- 2. 座位控管設定 ---
DEFAULT_LIMIT = 30
SPECIAL_DAYS = {
"2026-12-31": 10,
"2026-02-14": 15
}
# --- 3. 輔助函式 ---
def get_date_options():
options = []
today = datetime.now(TAIPEI_TZ)
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
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=[]), "日期格式錯誤"
slots = ["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", "01:30"]
if weekday == 4 or weekday == 5: slots.extend(["02:00", "02:30"])
now = datetime.now(TAIPEI_TZ)
if date_obj.date() == now.date():
current_time_str = now.strftime("%H:%M")
valid_slots = []
for s in slots:
h = int(s.split(":")[0])
if h < 5:
valid_slots.append(s)
elif s > current_time_str:
valid_slots.append(s)
slots = valid_slots
clean_date = date_str.split(" ")[0]
daily_limit = SPECIAL_DAYS.get(clean_date, DEFAULT_LIMIT)
try:
res = supabase.table("bookings").select("pax").eq("date", date_str).neq("status", "顧客已取消").execute()
current_total = sum([item['pax'] for item in res.data])
remaining = daily_limit - current_total
status_msg = f"✨ {date_str} (剩餘座位: {remaining} 位)"
except: status_msg = f"✨ {date_str}"
return gr.update(choices=slots, value=slots[0] if slots else None), status_msg
def init_ui():
fresh_dates = get_date_options()
default_date = fresh_dates[0]
time_update, status_msg = update_time_slots(default_date)
return (gr.update(choices=fresh_dates, value=default_date), time_update, status_msg)
# --- 4. 核心邏輯 (抓 ID 與 處理訂位) ---
def get_line_id_from_url(request: gr.Request):
if request:
return request.query_params.get("line_id", "")
return ""
def handle_booking(name, tel, email, date_str, time, pax, remarks, line_id):
# 1. 基本必填檢查
if not name or not tel or not date_str or not time:
return "⚠️ 請完整填寫必填欄位 (姓名、電話、日期、時間)"
# 🔥🔥🔥 [新增] 格式驗證邏輯 🔥🔥🔥
# 2. Email 驗證 (如果用戶有填寫 Email，就必須包含 @)
if email and "@" not in email:
return "⚠️ Email 格式錯誤 (請確認包含 @ 符號)"
# 3. 電話驗證 (寬鬆模式：相容外國人)
# 邏輯：不限制符號 (+, -, 空白, 括號都可)，但「數字」的總數至少要有 6 碼
# 例如: "+886 912-345-678" (數字12碼) -> 通過
# 例如: "Test Phone" (數字0碼) -> 失敗
digit_count = sum(c.isdigit() for c in tel)
if digit_count < 6:
return "⚠️ 電話號碼格式錯誤 (請填寫有效的電話號碼)"
# 🔥🔥🔥 驗證結束 🔥🔥🔥
clean_date = date_str.split(" ")[0]
daily_limit = SPECIAL_DAYS.get(clean_date, DEFAULT_LIMIT)
try:
res = supabase.table("bookings").select("pax").eq("date", date_str).neq("status", "顧客已取消").execute()
current_total = sum([item['pax'] for item in res.data])
if current_total + pax > daily_limit:
return "⚠️ 抱歉，該時段剩餘座位不足，請調整人數或日期。"
except: pass
try:
existing = supabase.table("bookings").select("id").eq("tel", tel).eq("date", date_str).eq("time", time).neq("status", "顧客已取消").execute()
if existing.data: return "⚠️ 您已預約過此時段，請勿重複提交。"
except: pass
data = {
"name": name, "tel": tel, "email": email, "date": date_str, "time": time,
"pax": pax, "remarks": remarks, "status": "待處理",
"user_id": line_id
}
try:
supabase.table("bookings").insert(data).execute()
# [LINE 通知老闆 - 新訂位]
if LINE_ACCESS_TOKEN and LINE_ADMIN_ID:
src = "🟢 LINE用戶" if line_id else "⚪ 訪客"
note = remarks if remarks else "無"
msg = (
f"🔥 新訂位 ({src})\n"
f"姓名：{name}\n"
f"電話：{tel}\n"
f"日期：{date_str}\n"
f"時間：{time}\n"
f"人數：{pax}人\n"
f"備註：{note}"
)
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": msg}]})
return """Request Received
🥂 預約申請已提交
請留意 Email 確認信 或 Line 訊息。