Update main.py

main.py

@@ -1,9 +1,119 @@
-# 在主程式中加入以下功能
-
-import re
+import json, os, glob, pathlib, time, re
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI, Request, Header, BackgroundTasks, HTTPException, status
+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
 from datetime import datetime, timedelta
 
-# ========== 日期解析與驗證 ==========
+# ============== 設定日誌 ==============
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# ============== API 金鑰檢查 ==============
+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 not files:
+    logger.warning("警告：docs/ 資料夾中找不到 PDF 檔案")
+else:
+    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 i in range(len(pdf_reader.pages)):
+                    page = pdf_reader.pages[i]
+                    pdf_content += page.extract_text()
+            logger.info(f"成功讀取: {filename}")
+        except Exception as e:
+            logger.error(f"讀取 {filename} 失敗: {str(e)}")
+
+# 限制 PDF 內容長度
+MAX_CONTENT_LENGTH = 30000
+if len(pdf_content) > MAX_CONTENT_LENGTH:
+    pdf_content = pdf_content[:MAX_CONTENT_LENGTH]
+
+logger.info(f"PDF 內容總長度: {len(pdf_content)} 字元")
+
+# ============== Gemini 設定 ==============
+system_instruction = (
+    "你是問答助手，使用以下參考資料回答問題。\n"
+    "規則：\n"
+    "1. 只用提供的資料回答\n"
+    "2. 不知道就說「請聯繫服務專員」\n"
+    "3. 答案簡潔，必要時用列表\n\n"
+    f"參考資料：\n{pdf_content}"
+)
+
+thinking_config = genai.types.ThinkingConfig(thinking_budget=0)
+generation_config = genai.types.GenerateContentConfig(
+    max_output_tokens=1000,
+    temperature=0.1,
+    top_p=0.2,
+    thinking_config=thinking_config,
+    system_instruction=system_instruction
+)
+
+# ============== LINE Bot 設定 ==============
+line_bot_api = LineBotApi(os.getenv("CHANNEL_ACCESS_TOKEN"))
+line_handler = WebhookHandler(os.getenv("CHANNEL_SECRET"))
+working_status = os.getenv("DEFALUT_TALKING", default="true").lower() == "true"
+
+# ============== 請求速率控制 ==============
+last_request_time = {}
+REQUEST_COOLDOWN = 2
+
+# ============== 預約系統 ==============
+user_booking_state = {}
+
+class BookingState:
+    """預約狀態管理"""
+    IDLE = "idle"
+    ASKING_DATE = "asking_date"
+    ASKING_TIME = "asking_time"
+    ASKING_PEOPLE = "asking_people"
+    ASKING_ROOM = "asking_room"
+    CONFIRMING = "confirming"
+
+AVAILABLE_ROOMS = ["A琴房", "B琴房", "C琴房", "D琴房", "任意"]
+
+def init_user_booking(user_id):
+    """初始化使用者的預約資料"""
+    user_booking_state[user_id] = {
+        "state": BookingState.IDLE,
+        "date": None,
+        "time": None,
+        "people": None,
+        "room": None,
+        "last_update": time.time()
+    }
+
+def get_user_booking(user_id):
+    """取得使用者的預約狀態"""
+    if user_id not in user_booking_state:
+        init_user_booking(user_id)
+    return user_booking_state[user_id]
+
+def reset_booking(user_id):
+    """重置使用者的預約狀態"""
+    init_user_booking(user_id)
+
+def is_booking_keyword(text):
+    """判斷是否為預約相關關鍵字"""
+    keywords = ["預約", "預定", "訂位", "訂房", "訂琴房", "借琴房", "租琴房", "我要預約", "想預約"]
+    return any(keyword in text for keyword in keywords)
+
 def parse_date(date_str):
     """解析並驗證日期"""
     today = datetime.now()
@@ -33,68 +143,56 @@ def parse_date(date_str):
                 
                 # 檢查是否為過去的日期
                 if date.date() < today.date():
