Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import re
+from typing import Optional
 
 import openai
 from dotenv import load_dotenv
@@ -14,7 +15,6 @@ from pydantic import BaseModel
 load_dotenv()
 api_key = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 
-# You can change the model later if needed.
 model_link = "meta-llama/Meta-Llama-3-8B-Instruct"
 base_url = "https://router.huggingface.co/v1"
 
@@ -27,7 +27,12 @@ class Message(BaseModel):
 
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
-    return """Welcome to Up to 12 Chat Processor"""
+    return "Welcome to Up to 12 Chat Processor"
+
+
+@app.get("/health")
+async def health():
+    return {"ok": True, "service": "up"}
 
 
 # =========================
@@ -163,7 +168,7 @@ def detect_intent(text: str) -> str:
     ):
         return "HUMAN_AGENT"
 
-    # 2) Specific course types first (more precise than general courses menu)
+    # 2) Specific course types first
     if any(x in t for x in ["online", "اونلاين", "اون لاين", "zoom", "زووم", "من البيت"]):
         return "ONLINE_COURSES"
 
@@ -227,7 +232,7 @@ def detect_intent(text: str) -> str:
     return "OTHER"
 
 
-def detect_course_subtype(text: str) -> str | None:
+def detect_course_subtype(text: str) -> Optional[str]:
     t = normalize_arabic(text)
 
     if any(x in t for x in ["express", "اكسبريس", "سريع", "super intensive"]):
@@ -340,8 +345,6 @@ def sanitize_reply(text: str) -> str:
         "يهتم بيك": "تحب",
         "اقترحك": "أقترح لك",
         "يمكنني": "أقدر",
-        "سأقوم": "ه",
-        "سوف": "ه",
         "العميل": "إنت",
         "المستخدم": "إنت",
     }
@@ -349,21 +352,16 @@ def sanitize_reply(text: str) -> str:
     for old, new in replacements.items():
         text = text.replace(old, new)
 
-    # Remove bullets spam
     text = re.sub(r"[•\-]{2,}", "-", text)
 
-    # Remove extra empty lines
     lines = [line.strip() for line in text.splitlines() if line.strip()]
     lines = lines[:5]
-
     text = "\n".join(lines)
 
-    # Avoid too much English-heavy reply
     english_words = re.findall(r"[A-Za-z]{4,}", text)
     if len(english_words) > 8:
         return fallback_customer_service_reply()
 
-    # Prevent invented addresses / prices / schedules if model sneaks them in
     blocked_patterns = [
         r"\b\d+\s*جنيه\b",
         r"\b\d+\s*egp\b",
@@ -371,9 +369,11 @@ def sanitize_reply(text: str) -> str:
         r"\bالمعاد\b",
         r"\bالساعة\b",
         r"\bالسعر\b",
+        r"\bالاسعار\b",
         r"\bالأسعار\b",
         r"\bفرع\s+\w+",
     ]
+
     lowered = normalize_arabic(text)
     for pattern in blocked_patterns:
         if re.search(pattern, lowered, flags=re.IGNORECASE):
@@ -497,13 +497,11 @@ def process_text(user_text: str):
 
     intent = detect_intent(user_text)
 
-    # 1) Rule-based first for known intents
     answer = generate_rule_based_reply(intent, user_text)
     if answer:
         answer = sanitize_reply(answer)
         return {"ok": True, "intent": intent, "reply": answer}
 
-    # 2) LLM only for unknown / out-of-scope style messages
     llm_answer = process_with_llm(user_text)
     return {"ok": True, "intent": intent, "reply": llm_answer}
 
@@ -514,7 +512,14 @@ def process_text(user_text: str):
 
 @app.post("/processtext")
 async def receive_updates(request: Request):
-    data = await request.json()
+    try:
+        data = await request.json()
+    except Exception:
+        return JSONResponse(
+            content={"ok": False, "error": "Invalid JSON payload"},
+            status_code=400
+        )
+
     print("Received Update:", data)
 
     result = process_text(data.get("message", ""))