Update main.py

main.py

@@ -1,177 +1,216 @@
-import json, os, glob, time
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi import FastAPI, Request, Header, BackgroundTasks, HTTPException
-from google import genai
-from linebot import LineBotApi, WebhookHandler
-from linebot.exceptions import InvalidSignatureError
-from linebot.models import MessageEvent, TextMessage, TextSendMessage
-import PyPDF2
-import logging
+# 在主程式中加入以下功能
 
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+import re
+from datetime import datetime, timedelta
 
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-if not GOOGLE_API_KEY:
-    raise ValueError("請設定 GOOGLE_API_KEY")
-
-client = genai.Client(api_key=GOOGLE_API_KEY)
-
-# 讀取 PDF
-files = glob.glob('docs/*.pdf')
-pdf_content = ''
-
-if files:
-    logger.info(f"找到 {len(files)} 個 PDF")
-    for filename in files:
-        try:
-            with open(filename, 'rb') as pdf_file:
-                pdf_reader = PyPDF2.PdfReader(pdf_file)
-                for page in pdf_reader.pages:
-                    pdf_content += page.extract_text()
-        except Exception as e:
-            logger.error(f"讀取失敗: {e}")
-
-logger.info(f"PDF 長度: {len(pdf_content)} 字元")
-
-# 限制長度
-if len(pdf_content) > 50000:
-    pdf_content = pdf_content[:50000]
-
-# 使用 Context Caching 儲存 PDF 內容
-cache_name = None
-try:
-    # 建立 cached content
-    cached_content = client.caches.create(
-        model="gemini-1.5-flash-001",
-        contents=[
-            genai.types.Content(
-                role="user",
-                parts=[genai.types.Part(text=f"""以下是參考資料，請記住這些內容：
-
-{pdf_content}
-
-你是問答助手，使用以上資料回答問題。規則：
-1. 只用提供的資料回答
-2. 不知道就說「請聯繫服務專員」
-3. 你是租借音樂空間的回覆助手，請你用音樂人員或產業人員來回覆""")]
-            )
-        ],
-        ttl="300s",  # 快取 5 分鐘
-    )
-    cache_name = cached_content.name
-    logger.info(f"成功建立 Cache: {cache_name}")
-except Exception as e:
-    logger.error(f"無法建立 Cache: {e}")
-    logger.info("將使用一般模式（不使用 Cache）")
-
-# LINE Bot 設定
-line_bot_api = LineBotApi(os.getenv("CHANNEL_ACCESS_TOKEN"))
-line_handler = WebhookHandler(os.getenv("CHANNEL_SECRET"))
-working_status = True
-
-last_request_time = {}
-REQUEST_COOLDOWN = 3
-
-app = FastAPI()
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# ========== 日期解析與驗證 ==========
+def parse_date(date_str):
+    """解析並驗證日期"""
+    today = datetime.now()
+    
+    # 處理相對日期
+    if "今天" in date_str or "今日" in date_str:
+        return today.strftime("%Y-%m-%d"), None
+    elif "明天" in date_str or "明日" in date_str:
+        return (today + timedelta(days=1)).strftime("%Y-%m-%d"), None
+    elif "後天" in date_str:
+        return (today + timedelta(days=2)).strftime("%Y-%m-%d"), None
+    
+    # 處理絕對日期格式
+    patterns = [
+        (r'(\d{4})[/-](\d{1,2})[/-](\d{1,2})', '%Y-%m-%d'),  # 2025-12-26
+        (r'(\d{1,2})[/-](\d{1,2})', '%m-%d'),  # 12/26
+    ]
+    
+    for pattern, fmt in patterns:
+        match = re.search(pattern, date_str)
+        if match:
+            try:
+                if fmt == '%m-%d':
+                    date = datetime.strptime(f"{today.year}-{match.group(0)}", f'%Y-{fmt}')
+                else:
+                    date = datetime.strptime(match.group(0), fmt)
+                
+                # 檢查是否為過去的日期
+                if date.date() < today.date():
+                    return None, "日期不能是過去的時間，請重新輸入。"
+                
+                return date.strftime("%Y-%m-%d"), None
+            except ValueError:
+                pass
+    
+    return None, "日期格式不正確，請使用如：明天、12/26、2025-12-30"
+
+# ========== 時間驗證 ==========
+def validate_time(time_str):
+    """驗證時段格式"""
+    # 簡單驗證是否包含時間相關字詞
+    time_keywords = ["點", "時", ":", "-", "到", "至"]
+    if any(kw in time_str for kw in time_keywords):
+        return True, None
+    return False, "時段格式不正確，請使用如：09:00-11:00 或 下午2點到4點"
+
+# ========== 人數驗證 ==========
+def parse_people(people_str):
+    """解析並驗證人數"""
+    # 提取數字
+    numbers = re.findall(r'\d+', people_str)
+    if numbers:
+        count = int(numbers[0])
+        if 1 <= count <= 10:  # 假設最多10人
+            return count, None
+        else:
+            return None, "人數必須在 1-10 人之間，請重新輸入。"
+    return None, "請輸入有效的人數，例如：2人、3"
+
+# ========== 琴房驗證 ==========
+AVAILABLE_ROOMS = ["A琴房", "B琴房", "C琴房", "D琴房", "任意"]
+
+def validate_room(room_str):
+    """驗證琴房選擇"""
+    # 標準化輸入
+    room_normalized = room_str.strip().upper()
+    
+    # 檢查是否在可用琴房列表中
+    for room in AVAILABLE_ROOMS:
+        if room.upper() in room_normalized or room_normalized in room.upper():
+            return room, None
+    
+    return None, f"請選擇有效的琴房：{', '.join(AVAILABLE_ROOMS)}"
 