-                    return None, "日期不能是過去的時間，請重新輸入。"
+                    return None, "❌ 日期不能是過去的時間，請重新輸入。\n例如：明天、12/26、2025-12-30"
                 
                 return date.strftime("%Y-%m-%d"), None
             except ValueError:
                 pass
     
-    return None, "日期格式不正確，請使用如：明天、12/26、2025-12-30"
+    return None, "❌ 日期格式不正確，請重新輸入。\n例如：明天、12/26、2025-12-30"
 
-# ========== 時間驗證 ==========
 def validate_time(time_str):
     """驗證時段格式"""
-    # 簡單驗證是否包含時間相關字詞
-    time_keywords = ["點", "時", ":", "-", "到", "至"]
+    time_keywords = ["點", "時", ":", "-", "到", "至", "~"]
     if any(kw in time_str for kw in time_keywords):
         return True, None
-    return False, "時段格式不正確，請使用如：09:00-11:00 或 下午2點到4點"
+    return False, "❌ 時段格式不正確，請重新輸入。\n例如：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人
+        if 1 <= count <= 10:
             return count, None
         else:
-            return None, "人數必須在 1-10 人之間，請重新輸入。"
-    return None, "請輸入有效的人數，例如：2人、3"
-
-# ========== 琴房驗證 ==========
-AVAILABLE_ROOMS = ["A琴房", "B琴房", "C琴房", "D琴房", "任意"]
+            return None, "❌ 人數必須在 1-10 人之間，請重新輸入。"
+    return None, "❌ 請輸入有效的人數。\n例如：2人、3、5位"
 
 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)}"
+    return None, f"❌ 請選擇有效的琴房：{', '.join(AVAILABLE_ROOMS)}"
 
-# ========== 將預約資料存入檔案 ==========
 def save_booking_to_file(user_id, booking_data):
     """將預約資料儲存到 JSON 檔案"""
-    import json
-    from pathlib import Path
-    
-    # 建立 bookings 資料夾
-    bookings_dir = Path("bookings")
+    bookings_dir = pathlib.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)
+            try:
+                bookings = json.load(f)
+            except json.JSONDecodeError:
+                bookings = []
     else:
         bookings = []
     
@@ -107,7 +205,7 @@ def save_booking_to_file(user_id, booking_data):
         "people": booking_data["people"],
         "room": booking_data["room"],
         "created_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-        "status": "pending"  # pending, confirmed, cancelled
+        "status": "pending"
     }
     
     bookings.append(booking_record)
@@ -116,63 +214,82 @@ def save_booking_to_file(user_id, booking_data):
     with open(booking_file, 'w', encoding='utf-8') as f:
         json.dump(bookings, f, ensure_ascii=False, indent=2)
     
+    logger.info(f"預約已儲存: {booking_record['booking_id']}")
     return booking_record["booking_id"]
 
-# ========== 修改後的 handle_booking_flow 函數 ==========
-def handle_booking_flow_enhanced(user_id, user_message):
-    """處理預約流程（加強版）"""
+def handle_booking_flow(user_id, user_message):
+    """處理預約流程"""
     booking = get_user_booking(user_id)
     current_state = booking["state"]
     
-    # 超時重置
+    # 檢查是否超時（10分鐘無互動則重置）
     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 ["取消", "取消預約", "重來"]:
+    if user_message in ["取消", "取消預約", "重來", "重新開始"]:
         reset_booking(user_id)
-        return "已取消預約流程。如需重新預約，請輸入「預約」。"
+        return "✅ 已取消預約流程。\n\n如需重新預約，請輸入「預約」。"
     
-    # 開始預約
+    # 開始預約流程
     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(輸入「取消」可取消預約)"
