Update app.py

app.py

@@ -26,7 +26,7 @@ handler = WebhookHandler(LINE_CHANNEL_SECRET)
 
 @app.get("/")
 async def root():
-    return {"message": "Triage Bot is running (Thai NLP + fuzzy + triage level)"}
+    return {"message": "Triage Bot with improved matching and response."}
 
 @app.post("/webhook")
 async def webhook(request: Request, x_line_signature: str = Header(None)):
@@ -65,33 +65,24 @@ def reply_message(reply_token, message):
     except Exception as e:
         logger.error(f"Send reply error: {e}")
 
-# ------------------ Emergency Keywords ------------------
+# ------------------ Emergency Keywords & Levels ------------------
+
+# คำที่สั้นเกินไป / ไม่ชัดเจน
+UNCLEAR_TOKENS = ["เจ็บ", "เวียน", "หน้ามืด", "ปวด", "ใจสั่น", "ตึง", "แน่น", "เมื่อย", "มึน"]
 
 EMERGENCY_SYMPTOMS = {
-    "หมดสติ / ไม่รู้สึกตัว": ["หมดสติ", "ไม่รู้สึกตัว", "เป็นลม", "ล้มแล้วไม่ตื่น", "ไม่ตอบสนอง",
-                               "หลับไม่รู้เรื่อง", "ไม่ขยับ", "ปลุกไม่ตื่น", "เหม่อลอย", "สลบ"],
-    "หายใจผิดปกติ": ["หายใจเร็ว", "หายใจเหนื่อย", "หายใจลำบาก", "หายใจติดขัด", "หอบ",
-                      "หายใจเสียงดัง", "แน่นหน้าอก", "หายใจไม่ทัน", "อึดอัด", "หายใจถี่"],
-    "เจ็บหน้าอก": ["เจ็บหน้าอก", "แน่นหน้าอก", "จุกหน้าอก", "เสียวหน้าอก", "ปวดอก",
-                   "บีบหน้าอก", "เจ็บลึกๆในอก", "อกแน่น", "อกอึดอัด", "หน้าอกปวดจี๊ด"],
-    "ชัก / หยุดหายใจ": ["ชัก", "เกร็ง", "หยุดหายใจ", "หมดลม", "ตัวสั่น",
-                        "ชักกระตุก", "หมดแรง", "ดิ้น", "ลืมตาไม่ขึ้น", "ตัวแข็ง"],
-    "อัมพฤกษ์ / พูดไม่ชัด": ["อัมพฤกษ์", "อัมพาต", "พูดไม่ชัด", "หน้าเบี้ยว", "แขนขาอ่อนแรง",
-                             "ยิ้มไม่เท่ากัน", "ยกแขนไม่ขึ้น", "เดินเซ", "ลิ้นแข็ง", "แขนขาไม่มีแรง"],
-    "เลือดออก / บาดแผล": ["เลือดออก", "เลือดไม่หยุด", "เลือดไหล", "แผลฉกรรจ์", "แผลลึก",
-                          "เลือดพุ่ง", "เลือดเปื้อนตัว", "แผลเปิด", "เลือดซึมตลอด", "แผลทะลุ"],
-    "กระดูกหัก": ["กระดูกหัก", "แขนหัก", "ขาหัก", "นิ้วหัก", "บวมผิดรูป",
-                 "ขางอผิดรูป", "ขาใช้การไม่ได้", "ขยับแล้วเจ็บมาก", "ปูด", "โค้งผิดปกติ"],
-    "อุบัติเหตุรุนแรง": ["รถชน", "รถล้ม", "ตกจากที่สูง", "หัวฟาด", "อุบัติเหตุแรง",
-                       "รถคว่ำ", "ล้มแรง", "โดนกระแทกแรง", "ลื่นหัวฟาด", "โดนรถชน"],
-    "เด็กมีไข้ / ชัก": ["ไข้สูง", "เด็กชัก", "ลูกชัก", "ชักตอนมีไข้", "ตัวร้อนจัด",
-                      "ไข้ขึ้นสูง", "ตาค้าง", "เด็กไม่ตอบสนอง", "มือแข็ง", "สั่นทั้งตัว"],
-    "ภาวะช็อก / หัวใจเต้นผิดจังหวะ": ["ช็อก", "ความดันต่ำ", "ใจสั่น", "หัวใจเต้นผิดจังหวะ", "ชีพจรเร็ว",
-                                  "หมดแรงกะทันหัน", "ตัวเย็น", "เหงื่อแตก", "ใจเต้นแรง", "แน่นท้องหายใจไม่ออก"],
-    "ได้รับสารพิษ / ยาเกินขนาด / สัตว์มีพิษกัด": ["สารพิษ", "กินยาเกิน", "ยาเกินขนาด", "พิษ", "ถูกงูกัด",
-                                               "แมงป่องต่อย", "แมลงกัด", "แพ้ยา", "แพ้อาหารรุนแรง", "คลื่นไส้อาเจียนมาก"],
-    "ผู้ป่วยตั้งครรภ์ใกล้คลอด": ["ตั้งครรภ์", "เจ็บท้องคลอด", "น้ำเดิน", "มูกเลือด", "ปวดท้องใกล้คลอด",
-                              "ท้องแข็ง", "มดลูกบีบ", "ปวดหน่วง", "เจ็บท้องถี่", "ลูกดิ้นแรงผิดปกติ"]
+    "หมดสติ / ไม่รู้สึกตัว": ["หมดสติ", "ไม่รู้สึกตัว", "สลบ", "วูบ", "หน้ามืด", "ไม่ตอบสนอง", "ล้มแล้วไม่รู้ตัว"],
+    "หายใจผิดปกติ": ["หายใจไม่ออก", "หายใจไม่ทัน", "หายใจลำบาก", "หอบ", "แน่นหน้าอก", "เหนื่อยหอบ", "หายใจติดขัด"],
+    "เจ็บหน้าอก": ["เจ็บหน้าอก", "แน่นในอก", "จุกกลางอก", "เจ็บร้าวไปหลัง", "ปวดหน้าอก", "แน่นจุก"],
+    "ชัก / หยุดหายใจ": ["ชัก", "เกร็ง", "กระตุก", "หยุดหายใจ", "ดิ้น", "ตาค้าง", "หมดลม"],
+    "อัมพฤกษ์ / พูดไม่ชัด": ["อัมพฤกษ์", "หน้าเบี้ยว", "พูดไม่ได้", "พูดไม่รู้เรื่อง", "แขนขาไม่มีแรง", "ยิ้มไม่เท่ากัน", "เดินเซ"],
+    "เลือดออก / บาดแผล": ["เลือดออก", "เลือดไม่หยุด", "แผลลึก", "แผลฉกรรจ์", "เลือดพุ่ง", "แผลเปิด", "เลือดซึม"],
+    "กระดูกหัก": ["หัก", "กระดูกหัก", "บวมผิดรูป", "งอผิดรูป", "ขยับไม่ได้", "ปูด", "ผิดรูป"],
+    "อุบัติเหตุรุนแรง": ["รถชน", "รถล้ม", "ตกจากที่สูง", "หัวฟาด", "โดนรถชน", "แรงกระแทก", "อุบัติเหตุรุนแรง"],
+    "เด็กมีไข้ / ชัก": ["ไข้สูง", "เด็กชัก", "ตัวร้อนจัด", "ไข้แล้วชัก", "ลูกชัก", "ไข้ขึ้น", "ตาค้าง", "เด็กไม่ตอบสนอง"],
+    "ภาวะช็อก / หัวใจเต้นผิดจังหวะ": ["ช็อก", "ใจสั่น", "ชีพจรเร็ว", "เหงื่อแตก", "ตัวเย็น", "หมดแรงกะทันหัน"],
+    "ได้รับสารพิษ / ยาเกินขนาด / สัตว์มีพิษกัด": ["กินยาเกิน", "ยาเกินขนาด", "โดนงูกัด", "แพ้รุนแรง", "โดนพิษ", "แมงป่องต่อย", "คลื่นไส้อาเจียนมาก"],
+    "ผู้ป่วยตั้งครรภ์ใกล้คลอด": ["น้ำเดิน", "มูกเลือด", "ปวดท้องคลอด", "ท้องแข็ง", "ใกล้คลอด", "เจ็บท้องถี่"]
 }
 
 EMERGENCY_LEVELS = {
@@ -114,6 +105,11 @@ FUZZY_THRESHOLD = 85
 def check_emergency(message):
     tokens = word_tokenize(message.lower(), engine="newmm")
     logger.info(f"Tokenized: {tokens}")
+
+    # ตรวจว่าข้อความอาจไม่ชัดเจนหรือสั้นเกินไป
+    if len(tokens) <= 2 or all(t in UNCLEAR_TOKENS for t in tokens):
+        return "❗ อาการของคุณยังไม่ชัดเจน กรุณาพิมพ์รายละเอียดเพิ่มเติม เช่น ตำแหน่ง ความรุนแรง หรืออาการร่วม"
+
     for label, keywords in EMERGENCY_SYMPTOMS.items():
         for token in tokens:
             for keyword in keywords:
@@ -124,9 +120,10 @@ def check_emergency(message):
                         return f"🚨 อาการของคุณเข้าข่ายฉุกเฉินระดับ **1 (ด่วนมาก)**: {label}\nกรุณารีบไปห้องฉุกเฉินทันที!"
                     elif level == 2:
                         return f"⚠️ อาการของคุณเข้าข่ายฉุกเฉิน���ะดับ **2 (เร่งด่วน)**: {label}\nควรไปโรงพยาบาลโดยเร็วที่สุด"
+
     return "🟢 อาการของคุณไม่เข้าข่ายฉุกเฉิน สามารถเข้ารับบริการที่ห้องตรวจทั่วไปในวันถัดไปได้"
 
-# ---------------------------------------------------------------
+# ------------------------------------------------------------
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)