Update app.py

app.py

@@ -36,96 +36,133 @@ async def receive_updates(request: Request):
     # ضمان إن الرجوع JSON دايمًا
     return JSONResponse(content=result)
 
-    
-# def process_text(user_text):
-#   """
-#   Processes user text using the Meta-Llama-3-8B model and returns the response.
-#   Args:
-#       user_text: The text entered by the user.
-#   Returns:
-#       The response generated by the model.
-#   """
-
-#   # Initialize OpenAI client
-#   client = openai.OpenAI(api_key=api_key, base_url=base_url)
-
-#   try:
-#     # Generate response using OpenAI chat completion API
-#     response = client.chat.completions.create(
-#         model=model_link,
-#         messages=[{"role": "user", "content": user_text}],
-#         max_tokens=3000,
-#         temperature=0.5,  # Adjust temperature for desired response randomness
-#         stream=False  # Disable streaming for function use
-#     )
-
-#     # Handle potential changes in response format
-#     if isinstance(response.choices, list):
-#       # Access response text if choices is a list (likely scenario)
-#       return response.choices[0].message.content.strip()
-#     else:
-#       # Handle potential alternative response format (less likely)
-#       return response.content.strip() if hasattr(response, 'content') else "An error occurred."
-
-#   except Exception as e:
-#     print(f"Error occurred: {e}")
-#     return "An error occurred while processing your request."
+import re
+
+INTENTS = {
+    "GREETING": "تحية/بداية محادثة",
+    "COURSES_MENU": "طلب قائمة الكورسات",
+    "COURSE_TYPE_DETAILS": "سؤال عن نوع كورس محدد",
+    "CENTER_INFO": "سؤال عن المركز",
+    "CHILDREN_COURSES": "كورس أطفال",
+    "ONLINE_COURSES": "كورسات أونلاين",
+    "WEEKEND_COURSES": "كورسات ويكند",
+    "HUMAN_AGENT": "طلب خدمة عملاء/موظف",
+    "OTHER": "غير معروف/خارج النطاق",
+}
+
+def detect_intent(text: str) -> str:
+    t = (text or "").strip().lower()
+
+    # 1) Greetings
+    if re.search(r"\b(hi|hello|hey|start)\b", t) or any(x in t for x in ["اهلا", "أهلا", "السلام", "هاي", "مرحبا", "ابدأ", "ابدء"]):
+        return "GREETING"
+
+    # 2) Human agent
+    if any(x in t for x in ["خدمة العملاء", "موظف", "حد يرد", "اكلم", "اتواصل", "رقم", "واتساب", "support", "agent"]):
+        return "HUMAN_AGENT"
+
+    # 3) Courses menu / asking about courses generally
+    if any(x in t for x in ["كورسات", "دورات", "courses", "الكورسات المتاحة", "عايز كورس", "عايز اتعلم", "الالماني عندكم", "الماني"]):
+        return "COURSES_MENU"
+
+    # 4) Specific course types
+    # Express / Intensive / Regular / Weekend / Online / Children
+    if any(x in t for x in ["express", "اكسبريس", "سريع", "super intensive"]):
+        return "COURSE_TYPE_DETAILS"
+    if any(x in t for x in ["intensive", "انتنسب", "مكثف", "مكثفة"]):
+        return "COURSE_TYPE_DETAILS"
+    if any(x in t for x in ["regular", "ريجولار", "عادي", "منتظم"]):
+        return "COURSE_TYPE_DETAILS"
+    if any(x in t for x in ["weekend", "ويكند", "الجمعة", "السبت", "عطلة"]):
+        return "WEEKEND_COURSES"
+    if any(x in t for x in ["online", "اونلاين", "أونلاين", "زووم", "من البيت"]):
+        return "ONLINE_COURSES"
+    if any(x in t for x in ["children", "kids", "اطفال", "أطفال", "طفل", "سن", "سنين"]):
+        return "CHILDREN_COURSES"
+
+    # 5) Center info
+    if any(x in t for x in ["المركز", "adk", "ädk", "فروع", "branch", "اتأسس", "تاريخ", "goethe", "معتمد"]):
+        return "CENTER_INFO"
+
+    return "OTHER"
+
+KB_TEXT = """
+المركز:
+(ÄDK)- Egyptian-German Cultural Centre was established in 1998. With various branches in Cairo, ÄDK offers German language courses for adults from A1 to C1 and for children from A1 to B1. Over the years thousands have graduated and have learned in ÄDK. In addition to our language courses, ÄDK organizes cultural projects independently, or in conjunction with the Goethe institute.
+ÄDK is the first institute to be accredited from the Goethe institute in the Middle East and North Africa region.
+
+أنواع الكورسات:
+1) Express Courses:
+- Super intensive courses لتعلم الألماني بسرعة وكفاءة. Challenging وتحتاج مجهود.
+- ممكن تحجز Stage كاملة (A1-A2-B1).
+- 3 محاضرات في الأسبوع، كل محاضرة 4 ساعات.
+- مدة الـ stage: شهرين ونصف.
+
+2) Intensive Courses:
+- ممكن تحجز Stage كاملة (A1-A2-B1).
+- محاضرتين في الأسبوع، كل محاضرة 4 ساعات.
+- مدة الـ stage: 3 شهور ونصف.
+
+3) Regular Courses:
+- ممكن تحجز كل level لوحده من (A1.1 إلى B1.3).
+- محاضرتين في الأسبوع، كل محاضرة 4 ساعات.
+- مدة الكورس: شهر ونصف.
+
+4) Weekend Courses:
+- للناس اللي مش فاضيين خلال الأسبوع بسبب شغل/دراسة.
+- بتتبع نفس outline وساعات الـ Regular (intensive و normal).
+
+5) Online Courses:
+- attending online.
+
+6) Children Courses:
+- من 5 سنوات حتى 15 سنة.
+- الهدف: تنمية مهارات الطفل واللغة وتشجيع التفكير الحر والنقدي والاستقلالية.
+"""
+
+def build_system_prompt(intent: str) -> str:
+    return f"""
+أنت مساعد واتساب رسمي ولطيف لمركز ÄDK لتعليم اللغة الألمانية.
+لازم ترد بالعربي المصري بشكل friendly ومختصر وواضح.
+
+قواعد مهمة جدًا:
+- استخدم فقط المعلومات الموجودة داخل "قاعدة المعرفة" أدناه.
+- ممنوع تمامًا تخترع أسعار، مواعيد، فروع بالتفصيل، أماكن، أو أي تفاصيل غير موجودة.
+- ممنوع تتكلم عن المنافسين أو تقارن أو تقدم معلومات عامة عن كورسات الألماني خارج قاعدة المعرفة.
+- لو المستخدم سأل عن حاجة مش موجودة في قاعدة المعرفة: قول له بصراحة إن المعلومة مش متاحة عندك دلوقتي، واقترح "التحدث مع خدمة العملاء".
+- لو الـ intent = GREETING: قدم رسالة ترحيب + خيارات قصيرة.
+- لو الـ intent = COURSES_MENU: اعرض الأنواع المتاحة (Regular/Express/Intensive/Weekend/Online/Children) واسأل يختار نوع.
+- لو الـ intent متعلق بنوع كورس: اعرض تفاصيل النوع من قاعدة المعرفة فقط.
+- الرد يكون نص فقط (بدون JSON)، وجمل قصيرة.
+
+الـ intent الحالي: {intent}
+
+قاعدة المعرفة:
+{KB_TEXT}
+""".strip()
 
 def process_text(user_text: str):
     client = openai.OpenAI(api_key=api_key, base_url=base_url)
 
