Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 from supabase import create_client, Client
 from datetime import datetime, timedelta, timezone
 
-# 設定台北時區
+# ✅ 補回：設定台北時區
 TAIPEI_TZ = timezone(timedelta(hours=8))
 
 # --- 設定 ---
@@ -13,12 +13,13 @@ SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_KEY")
 GAS_MAIL_URL = os.getenv("GAS_MAIL_URL")
 LINE_ACCESS_TOKEN = os.getenv("LINE_ACCESS_TOKEN")
-# ⚠️ 請確認這是您 Space A 的正確網址 (結尾不要有斜線)
 PUBLIC_SPACE_URL = "https://deeplearning101-ciecietaipei.hf.space" 
 
+# 取得帳密 (若沒設定則使用預設值)
+REAL_ADMIN_USER = os.getenv("ADMIN_USER") or "Deep Learning 101"
+REAL_ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD") or "2016-11-11"
+
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
-ADMIN_USER = os.getenv("ADMIN_USER")
-ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
 
 def get_bookings():
     res = supabase.table("bookings").select("*").order("created_at", desc=True).execute()
@@ -37,20 +38,16 @@ def send_confirmation_hybrid(booking_id):
         email, user_id = booking.get('email'), booking.get('user_id')
         log_msg = ""
         
-        # 產生確認連結
         confirm_link = f"{PUBLIC_SPACE_URL}/?id={booking_id}&action=confirm"
         cancel_link = f"{PUBLIC_SPACE_URL}/?id={booking_id}&action=cancel"
 
