File size: 5,532 Bytes
cd9bca9 4514838 cd9bca9 4514838 cd9bca9 4514838 cd9bca9 4514838 cd9bca9 4514838 cd9bca9 4514838 cd9bca9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 |
import os
from fastapi import FastAPI, Request, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from linebot import LineBotApi, WebhookHandler
from linebot.exceptions import InvalidSignatureError
from linebot.models import MessageEvent, TextMessage, TextSendMessage
import logging
from backend.services.nlp_service import NLPService
from backend.services.database_service import DatabaseService
from backend.services.message_router import MessageRouter
from backend.config import settings
from backend.database.connection import test_database_connection
# 設定日誌
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
app = FastAPI(title="LINE Bot API", version="1.0.0")
# CORS 設定
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# LINE Bot 初始化
line_bot_api = LineBotApi(settings.LINE_CHANNEL_ACCESS_TOKEN)
handler = WebhookHandler(settings.LINE_CHANNEL_SECRET)
# 服務初始化
nlp_service = NLPService()
db_service = DatabaseService()
message_router = MessageRouter()
@app.get("/")
def greet_json():
return {"Hello": "LINE Bot API is running!"}
@app.get("/health")
def health_check():
# 檢查資料庫連線
db_status = "connected" if test_database_connection() else "disconnected"
return {
"status": "healthy",
"message": "LINE Bot API is operational",
"database": db_status,
"database_url": f"postgresql://{settings.DB_HOST}:{settings.DB_PORT}/{settings.DB_NAME}"
}
@app.post("/webhook")
async def callback(request: Request):
"""LINE Bot Webhook 端點"""
signature = request.headers.get('X-Line-Signature', '')
body = await request.body()
try:
handler.handle(body.decode('utf-8'), signature)
except InvalidSignatureError:
logger.error("Invalid signature")
raise HTTPException(status_code=400, detail="Invalid signature")
return {"status": "ok"}
@handler.add(MessageEvent, message=TextMessage)
def handle_message(event):
"""處理 LINE 訊息 - 使用新的路由系統"""
try:
user_message = event.message.text
user_id = event.source.user_id
logger.info(f"收到訊息: {user_message} from {user_id}")
# 使用訊息路由器處理
response_data = message_router.route_message(user_message, user_id)
# 回覆訊息
line_bot_api.reply_message(
event.reply_token,
TextSendMessage(text=response_data["text"])
)
except Exception as e:
logger.error(f"處理訊息時發生錯誤: {str(e)}")
line_bot_api.reply_message(
event.reply_token,
TextSendMessage(text="抱歉，處理您的訊息時發生錯誤，請稍後再試。")
)
# 新增直接 API 端點
@app.post("/chat")
async def direct_chat(request: dict):
"""直接聊天 API"""
try:
message = request.get("message", "")
user_id = request.get("user_id", "api_user")
if not message:
return {
"success": False,
"error": "訊息內容不能為空"
}
# 強制使用聊天模式
result = message_router._handle_chat_mode(message, user_id)
return result
except Exception as e:
logger.error(f"直接聊天 API 錯誤: {str(e)}")
return {
"success": False,
"error": str(e),
"text": "聊天服務暫時無法使用"
}
@app.post("/search")
async def direct_search(request: dict):
"""直接搜尋 API"""
try:
message = request.get("message", "")
user_id = request.get("user_id", "api_user")
if not message:
return {
"success": False,
"error": "查詢內容不能為空"
}
# 強制使用搜尋模式
result = message_router._handle_search_mode(message, user_id)
return result
except Exception as e:
logger.error(f"直接搜尋 API 錯誤: {str(e)}")
return {
"success": False,
"error": str(e),
"text": "搜尋服務暫時無法使用"
}
@app.post("/route")
async def smart_route(request: dict):
"""智能路由 API"""
try:
message = request.get("message", "")
user_id = request.get("user_id", "api_user")
if not message:
return {
"success": False,
"error": "訊息內容不能為空"
}
# 使用智能路由
result = message_router.route_message(message, user_id)
return result
except Exception as e:
logger.error(f"智能路由 API 錯誤: {str(e)}")
return {
"success": False,
"error": str(e),
"text": "路由服務暫時無法使用"
}
@app.get("/stats")
async def get_stats():
"""取得路由統計"""
try:
stats = message_router.get_route_statistics()
return {
"success": True,
"data": stats
}
except Exception as e:
logger.error(f"統計 API 錯誤: {str(e)}")
return {
"success": False,
"error": str(e)
}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860) |