+            return (
+                "🎹 歡迎使用琴房預約系統！\n\n"
+                "📅 請問您想要預約哪一天呢？\n"
+                "例如：明天、12/26、2025/12/30\n\n"
+                "💡 隨時輸入「取消」可取消預約"
+            )
         else:
-            return None
+            return None  # 不是預約，交給 AI 處理
     
-    # 詢問日期（加入驗證）
+    # 詢問日期
     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點"
+        return (
+            f"✅ 預約日期：{parsed_date}\n\n"
+            f"⏰ 請問您想要預約哪個時段呢？\n"
+            f"例如：09:00-11:00、下午2點到4點、14:00-16:00"
+        )
     
-    # 詢問時段（加入驗證）
+    # 詢問時段
     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人"
+        return (
+            f"✅ 預約時段：{user_message}\n\n"
+            f"👥 請問有幾位使用呢？\n"
+            f"例如：1人、2人、3位"
+        )
     
-    # 詢問人數（加入驗證）
+    # 詢問人數
     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)}"
+        return (
+            f"✅ 使用人數：{people_count}人\n\n"
+            f"🎹 請問您想使用哪個琴房呢？\n"
+            f"可選擇：{', '.join(AVAILABLE_ROOMS)}\n\n"
+            f"💡 如不指定可輸入「任意」"
+        )
     
-    # 詢問琴房（加入驗證）
+    # 詢問琴房
     elif current_state == BookingState.ASKING_ROOM:
         room, error = validate_room(user_message)
         if error:
@@ -181,36 +298,195 @@ def handle_booking_flow_enhanced(user_id, user_message):
         booking["state"] = BookingState.CONFIRMING
         
         summary = (
-            f"📋 請確認您的預約資訊：\n\n"
+            f"📋 請確認您的預約資訊：\n"
+            f"{'='*25}\n"
             f"📅 日期：{booking['date']}\n"
             f"⏰ 時段：{booking['time']}\n"
             f"👥 人數：{booking['people']}人\n"
-            f"🎹 琴房：{booking['room']}\n\n"
-            f"請輸入「確認」送出預約，或輸入「取消」重新填寫。"
+            f"🎹 琴房：{booking['room']}\n"
+            f"{'='*25}\n\n"
+            f"✅ 請輸入「確認」送出預約\n"
+            f"❌ 輸入「取消」重新填寫"
         )
         return summary
     
-    # 確認預約（儲存資料）
+    # 確認預約
     elif current_state == BookingState.CONFIRMING:
-        if user_message in ["確認", "確定", "送出", "ok", "OK"]:
+        if user_message in ["確認", "確定", "送出", "ok", "OK", "沒錯", "正確"]:
             # 儲存預約資料
-            booking_id = save_booking_to_file(user_id, booking)
-            
-            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"如有任何問題，歡迎隨時詢問！"
-            )
-            
-            logger.info(f"新預約已儲存：{booking_id}")
-            reset_booking(user_id)
-            return summary
+            try:
+                booking_id = save_booking_to_file(user_id, booking)
+                
+                summary = (
+                    f"🎉 預約成功！\n"
+                    f"{'='*25}\n"
+                    f"📅 日期：{booking['date']}\n"
+                    f"⏰ 時段：{booking['time']}\n"
+                    f"👥 人數：{booking['people']}人\n"
+                    f"🎹 琴房：{booking['room']}\n"
+                    f"📝 預約編號：{booking_id}\n"
+                    f"{'='*25}\n\n"
+                    f"✅ 我們已收到您的預約！\n"
+                    f"服務專員將盡快與您聯繫確認。\n\n"
+                    f"如有任何問題，歡迎隨時詢問！"
+                )
+                
+                # 重置狀態
+                reset_booking(user_id)
+                return summary
+            except Exception as e:
+                logger.error(f"儲存預約失敗: {str(e)}")
+                reset_booking(user_id)
+                return "❌ 預約儲存失敗，請稍後再試或聯繫服務專員。"
         else:
             return "請輸入「確認」來完成預約，或輸入「取消」重新開始。"
     