-@app.get("/")
-def root():
-    return {
-        "status": "running",
-        "cache_enabled": cache_name is not None,
-        "pdf_length": len(pdf_content)
+# ========== 將預約資料存入檔案 ==========
+def save_booking_to_file(user_id, booking_data):
+    """將預約資料儲存到 JSON 檔案"""
+    import json
+    from pathlib import Path
+    
+    # 建立 bookings 資料夾
+    bookings_dir = Path("bookings")
+    bookings_dir.mkdir(exist_ok=True)
+    
+    # 預約檔案路徑
+    booking_file = bookings_dir / "bookings.json"
+    
+    # 讀取現有預約
+    if booking_file.exists():
+        with open(booking_file, 'r', encoding='utf-8') as f:
+            bookings = json.load(f)
+    else:
+        bookings = []
+    
+    # 新增預約
+    booking_record = {
+        "booking_id": f"BK{int(time.time())}",
+        "user_id": user_id,
+        "date": booking_data["date"],
+        "time": booking_data["time"],
+        "people": booking_data["people"],
+        "room": booking_data["room"],
+        "created_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "status": "pending"  # pending, confirmed, cancelled
     }
+    
+    bookings.append(booking_record)
+    
+    # 寫入檔案
+    with open(booking_file, 'w', encoding='utf-8') as f:
+        json.dump(bookings, f, ensure_ascii=False, indent=2)
+    
+    return booking_record["booking_id"]
 
-@app.post("/webhook")
-async def webhook(request: Request, background_tasks: BackgroundTasks, x_line_signature=Header(None)):
-    body = await request.body()
-    try:
-        background_tasks.add_task(line_handler.handle, body.decode("utf-8"), x_line_signature)
-    except InvalidSignatureError:
-        raise HTTPException(status_code=400, detail="Invalid signature")
-    return "ok"
-
-@line_handler.add(MessageEvent, message=TextMessage)
-def handle_message(event):
-    if event.type != "message" or event.message.type != "text":
-        return
+# ========== 修改後的 handle_booking_flow 函數 ==========
+def handle_booking_flow_enhanced(user_id, user_message):
+    """處理預約流程（加強版）"""
+    booking = get_user_booking(user_id)
+    current_state = booking["state"]
+    
+    # 超時重置
+    if time.time() - booking["last_update"] > 600:
+        reset_booking(user_id)
+        booking = get_user_booking(user_id)
+        current_state = booking["state"]
+    
+    booking["last_update"] = time.time()
+    
+    # 取消預約
+    if user_message in ["取消", "取消預約", "重來"]:
+        reset_booking(user_id)
+        return "已取消預約流程。如需重新預約，請輸入「預約」。"
+    
+    # 開始預約
+    if current_state == BookingState.IDLE:
+        if is_booking_keyword(user_message):
+            booking["state"] = BookingState.ASKING_DATE
+            return "好的！請問您想要預約哪一天呢？\n例如：明天、12/26、2025/12/30\n\n(輸入「取消」可取消預約)"
+        else:
+            return None
+    
+    # 詢問日期（加入驗證）
+    elif current_state == BookingState.ASKING_DATE:
+        parsed_date, error = parse_date(user_message)
+        if error:
+            return error
+        booking["date"] = parsed_date
+        booking["state"] = BookingState.ASKING_TIME
+        return f"好的，預約日期：{parsed_date}\n\n請問您想要預約哪個時段呢？\n例如：09:00-11:00、下午2點到4點"
+    
+    # 詢問時段（加入驗證）
+    elif current_state == BookingState.ASKING_TIME:
+        is_valid, error = validate_time(user_message)
+        if not is_valid:
+            return error
+        booking["time"] = user_message
+        booking["state"] = BookingState.ASKING_PEOPLE
+        return f"好的，時段：{user_message}\n\n請問有幾位使用呢？\n例如：1人、2人"
+    
+    # 詢問人數（加入驗證）
+    elif current_state == BookingState.ASKING_PEOPLE:
+        people_count, error = parse_people(user_message)
+        if error:
+            return error
+        booking["people"] = people_count
+        booking["state"] = BookingState.ASKING_ROOM
+        return f"好的，人數：{people_count}人\n\n請問您想使用哪個琴房呢？\n可選擇：{', '.join(AVAILABLE_ROOMS)}"
+    
+    # 詢問琴房（加入驗證）
+    elif current_state == BookingState.ASKING_ROOM:
+        room, error = validate_room(user_message)
+        if error:
+            return error
+        booking["room"] = room
+        booking["state"] = BookingState.CONFIRMING
         
-    if event.message.text == "再見":
-        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="Bye!"))
-        return
-    
-    # 速率限制
-    user_id = event.source.user_id
-    current_time = time.time()
-    
-    if user_id in last_request_time:
-        elapsed = current_time - last_request_time[user_id]
-        if elapsed < REQUEST_COOLDOWN:
-            line_bot_api.reply_message(
-                event.reply_token,
-                TextSendMessage(text=f"請稍等 {REQUEST_COOLDOWN - int(elapsed)} 秒")
-            )
-            return
-    
-    last_request_time[user_id] = current_time
-       
-    if working_status:
-        try: 
-            prompt = event.message.text
-            logger.info(f"提問: {prompt[:50]}")
-            
-            # 如果有 Cache，使用 Cache
-            if cache_name:
-                response = client.models.generate_content(
-                    model="gemini-1.5-flash-001",
-                    contents=prompt,
-                    config=genai.types.GenerateContentConfig(
-                        max_output_tokens=1000,
-                        temperature=0.1,
-                        cached_content=cache_name
-                    )
-                )
-            else:
-                # 沒有 Cache，使用一般模式
-                response = client.models.generate_content(
-                    model="gemini-2.5-flash",
-                    contents=prompt,
-                    config=genai.types.GenerateContentConfig(
-                        max_output_tokens=1000,
-                        temperature=0.1,
-                        system_instruction=f"""你是問答助手。參考資料：
-{pdf_content[:10000]}
-規則：只用資料回答，不知道就說聯繫專員。"""
-                    )
-                )
+        summary = (
+            f"📋 請確認您的預約資訊：\n\n"
+            f"📅 日期：{booking['date']}\n"
+            f"⏰ 時段：{booking['time']}\n"
+            f"👥 人數：{booking['people']}人\n"
+            f"🎹 琴房：{booking['room']}\n\n"
+            f"請輸入「確認」送出預約，或輸入「取消」重新填寫。"
+        )
+        return summary
+    
+    # 確認預約（儲存資料）
+    elif current_state == BookingState.CONFIRMING:
+        if user_message in ["確認", "確定", "送出", "ok", "OK"]:
+            # 儲存預約資料
+            booking_id = save_booking_to_file(user_id, booking)
             
