Update app.py

app.py

@@ -1,42 +1,38 @@
-import openai
 import os
-from dotenv import load_dotenv, dotenv_values
+import re
+
+import openai
+from dotenv import load_dotenv
 from fastapi import FastAPI, Request
+from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
-from fastapi.responses import RedirectResponse, HTMLResponse
-from fastapi.responses import JSONResponse
-import urllib.parse
-import requests
 
-# Load environment variables (assuming your API key is stored in a `.env` file)
+# =========================
+# Environment / Model Setup
+# =========================
+
 load_dotenv()
-api_key = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
+api_key = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 
-# OpenAI API configuration (specific to Meta-Llama-3-8B)
+# You can change the model later if needed.
 model_link = "meta-llama/Meta-Llama-3-8B-Instruct"
 base_url = "https://router.huggingface.co/v1"
 
 app = FastAPI()
 
+
 class Message(BaseModel):
     message: str
 
+
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
-        return """Welcome to Up to 12 Chat Processor"""
-    
-@app.post("/processtext")
-async def receive_updates(request: Request):
-    data = await request.json()
-    print("Received Update:", data)
+    return """Welcome to Up to 12 Chat Processor"""
 
-    result = process_text(data.get("message", ""))
-    print("Assistant:", result)
-
-    # ضمان إن الرجوع JSON دايمًا
-    return JSONResponse(content=result)
 