-    return None
+    return None
+
+# ============== FastAPI 應用 ==============
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.get("/")
+def root():
+    return {
+        "title": "Line Bot with Booking System",
+        "status": "running",
+        "version": "2.0",
+        "pdf_loaded": len(pdf_content) > 0,
+        "pdf_length": len(pdf_content),
+        "active_bookings": len(user_booking_state),
+        "features": ["AI Q&A", "Booking System", "PDF Knowledge Base"]
+    }
+
+@app.get("/health")
+def health_check():
+    """健康檢查端點"""
+    return {
+        "status": "healthy",
+        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "pdf_loaded": len(pdf_content) > 0,
+        "api_configured": bool(GOOGLE_API_KEY)
+    }
+
+@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:
+            bookings = json.load(f)
+        return {"total": len(bookings), "bookings": bookings}
+    return {"total": 0, "bookings": []}
+
+@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:
+        logger.error("無效的簽章")
+        raise HTTPException(status_code=400, detail="Invalid signature")
+    return "ok"
+
+@line_handler.add(MessageEvent, message=TextMessage)
+def handle_message(event):
+    global working_status
+    
+    if event.type != "message" or event.message.type != "text":
+        return
+        
+    user_message = event.message.text.strip()
+    user_id = event.source.user_id
+    
+    logger.info(f"收到訊息 - 使用者: {user_id[:8]}..., 內容: {user_message[:50]}")
+    
+    # 特殊指令
+    if user_message == "再見":
+        line_bot_api.reply_message(
+            event.reply_token,
+            TextSendMessage(text="👋 Bye! 期待下次為您服務！")
+        )
+        return
+    
+    # 速率限制
+    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
+    
+    # ============== 優先處理預約流程 ==============
+    booking_response = handle_booking_flow(user_id, user_message)
+    
+    if booking_response:
+        # 預約流程有回應，直接返回
+        line_bot_api.reply_message(
+            event.reply_token,
+            TextSendMessage(text=booking_response)
+        )
+        return
+    
+    # ============== 一般問答（使用 Gemini） ==============
+    if working_status:
+        try: 
+            logger.info(f"呼叫 Gemini API - 使用者提問: {user_message[:50]}")
+            
+            response = client.models.generate_content(
+                model="gemini-1.5-flash",
+                contents=user_message,
+                config=generation_config
+            )
+            
+            if response and response.text:
+                out = response.text
+                logger.info(f"AI 回應成功 - 長度: {len(out)} 字元")
+            else:
+                out = "抱歉，我無法回答這個問題。請聯繫服務專員。"
+                logger.warning("Gemini 回應為空")
+                
+        except Exception as e:
+            error_msg = str(e)
+            logger.error(f"Gemini API 錯誤: {error_msg}")
+            
+            if "quota" in error_msg.lower() or "limit" in error_msg.lower() or "resource_exhausted" in error_msg.lower():
+                out = "⚠️ 系統忙碌中，請稍後再試或聯繫服務專員。"
+            elif "429" in error_msg:
+                out = "⚠️ 請求過於頻繁，請稍後再試。"
+            elif "key" in error_msg.lower() or "auth" in error_msg.lower():
+                out = "⚠️ 系統設定錯誤，請聯繫服務專員。"
+            else:
+                out = "⚠️ 系統暫時無法處理，請聯繫服務專員。"
+  
+        line_bot_api.reply_message(
+            event.reply_token,
+            TextSendMessage(text=out)
+        )
+
+if __name__ == "__main__":
+    import uvicorn
+    logger.info("="*50)
+    logger.info("啟動琴房預約 LINE Bot")
+    logger.info(f"PDF 內容長度: {len(pdf_content)} 字元")
+    logger.info(f"可用琴房: {', '.join(AVAILABLE_ROOMS)}")
+    logger.info("="*50)
+    uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=True)