ragline

Sleeping

App Files Files Community

AlanRex commited on 16 days ago

Commit

2e431ca

verified ·

1 Parent(s): b7c3870

Update main.py

Browse files

Files changed (1) hide show

main.py +217 -2

main.py CHANGED Viewed

@@ -7,6 +7,7 @@ from linebot.exceptions import InvalidSignatureError
 from linebot.models import MessageEvent, TextMessage, TextSendMessage
 import PyPDF2
 import logging
 from datetime import datetime, timedelta
 from google.oauth2.credentials import Credentials
 from google.oauth2 import service_account
@@ -275,7 +276,7 @@ def parse_date(date_str):
             except ValueError:
                 pass
-    return None, "❌ 日期格式不正確，請重新輸入。\n例如：明天、12/26、2025-12-30"
 def validate_time(time_str):
     time_keywords = ["點", "時", ":", "-", "到", "至", "~"]
@@ -496,11 +497,186 @@ def root():
             "google_calendar": calendar_service is not None,
             "ai_qa": ai_enabled,
             "pdf_loaded": len(pdf_content) > 0
         }
     }
 @app.get("/bookings")
 def get_bookings():
     bookings_file = pathlib.Path("bookings/bookings.json")
     if bookings_file.exists():
         with open(bookings_file, 'r', encoding='utf-8') as f:
@@ -508,6 +684,45 @@ def get_bookings():
         return {"total": len(bookings), "bookings": bookings}
     return {"total": 0, "bookings": []}
 @app.post("/webhook")
 async def webhook(
     request: Request,
@@ -573,7 +788,7 @@ def handle_message(event):
                 full_prompt = user_message
             response = client.models.generate_content(
-                model="gemini-1.5-flash",
                 contents=full_prompt
             )

 from linebot.models import MessageEvent, TextMessage, TextSendMessage
 import PyPDF2
 import logging
+from fastapi.responses import HTMLResponse
 from datetime import datetime, timedelta
 from google.oauth2.credentials import Credentials
 from google.oauth2 import service_account
             except ValueError:
                 pass
+    return None, "❌ 日期格式不正確，請重新輸入。\n例如：明天、2025-12-30"
 def validate_time(time_str):
     time_keywords = ["點", "時", ":", "-", "到", "至", "~"]
             "google_calendar": calendar_service is not None,
             "ai_qa": ai_enabled,
             "pdf_loaded": len(pdf_content) > 0
+        },
+        "endpoints": {
+            "查看所有預約": "/bookings",
+            "下載預約資料": "/bookings/download",
+            "查看最新預約": "/bookings/latest",
+            "管理介面": "/admin"
         }
     }