-        # ---------------------------------------------------------
-        # 1. Email 發送 (改良版：使用 Table 排版，解決跑版問題)
-        # ---------------------------------------------------------
+        # 1. Email 發送
         if email and "@" in email:
             try:
                 html = f"""
                 <div style="padding: 20px; background: #111; color: #d4af37; border-radius: 10px; max-width: 600px; margin: 0 auto; font-family: sans-serif;">
                     <h2 style="border-bottom: 1px solid #d4af37; padding-bottom: 15px; text-align: center; letter-spacing: 2px;">Cié Cié Taipei</h2>
                     <p style="font-size: 16px; margin-top: 20px; color: #eee;">{booking['name']} 您好，已為您保留座位：</p>
-                    
                     <div style="background: #222; padding: 15px; border-radius: 8px; margin: 20px 0; border-left: 4px solid #d4af37;">
                         <ul style="color: #eee; list-style: none; padding: 0; margin: 0; line-height: 2;">
                             <li>📅 日期：<strong style="color:#fff;">{booking['date']}</strong></li>
@@ -59,7 +56,6 @@ def send_confirmation_hybrid(booking_id):
                             <li>📝 備註：{booking.get('remarks') or '無'}</li>
                         </ul>
                     </div>
-
                     <table width="100%" border="0" cellspacing="0" cellpadding="0">
                         <tr>
                             <td align="center">
@@ -68,7 +64,6 @@ def send_confirmation_hybrid(booking_id):
                             </td>
                         </tr>
                     </table>
-                    
                     <hr style="border: 0; border-top: 1px solid #333; margin-top: 30px;">
                     <p style="color: #666; font-size: 12px; text-align: center;">如需更改，請直接回覆此信件。</p>
                 </div>
@@ -78,115 +73,98 @@ def send_confirmation_hybrid(booking_id):
             except Exception as e:
                 log_msg += f"⚠️ Email 失敗: {e} "
         
-        # ---------------------------------------------------------
-        # 2. LINE 發送 (升級版：Flex Message 卡片按鈕)
-        # ---------------------------------------------------------
+        # 2. LINE 發送
         if not LINE_ACCESS_TOKEN:
             log_msg += "| ⚠️ 未設定 LINE_ACCESS_TOKEN"
         elif not user_id or len(str(user_id)) < 10:
             log_msg += "| ℹ️ 無 LINE ID"
         else:
             try:
-                # 定義 Flex Message 內容
                 flex_payload = {
                     "type": "flex",
                     "altText": "您有一筆訂位確認通知",
                     "contents": {
                         "type": "bubble",
                         "styles": { "header": {"backgroundColor": "#222222"}, "body": {"backgroundColor": "#2c2c2c"}, "footer": {"backgroundColor": "#2c2c2c"} },
-                        "header": {
-                            "type": "box",
-                            "layout": "vertical",
-                            "contents": [
-                                {"type": "text", "text": "Cié Cié Taipei", "color": "#d4af37", "weight": "bold", "size": "xl", "align": "center"}
-                            ]
-                        },
+                        "header": { "type": "box", "layout": "vertical", "contents": [ {"type": "text", "text": "Cié Cié Taipei", "color": "#d4af37", "weight": "bold", "size": "xl", "align": "center"} ] },
                         "body": {
-                            "type": "box",
-                            "layout": "vertical",
+                            "type": "box", "layout": "vertical",
                             "contents": [
                                 {"type": "text", "text": "訂位確認", "weight": "bold", "size": "lg", "color": "#ffffff", "align": "center", "margin": "md"},
                                 {"type": "separator", "margin": "lg", "color": "#444444"},
                                 {"type": "box", "layout": "vertical", "margin": "lg", "spacing": "sm", "contents": [
-                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [
-                                        {"type": "text", "text": "姓名", "color": "#aaaaaa", "size": "sm", "flex": 2},
-                                        {"type": "text", "text": f"{booking['name']}", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4}
-                                    ]},
-                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [
-                                        {"type": "text", "text": "日期", "color": "#aaaaaa", "size": "sm", "flex": 2},
-                                        {"type": "text", "text": f"{booking['date']}", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4}
-                                    ]},
-                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [
-                                        {"type": "text", "text": "時間", "color": "#aaaaaa", "size": "sm", "flex": 2},
-                                        {"type": "text", "text": f"{booking['time']}", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4}
-                                    ]},
-                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [
-                                        {"type": "text", "text": "人數", "color": "#aaaaaa", "size": "sm", "flex": 2},
-                                        {"type": "text", "text": f"{booking['pax']} 位", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4}
-                                    ]}
+                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [ {"type": "text", "text": "姓名", "color": "#aaaaaa", "size": "sm", "flex": 2}, {"type": "text", "text": f"{booking['name']}", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4} ]},
+                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [ {"type": "text", "text": "日期", "color": "#aaaaaa", "size": "sm", "flex": 2}, {"type": "text", "text": f"{booking['date']}", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4} ]},
+                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [ {"type": "text", "text": "時間", "color": "#aaaaaa", "size": "sm", "flex": 2}, {"type": "text", "text": f"{booking['time']}", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4} ]},
+                                    {"type": "box", "layout": "baseline", "spacing": "sm", "contents": [ {"type": "text", "text": "人數", "color": "#aaaaaa", "size": "sm", "flex": 2}, {"type": "text", "text": f"{booking['pax']} 位", "wrap": True, "color": "#ffffff", "size": "sm", "flex": 4} ]}
                                 ]}
                             ]
                         },
                         "footer": {
-                            "type": "box",
-                            "layout": "vertical",
-                            "spacing": "sm",
+                            "type": "box", "layout": "vertical", "spacing": "sm",
                             "contents": [
-                                {
-                                    "type": "button",
-                                    "style": "primary",
-                                    "color": "#d4af37",
-                                    "height": "sm",
-                                    "action": {
-                                        "type": "uri",
-                                        "label": "✅ 確認出席",
-                                        "uri": confirm_link
-                                    }
-                                },
-                                {
-                                    "type": "button",
-                                    "style": "secondary",
-                                    "height": "sm",
-                                    "color": "#aaaaaa",
-                                    "action": {
-                                        "type": "uri",
-                                        "label": "🚫 取消訂位",
-                                        "uri": cancel_link
-                                    }
-                                }
+                                { "type": "button", "style": "primary", "color": "#d4af37", "height": "sm", "action": { "type": "uri", "label": "✅ 確認出席", "uri": confirm_link } },
+                                { "type": "button", "style": "secondary", "height": "sm", "color": "#aaaaaa", "action": { "type": "uri", "label": "🚫 取消訂位", "uri": cancel_link } }
                             ]
                         }
                     }
                 }
-
-                # 發送請求
-                r = requests.post(
-                    "https://api.line.me/v2/bot/message/push", 
-                    headers={"Authorization": f"Bearer {LINE_ACCESS_TOKEN}", "Content-Type": "application/json"}, 
-                    json={"to": user_id, "messages": [flex_payload]}
-                )
-                
-                if r.status_code == 200:
-                    log_msg += "| ✅ LINE Flex ok"
-                else:
-                    log_msg += f"| ❌ LINE 錯誤: {r.text}"
-            except Exception as e: 
-                log_msg += f"| ❌ LINE 例外: {e}"
+                r = requests.post("https://api.line.me/v2/bot/message/push", headers={"Authorization": f"Bearer {LINE_ACCESS_TOKEN}", "Content-Type": "application/json"}, json={"to": user_id, "messages": [flex_payload]})
+                if r.status_code == 200: log_msg += "| ✅ LINE Flex ok"
+                else: log_msg += f"| ❌ LINE 錯誤: {r.text}"
+            except Exception as e: log_msg += f"| ❌ LINE 例外: {e}"
 
         supabase.table("bookings").update({"status": "已發確認信"}).eq("id", booking_id).execute()
         return log_msg
     except Exception as e: return f"❌ Error: {str(e)}"
 
+# --- [修正點] 登入邏輯 ---
+def check_login(user, password):
+    if user == REAL_ADMIN_USER and password == REAL_ADMIN_PASSWORD:
+        # 登入成功：隱藏登入頁，顯示後台，並清空錯誤訊息
+        return {
+            login_row: gr.update(visible=False),
+            admin_row: gr.update(visible=True),
+            error_msg: ""
+        }
+    else:
+        return {
+            error_msg: "❌ 帳號或密碼錯誤"
+        }
+
+# --- 介面開始 ---
 with gr.Blocks(title="Admin") as demo:
-    gr.Markdown("# 🍷 訂位管理後台")
-    refresh_btn = gr.Button("🔄 重新整理")
-    booking_table = gr.Dataframe(interactive=False)
-    with gr.Row():
-        id_input = gr.Number(label="訂單 ID", precision=0)
-        action_btn = gr.Button("📧 發送確認信 (Hybrid)", variant="primary")
-    log_output = gr.Textbox(label="結果")
-    refresh_btn.click(get_bookings, outputs=booking_table)
-    action_btn.click(send_confirmation_hybrid, inputs=id_input, outputs=log_output)
+    
+    # 1. 登入介面 (預設顯示)
+    with gr.Group(visible=True) as login_row:
+        gr.Markdown("# 🔒 請登入後台")
+        with gr.Row():
+            username_input = gr.Textbox(label="帳號 Username", placeholder="Enter username")
+            password_input = gr.Textbox(label="密碼 Password", type="password", placeholder="Enter password")
+        login_btn = gr.Button("登入 Login", variant="primary")
+        error_msg = gr.Markdown("", style="color: red;")
+        
+    # 2. 後台介面 (預設隱藏)
+    with gr.Group(visible=False) as admin_row:
+        gr.Markdown("# 🍷 訂位管理後台 (Dashboard)")
+        refresh_btn = gr.Button("🔄 重新整理")
+        booking_table = gr.Dataframe(interactive=False)
+        with gr.Row():
+            id_input = gr.Number(label="訂單 ID", precision=0)
+            action_btn = gr.Button("📧 發送確認信 (Hybrid)", variant="primary")
+        log_output = gr.Textbox(label="結果")
+        
+        # 綁定後台按鈕功能
+        refresh_btn.click(get_bookings, outputs=booking_table)
+        action_btn.click(send_confirmation_hybrid, inputs=id_input, outputs=log_output)
+
+    # 3. 綁定登入按鈕事件
+    login_btn.click(
+        check_login, 
+        inputs=[username_input, password_input], 
+        outputs=[login_row, admin_row, error_msg]
+    )
 
 if __name__ == "__main__":
-    demo.launch(auth=(ADMIN_USER or "admin", ADMIN_PASSWORD or "123456"))
+    # ⚠️ 注意：這裡不再使用 auth=(...)，改用上面寫的 Python 邏輯來切換顯示
+    demo.launch()