-            if response and response.text:
-                out = response.text
-                logger.info("回應成功")
-            else:
-                out = "無法回答，請聯繫服務專員。"
-                
-        except Exception as e:
-            error_msg = str(e)
-            logger.error(f"錯誤: {error_msg}")
+            summary = (
+                f"✅ 預約成功！\n\n"
+                f"📅 日期：{booking['date']}\n"
+                f"⏰ 時段：{booking['time']}\n"
+                f"👥 人數：{booking['people']}人\n"
+                f"🎹 琴房：{booking['room']}\n"
+                f"📝 預約編號：{booking_id}\n\n"
+                f"我們已收到您的預約，服務專員將盡快與您聯繫確認。\n"
+                f"如有任何問題，歡迎隨時詢問！"
+            )
             
-            if "quota" in error_msg.lower() or "resource_exhausted" in error_msg.lower():
-                out = "系統忙碌，請稍後再試。"
-            elif "429" in error_msg:
-                out = "請求過於頻繁，請稍候。"
-            else:
-                out = "系統錯誤，請聯繫服務專員。"
-  
-        line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
-
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=True)
+            logger.info(f"新預約已儲存：{booking_id}")
+            reset_booking(user_id)
+            return summary
+        else:
+            return "請輸入「確認」來完成預約，或輸入「取消」重新開始。"
+    
+    return None