+@app.get("/admin")
+def admin_panel():
+    """簡單的管理介面"""
+    bookings_file = pathlib.Path("bookings/bookings.json")
+    bookings = []
+    if bookings_file.exists():
+        with open(bookings_file, 'r', encoding='utf-8') as f:
+            bookings = json.load(f)
+    # 生成 HTML
+    html = """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>琴房預約管理</title>
+        <style>
+            body {
+                font-family: Arial, sans-serif;
+                max-width: 1200px;
+                margin: 0 auto;
+                padding: 20px;
+                background: #f5f5f5;
+            }
+            h1 {
+                color: #333;
+                text-align: center;
+            }
+            .stats {
+                display: flex;
+                gap: 20px;
+                margin-bottom: 30px;
+            }
+            .stat-card {
+                flex: 1;
+                background: white;
+                padding: 20px;
+                border-radius: 8px;
+                box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+            }
+            .stat-card h3 {
+                margin: 0 0 10px 0;
+                color: #666;
+                font-size: 14px;
+            }
+            .stat-card .number {
+                font-size: 32px;
+                font-weight: bold;
+                color: #4CAF50;
+            }
+            table {
+                width: 100%;
+                background: white;
+                border-collapse: collapse;
+                border-radius: 8px;
+                overflow: hidden;
+                box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+            }
+            th {
+                background: #4CAF50;
+                color: white;
+                padding: 12px;
+                text-align: left;
+            }
+            td {
+                padding: 12px;
+                border-bottom: 1px solid #ddd;
+            }
+            tr:hover {
+                background: #f9f9f9;
+            }
+            .status-pending {
+                color: #FF9800;
+                font-weight: bold;
+            }
+            .calendar-link {
+                color: #2196F3;
+                text-decoration: none;
+            }
+            .calendar-link:hover {
+                text-decoration: underline;
+            }
+            .refresh-btn {
+                background: #4CAF50;
+                color: white;
+                padding: 10px 20px;
+                border: none;
+                border-radius: 4px;
+                cursor: pointer;
+                font-size: 16px;
+                margin-bottom: 20px;
+            }
+            .refresh-btn:hover {
+                background: #45a049;
+            }
+        </style>
+    </head>
+    <body>
+        <h1>🎹 琴房預約管理系統</h1>
+        <div class="stats">
+            <div class="stat-card">
+                <h3>總預約數</h3>
+                <div class="number">""" + str(len(bookings)) + """</div>
+            </div>
+            <div class="stat-card">
+                <h3>待確認</h3>
+                <div class="number">""" + str(len([b for b in bookings if b.get('status') == 'pending'])) + """</div>
+            </div>
+            <div class="stat-card">
+                <h3>Google Calendar</h3>
+                <div class="number">""" + ('✓' if calendar_service else '✗') + """</div>
+            </div>
+        </div>
+        <button class="refresh-btn" onclick="location.reload()">🔄 重新整理</button>
+        <table>
+            <thead>
+                <tr>
+                    <th>預約編號</th>
+                    <th>日期</th>
+                    <th>時段</th>
+                    <th>人數</th>
+                    <th>琴房</th>
+                    <th>狀態</th>
+                    <th>建立時間</th>
+                    <th>日曆</th>
+                </tr>
+            </thead>
+            <tbody>
+    """
+    # 按時間倒序排列
+    for booking in reversed(bookings):
+        calendar_icon = '📆' if booking.get('calendar_link') else '❌'
+        html += f"""
+                <tr>
+                    <td>{booking['booking_id']}</td>
+                    <td>{booking['date']}</td>
+                    <td>{booking['time']}</td>
+                    <td>{booking['people']}人</td>
+                    <td>{booking['room']}</td>
+                    <td class="status-pending">{booking['status']}</td>
+                    <td>{booking['created_at']}</td>
+                    <td>
+        """
+        if booking.get('calendar_link'):
+            html += f'<a href="{booking["calendar_link"]}" target="_blank" class="calendar-link">{calendar_icon} 查看</a>'
+        else:
+            html += calendar_icon
+        html += """
+                    </td>
+                </tr>
+        """
+    html += """
+            </tbody>
+        </table>
+    </body>
+    </html>
+    """
+    from fastapi.responses import HTMLResponse
+    return HTMLResponse(content=html)
 @app.get("/bookings")
 def get_bookings():
+    """查看所有預約（管理用）"""
     bookings_file = pathlib.Path("bookings/bookings.json")
     if bookings_file.exists():
         with open(bookings_file, 'r', encoding='utf-8') as f:
         return {"total": len(bookings), "bookings": bookings}
     return {"total": 0, "bookings": []}
+@app.get("/bookings/download")
+def download_bookings():
+    """下載預約資料（JSON 格式）"""
+    bookings_file = pathlib.Path("bookings/bookings.json")
+    if bookings_file.exists():
+        with open(bookings_file, 'r', encoding='utf-8') as f:
+            return json.load(f)
+    return []
+@app.get("/bookings/latest")
+def get_latest_booking():
+    """查看最新一筆預約"""
+    bookings_file = pathlib.Path("bookings/bookings.json")
+    if bookings_file.exists():
+        with open(bookings_file, 'r', encoding='utf-8') as f:
+            bookings = json.load(f)
+            if bookings:
+                return bookings[-1]  # 返回最後一筆
+    return {"message": "尚無預約資料"}
+@app.delete("/bookings/{booking_id}")
+def delete_booking(booking_id: str):
+    """刪除指定預約"""
+    bookings_file = pathlib.Path("bookings/bookings.json")
+    if bookings_file.exists():
+        with open(bookings_file, 'r', encoding='utf-8') as f:
+            bookings = json.load(f)
+        # 找到並移除指定預約
+        updated_bookings = [b for b in bookings if b['booking_id'] != booking_id]
+        if len(updated_bookings) < len(bookings):
+            with open(bookings_file, 'w', encoding='utf-8') as f:
+                json.dump(updated_bookings, f, ensure_ascii=False, indent=2)
+            return {"message": f"已刪除預約 {booking_id}"}
+        else:
+            return {"message": f"找不到預約 {booking_id}"}
+    return {"message": "沒有預約資料"}
 @app.post("/webhook")
 async def webhook(
     request: Request,
                 full_prompt = user_message
             response = client.models.generate_content(
+                model="gemini-2.5-flash",  # 使用最新版本
                 contents=full_prompt
             )