Update app.py

app.py

@@ -1,34 +1,39 @@
 import os
 import threading
-from flask import Flask, request, jsonify
+from fastapi import FastAPI, Request
+from pydantic import BaseModel
 from linebot import LineBotApi, WebhookHandler
 from linebot.models import MessageEvent, TextMessage, TextSendMessage
 from model import classify_esi  # นำเข้าโมเดลจาก model.py
 
-# ✅ โหลด Environment Variables
+# โหลด Environment Variables
 LINE_ACCESS_TOKEN = os.getenv("LINE_ACCESS_TOKEN")
 LINE_CHANNEL_SECRET = os.getenv("LINE_CHANNEL_SECRET")
 
-# ✅ ตั้งค่า Flask App
-app = Flask(__name__)
+# ✅ ตั้งค่า FastAPI แทน Flask
+app = FastAPI()
 line_bot_api = LineBotApi(LINE_ACCESS_TOKEN)
 handler = WebhookHandler(LINE_CHANNEL_SECRET)
 
-@app.route("/")
+# ✅ Route หน้าแรก
+@app.get("/")
 def home():
-    return "DeepSeek AI ESI Classifier is running on Hugging Face Space!"
-
-@app.route("/webhook", methods=["POST"])
-def webhook():
-    signature = request.headers.get("X-Line-Signature", "No Signature")
-    body = request.get_data(as_text=True)
-
-    if not signature:
-        return "Missing Signature", 400
-
-    threading.Thread(target=handler.handle, args=(body, signature)).start()
-    return "OK", 200
-
+    return {"message": "DeepSeek AI ESI Classifier is running on Hugging Face Space!"}
+
+# ✅ Webhook สำหรับรับข้อความจาก LINE
+class LineRequest(BaseModel):
+    signature: str
+    body: str
+
+@app.post("/webhook")
+async def webhook(request: LineRequest):
+    if not request.signature:
+        return {"error": "Missing Signature"}
+    
+    threading.Thread(target=handler.handle, args=(request.body, request.signature)).start()
+    return {"status": "OK"}
+
+# ✅ ฟังก์ชันตอบกลับ LINE
 @handler.add(MessageEvent, message=TextMessage)
 def handle_message(event):
     user_message = event.message.text
@@ -39,12 +44,13 @@ def handle_message(event):
     elif esi_level == 3:
         response_text = f"🩺 คุณควรพบแพทย์เร็วที่สุดเพื่อตรวจสอบเพิ่มเติม (ESI {esi_level})"
     elif esi_level in [4, 5]:
-        response_text = f"💊 อาการของคุณสามารถรอพบแพทย์ที่ OPD ได้ (ESI {esi_level})"
+        response_text = f"💊 อาการของคุณสามารถรอพบแพทย์ที่ OPD ในวันพรุ่งนี้เช้าได้ (ESI {esi_level})"
     else:
         response_text = "❌ ไม่สามารถประเมินได้ กรุณาลองอธิบายอาการให้ละเอียดขึ้น"
 
     threading.Thread(target=lambda: line_bot_api.reply_message(event.reply_token, TextSendMessage(text=response_text))).start()
 
-# ✅ รันแอป
+# ✅ รันแอปที่ Port 7860 (ค่าเริ่มต้นของ Hugging Face Space)
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)