-import re
+# =========================
+# Intents
+# =========================
 
 INTENTS = {
     "GREETING": "تحية/بداية محادثة",
@@ -50,140 +46,478 @@ INTENTS = {
     "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"
+# =========================
+# Knowledge Base
+# =========================
+
+KB = {
+    "center": {
+        "name": "ÄDK - Egyptian-German Cultural Centre",
+        "established": "1998",
+        "branches": "له فروع مختلفة في القاهرة",
+        "adults_levels": "للكبار من A1 لحد C1",
+        "children_levels": "للأطفال من A1 لحد B1",
+        "goethe": "أول معهد معتمد من Goethe Institute في الشرق الأوسط وشمال أفريقيا",
+    },
+    "courses": {
+        "express": {
+            "title": "Express Courses",
+            "details": [
+                "كورسات Super intensive لتعلم الألماني بسرعة وكفاءة",
+                "النظام ده Challenging وبيحتاج مجهود",
+                "ممكن تحجز Stage كاملة (A1 - A2 - B1)",
+                "3 محاضرات في الأسبوع",
+                "كل محاضرة 4 ساعات",
+                "مدة الـ stage شهرين ونص",
+            ],
+        },
+        "intensive": {
+            "title": "Intensive Courses",
+            "details": [
+                "ممكن تحجز Stage كاملة (A1 - A2 - B1)",
+                "محاضرتين في الأسبوع",
+                "كل محاضرة 4 ساعات",
+                "مدة الـ stage 3 شهور ونص",
+            ],
+        },
+        "regular": {
+            "title": "Regular Courses",
+            "details": [
+                "ممكن تحجز كل level لوحده من A1.1 لحد B1.3",
+                "محاضرتين في الأسبوع",
+                "كل محاضرة 4 ساعات",
+                "مدة الكورس شهر ونص",
+            ],
+        },
+        "weekend": {
+            "title": "Weekend Courses",
+            "details": [
+                "مناسبة للناس اللي مش فاضيين خلال الأسبوع بسبب شغل أو دراسة",
+                "بتتبع نفس outline وساعات الـ Regular",
+            ],
+        },
+        "online": {
+            "title": "Online Courses",
+            "details": [
+                "متاحة بنظام أونلاين",
+                "الحضور بيكون online",
+            ],
+        },
+        "children": {
+            "title": "Children Courses",
+            "details": [
+                "من سن 5 سنين لحد 15 سنة",
+                "الهدف منها تنمية مهارات الطفل واللغة",
+                "وكمان تشجيع التفكير الحر والنقدي والاستقلالية",
+            ],
+        },
+    },
+}
+
+
+# =========================
+# Helpers
+# =========================
+
+def normalize_arabic(text: str) -> str:
+    if not text:
+        return ""
+
+    text = text.strip().lower()
+
+    replacements = {
+        "أ": "ا",
+        "إ": "ا",
+        "آ": "ا",
+        "ة": "ه",
+        "ى": "ي",
+        "ؤ": "و",
+        "ئ": "ي",
+    }
 
-    # 2) Human agent
-    if any(x in t for x in ["خدمة العملاء", "موظف", "حد يرد", "اكلم", "اتواصل", "رقم", "واتساب", "support", "agent"]):
+    for old, new in replacements.items():
+        text = text.replace(old, new)
+
+    return text
+
+
+def detect_intent(text: str) -> str:
+    t = normalize_arabic(text)
+
+    # 1) Human agent first
+    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"
+    # 2) Specific course types first (more precise than general courses menu)
+    if any(x in t for x in ["online", "اونلاين", "اون لاين", "zoom", "زووم", "من البيت"]):
+        return "ONLINE_COURSES"
+
+    if any(x in t for x in ["weekend", "ويكند", "الجمعة", "السبت", "عطله", "عطلة"]):
+        return "WEEKEND_COURSES"
+
+    if any(x in t for x in ["children", "kids", "اطفال", "طفل", "سن", "سنين"]):
+        return "CHILDREN_COURSES"
 
-    # 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", "انتنسب", "مكثف", "مكثفة"]):
+
+    if any(x in t for x in ["intensive", "مكثف", "مكثفه"]):
         return "COURSE_TYPE_DETAILS"
-    if any(x in t for x in ["regular", "ريجولار", "عادي", "منتظم"]):
+
+    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", "معتمد"]):
+    # 3) Center info
+    if any(
+        x in t
+        for x in [
+            "المركز",
+            "adk",
+            "ädk",
+            "فروع",
+            "branch",
+            "اتاسس",
+            "اتأسس",
+            "تاريخ",
+            "goethe",
+            "معتمد",
+        ]
+    ):
         return "CENTER_INFO"
 
+    # 4) General courses
+    if any(
+        x in t
+        for x in [
+            "كورسات",
+            "دورات",
+            "courses",
+            "الكورسات المتاحه",
+            "الكورسات المتاحة",
+            "عايز كورس",
+            "عايز اتعلم",
+            "الماني عندكم",
+            "ألماني",
+            "الماني",
+        ]
+    ):
+        return "COURSES_MENU"
+
+    # 5) Greeting
+    if re.search(r"\b(hi|hello|hey|start)\b", t) or any(
+        x in t for x in ["اهلا", "السلام", "هاي", "مرحبا", "ابدأ", "ابدء", "مسا", "صباح الخير"]
+    ):
+        return "GREETING"
+
     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"""
-You are the official WhatsApp assistant for ÄDK (Egyptian-German Cultural Centre).
-
-OUTPUT LANGUAGE:
-- Reply in Egyptian Arabic (عامية مصرية) using short, natural sentences.
-- Do NOT use formal greetings like "مرحباً" or "أهلاً". Reply directly to the user's message.
-
-STYLE RULES (must follow):
-- Do NOT produce Arabic grammar mistakes such as "اقترحك" or "يهتم بيك".
-  Use: "أقترح" / "أنسب" / "تحب" / "ممكن".
-- Use the second-person pronoun consistently: use "إنت" (not حضرتك).
-- Ask choices correctly: "إنت مهتم بأي نوع كورس؟" / "تحب تعرف تفاصيل أنهي نوع؟"
-  Never say: "يهتم بيك".
-- Max 5 lines. No JSON. No headings. No bullet spam.
-
-SCOPE RULES:
-- Use ONLY the information inside KNOWLEDGE BASE below.
-- Never invent prices, schedules, branch addresses, locations, or any details not in KB.
-- Never compare competitors or provide general German-course information outside KB.
-
-OUT-OF-SCOPE HANDLING:
-- Do NOT say "المعلومة مش عندي" or apologize at length.
-- Instead, use one of these Arabic templates:
-  1) "تمام — هحوّل سؤالك لفريق خدمة العملاء عشان نديك إجابة دقيقة."
-  2) "تمام — هسجّل استفسارك، وخدمة العملاء هيردّوا عليك بالتفاصيل."
-  3) "تمام — عشان نضمن الدقة، فريق خدمة العملاء هيتواصل معاك ويوضح لك كل التفاصيل."
-- Then ask ONE helpful follow-up question (if needed): "تحب أونلاين ولا حضور؟" (do NOT ask for phone number).
-
-INTENT BEHAVIOR:
-- GREETING: send a short welcome + 3 options.
-- COURSES_MENU: list available types only (Regular / Intensive / Express / Weekend / Online / Children), then ask which type.
-- COURSE_TYPE_DETAILS / ONLINE_COURSES / WEEKEND_COURSES / CHILDREN_COURSES:
-  answer ONLY from KB, then ask 1 follow-up question.
-- CENTER_INFO: answer ONLY from KB, then offer connecting to customer service.
-
-CURRENT INTENT: {intent}
-
-KNOWLEDGE BASE:
-{KB_TEXT}
+
+def detect_course_subtype(text: str) -> str | None:
+    t = normalize_arabic(text)
+
+    if any(x in t for x in ["express", "اكسبريس", "سريع", "super intensive"]):
+        return "express"
+    if any(x in t for x in ["intensive", "مكثف", "مكثفه"]):
+        return "intensive"
+    if any(x in t for x in ["regular", "ريجولار", "منتظم"]):
+        return "regular"
+    if any(x in t for x in ["weekend", "ويكند", "الجمعة", "السبت", "عطله", "عطلة"]):
+        return "weekend"
+    if any(x in t for x in ["online", "اونلاين", "اون لاين", "zoom", "زووم", "من البيت"]):
+        return "online"
+    if any(x in t for x in ["children", "kids", "اطفال", "طفل", "سن", "سنين"]):
+        return "children"
+
+    return None
+
+
+def fallback_customer_service_reply() -> str:
+    return (
+        "تمام — عشان نضمن لك إجابة دقيقة، فريق خدمة العملاء هيقدر يفيدك أكتر.\n"
+        "تحب أوجّه استفسارك لهم بخصوص أونلاين ولا حضور؟"
+    )
+
+
+def build_courses_menu_reply() -> str:
+    return (
+        "عندنا الأنواع دي من كورسات الألماني:\n"
+        "Regular / Intensive / Express / Weekend / Online / Children\n"
+        "تحب تعرف تفاصيل أنهي نوع؟"
+    )
+
+
+def build_center_info_reply() -> str:
+    center = KB["center"]
+    return (
+        f"ÄDK اتأسس سنة {center['established']}، و{center['branches']}.\n"
+        f"الكورسات للكبار {center['adults_levels']}، وللأطفال {center['children_levels']}.\n"
+        f"وكمان هو {center['goethe']}.\n"
+        "لو عندك سؤال محدد، أقدر أوصله لخدمة العملاء."
+    )
+
+
+def build_course_details_reply(course_key: str) -> str:
+    course = KB["courses"][course_key]
+    details = course["details"]
+
+    if course_key == "express":
+        return (
+            "الـ Express مناسب لو عايز تتعلم بسرعة.\n"
+            f"{details[0]}، و{details[1]}.\n"
+            f"ممكن تحجز Stage كاملة، والنظام {details[3]}، {details[4]}.\n"
+            f"مدة الـ stage {details[5].replace('مدة الـ stage ', '')}.\n"
+            "تحب تعرف الفرق بينه وبين الـ Intensive؟"
+        )
+
+    if course_key == "intensive":
+        return (
+            "الـ Intensive ممكن تحجز فيه Stage كاملة من A1 لحد B1.\n"
+            f"النظام {details[1]}، و{details[2]}.\n"
+            f"مدة الـ stage {details[3].replace('مدة الـ stage ', '')}.\n"
+            "تحب أقارن لك بينه وبين الـ Regular؟"
+        )
+
+    if course_key == "regular":
+        return (
+            "الـ Regular مناسب لو تحب تمشي level level.\n"
+            f"ممكن تحجز كل level لوحده من A1.1 لحد B1.3.\n"
+            f"النظام {details[1]}، و{details[2]}.\n"
+            f"مدة الكورس {details[3].replace('مدة الكورس ', '')}.\n"
+            "تحب تعرف لو ده أنسب لك ولا الـ Intensive؟"
+        )
+
+    if course_key == "weekend":
+        return (
+            "الـ Weekend مناسب لو إنت مشغول خلال الأسبوع بسبب شغل أو دراسة.\n"
+            "وبيمشي بنفس outline وساعات الـ Regular.\n"
+            "تحب حضور في الويكند ولا كنت بتدور على أونلاين؟"
+        )
+
+    if course_key == "online":
+        return (
+            "عندنا كورسات أونلاين، والحضور بيكون online.\n"
+            "لو مناسب لك النظام ده، أقدر أساعدك تختار النوع الأقرب لاحتياجك.\n"
+            "تحب أقولك على الأنواع المتاحة؟"
+        )
+
+    if course_key == "children":
+        return (
+            "عندنا كورسات أطفال من سن 5 لحد 15 سنة.\n"
+            "الهدف منها تنمية مهارات الطفل واللغة.\n"
+            "وكمان بتشجع التفكير الحر والنقدي والاستقلالية.\n"
+            "تحب أعرفك على نظام الأطفال بشكل أقرب؟"
+        )
+
+    return fallback_customer_service_reply()
+
+
+def sanitize_reply(text: str) -> str:
+    if not text or not text.strip():
+        return fallback_customer_service_reply()
+
+    text = text.strip()
+
+    replacements = {
+        "حضرتك": "إنت",
+        "مرحبا": "أهلا",
+        "مرحباً": "أهلا",
+        "يسعدني": "أقدر",
+        "يهتم بيك": "تحب",
+        "اقترحك": "أقترح لك",
+        "يمكنني": "أقدر",
+        "سأقوم": "ه",
+        "سوف": "ه",
+        "العميل": "إنت",
+        "المستخدم": "إنت",
+    }
+
+    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",
+        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):
+            return fallback_customer_service_reply()
+
+    return text
+
+
+def generate_rule_based_reply(intent: str, user_text: str) -> str:
+    subtype = detect_course_subtype(user_text)
+
+    if intent == "GREETING":
+        return (
+            "أهلا بيك في ÄDK.\n"
+            "أقدر أساعدك في أنواع الكورسات، تفاصيل نوع معين، أو معلومات عن المركز.\n"
+            "تحب تبدأ بإيه؟"
+        )
+
+    if intent == "HUMAN_AGENT":
+        return (
+            "تمام — هسجّل استفسارك لفريق خدمة العملاء عشان يردوا عليك بشكل أدق.\n"
+            "تحب استفسارك يكون عن أونلاين ولا حضور؟"
+        )
+
+    if intent == "COURSES_MENU":
+        return build_courses_menu_reply()
+
+    if intent == "CENTER_INFO":
+        return build_center_info_reply()
+
+    if intent == "ONLINE_COURSES":
+        return build_course_details_reply("online")
+
+    if intent == "WEEKEND_COURSES":
+        return build_course_details_reply("weekend")
+
+    if intent == "CHILDREN_COURSES":
+        return build_course_details_reply("children")
+
+    if intent == "COURSE_TYPE_DETAILS":
+        if subtype in ["express", "intensive", "regular"]:
+            return build_course_details_reply(subtype)
+        return build_courses_menu_reply()
+
+    return ""
+
+
+# =========================
+# LLM Fallback (ONLY for OTHER)
+# =========================
+
+def build_system_prompt() -> str:
+    return """
+أنت مساعد واتساب رسمي لمركز ÄDK.
+المطلوب:
+- ترد بالعربية المصرية بشكل مهذب ورسمي وخفيف.
+- الرد يكون قصير جدًا: من 2 إلى 4 سطور.
+- ممنوع اختراع أي معلومة غير الموجودة هنا.
+- ممنوع ذكر أسعار أو مواعيد أو عناوين أو أرقام فروع.
+- لو السؤال خارج المعلومات المتاحة، استخدم النص التالي حرفيًا:
+"تمام — عشان نضمن لك إجابة دقيقة، فريق خدمة العملاء هيقدر يفيدك أكتر.
+تحب أوجّه استفسارك لهم بخصوص أونلاين ولا حضور؟"
+
+المعلومات المتاحة فقط:
+1) المركز:
+- ÄDK اتأسس سنة 1998.
+- له فروع مختلفة في القاهرة.
+- بيقدم كورسات للكبار من A1 لحد C1.
+- وبيقدم كورسات للأطفال من A1 لحد B1.
+- وهو أول معهد معتمد من Goethe Institute في الشرق الأوسط وشمال أفريقيا.
+
+2) أنواع الكورسات:
+- Express: Super intensive، 3 محاضرات أسبوعيًا، كل محاضرة 4 ساعات، مدة الـ stage شهرين ونص، وممكن تحجز Stage كاملة.
+- Intensive: محاضرتين أسبوعيًا، كل محاضرة 4 ساعات، مدة الـ stage 3 شهور ونص، وممكن تحجز Stage كاملة.
+- Regular: تحجز كل level لوحده من A1.1 لحد B1.3، محاضرتين أسبوعيًا، كل محاضرة 4 ساعات، مدة الكورس شهر ونص.
+- Weekend: مناسب للمشغولين خلال الأسبوع، وبيتبع نفس outline وساعات الـ Regular.
+- Online: متاح بنظام أونلاين.
+- Children: من سن 5 لحد 15 سنة، والهدف تنمية مهارات الطفل واللغة وتشجيع التفكير الحر والنقدي والاستقلالية.
 """.strip()
 
-def process_text(user_text: str):
-    client = openai.OpenAI(api_key=api_key, base_url=base_url)
 
-    intent = detect_intent(user_text)
-    system_prompt = build_system_prompt(intent)
+def process_with_llm(user_text: str) -> str:
+    if not api_key:
+        return fallback_customer_service_reply()
+
+    client = openai.OpenAI(api_key=api_key, base_url=base_url)
 
     try:
         resp = client.chat.completions.create(
             model=model_link,
             messages=[
-                {"role": "system", "content": system_prompt},
+                {"role": "system", "content": build_system_prompt()},
                 {"role": "user", "content": user_text},
             ],
-            max_tokens=500,
-            temperature=0.2,
+            max_tokens=120,
+            temperature=0.0,
+            top_p=0.3,
             stream=False,
         )
 
         answer = resp.choices[0].message.content.strip()
+        return sanitize_reply(answer)
+
+    except Exception:
+        return fallback_customer_service_reply()
+
+
+# =========================
+# Main Processing
+# =========================
+
+def process_text(user_text: str):
+    user_text = (user_text or "").strip()
+
+    if not user_text:
+        return {
+            "ok": True,
+            "intent": "OTHER",
+            "reply": "أرسل سؤالك عن الكورسات أو عن المركز، وأنا أساعدك.",
+        }
+
+    intent = detect_intent(user_text)
 
-        # لو تحب ترجع intent كمان (مفيد للـ webhook logic)
+    # 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}
 
-    except Exception as e:
-        return {"ok": False, "error": str(e)}
+    # 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}
+
+
+# =========================
+# API Endpoint
+# =========================
+
+@app.post("/processtext")
+async def receive_updates(request: Request):
+    data = await request.json()
+    print("Received Update:", data)
+
+    result = process_text(data.get("message", ""))
+    print("Assistant:", result)
+
+    return JSONResponse(content=result)