-    # 1) تصحيح النص
-    try:
-        system_prompt = f"""
-You are a helpful English grammar corrector for A1/A2 CEFR students.
-Correct grammar, spelling, and literal translation mistakes.
-If the input is only a list of words (not sentences), only correct spelling.
-Return ONLY the revised text (no explanations).
-"""
+    intent = detect_intent(user_text)
+    system_prompt = build_system_prompt(intent)
 
+    try:
         resp = client.chat.completions.create(
             model=model_link,
             messages=[
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": user_text},
             ],
-            max_tokens=800,
-            temperature=0.1,
-            stream=False,
-        )
-
-        corrected_text = resp.choices[0].message.content.strip()
-        print("After correction:", corrected_text)
-        print("--------------------------------------")
-
-    except Exception as e:
-        # مهم جدًا: رجّع نص error مش object
-        return {"ok": False, "when": "During correcting sentences", "error": str(e)}
-
-    # 2) استخراج الأخطاء (diff)
-    try:
-        mistakes_prompt = f"""
-Compare the user text and the corrected text, and list EVERY correction made.
-Return ONLY a list of mistakes/corrections (no revised text).
-
-User text: "{user_text}"
-Corrected text: "{corrected_text}"
-"""
-
-        resp2 = client.chat.completions.create(
-            model=model_link,
-            messages=[
-                {"role": "system", "content": "You are a grammar checker bot."},
-                {"role": "user", "content": mistakes_prompt},
-            ],
-            max_tokens=800,
-            temperature=0,
+            max_tokens=500,
+            temperature=0.2,
             stream=False,
         )
 
-        mistakes = resp2.choices[0].message.content.strip()
+        answer = resp.choices[0].message.content.strip()
 
-        return {"ok": True, "Corrected_text": corrected_text, "Mistakes": mistakes}
+        # لو تحب ترجع intent كمان (مفيد للـ webhook logic)
+        return {"ok": True, "intent": intent, "reply": answer}
 
     except Exception as e:
-        return {"ok": False, "when": "During checking mistakes", "error": str(e), "Corrected_text": corrected_text}
+        return {"ok": False, "error": str(e)}