Update engine/flow_router.py

engine/flow_router.py

@@ -85,7 +85,8 @@ def is_adults(text: str) -> bool:
 def is_children(text: str) -> bool:
     t = normalize_text(text)
     return contains_any(t, [
-        "اطفال", "أطفال", "طفل", "children", "kids", "كورسات الأطفال", "كورسات الاطفال"
+        "اطفال", "أطفال", "طفل", "children", "kids",
+        "كورسات الأطفال", "كورسات الاطفال"
     ])
 
 
@@ -185,6 +186,51 @@ def wants_new_topic(text: str) -> bool:
     ])
 
 
+def wants_courses_info(text: str) -> bool:
+    t = normalize_text(text)
+    return contains_any(t, [
+        "كورسات",
+        "الكورسات",
+        "ايه الكورسات",
+        "ما هي الكورسات",
+        "الأنواع",
+        "الانواع",
+        "عايز اعرف الكورسات",
+        "عايزة اعرف الكورسات",
+        "ايه الكورسات المتاحة",
+        "الكورسات المتاحة"
+    ])
+
+
+def asks_about_prior_study_case(text: str) -> bool:
+    t = normalize_text(text)
+    return contains_any(t, [
+        "لو كنت درست",
+        "لو كنت دارس",
+        "لو درست قبل كده",
+        "طب لو درست",
+        "ولو درست",
+        "اذا كنت درست",
+        "إذا كنت درست",
+        "اختبار تحديد مستوى",
+        "تحديد مستوى"
+    ])
+
+
+def asks_about_beginner_case(text: str) -> bool:
+    t = normalize_text(text)
+    return contains_any(t, [
+        "لو مكنتش درست",
+        "لو ما درستش",
+        "لو مدرستش",
+        "لو لسه جديد",
+        "لو مبتدئ",
+        "لو بادئ",
+        "لو اول مرة",
+        "لو أول مرة"
+    ])
+
+
 def route_message(state: str, text: str, session: dict):
     t = normalize_text(text)
     flow_data = session.get("flow_data", {}) or {}
@@ -244,6 +290,9 @@ def route_message(state: str, text: str, session: dict):
             flow_data["customer_type"] = "current"
             return current_student_menu_reply(flow_data)
 
+        if wants_courses_info(t):
+            return ask_new_or_current_reply(flow_data)
+
         return ask_new_or_current_reply(flow_data)
 
     # =========================
@@ -304,6 +353,16 @@ def route_message(state: str, text: str, session: dict):
     # WAITING_BEGINNER_SCHEDULE_CHOICE
     # =========================
     if state == WAITING_BEGINNER_SCHEDULE_CHOICE:
+        if asks_about_prior_study_case(t):
+            new_flow = dict(flow_data)
+            new_flow["prior_study"] = True
+            return placement_test_reply(new_flow)
+
+        if asks_about_beginner_case(t):
+            new_flow = dict(flow_data)
+            new_flow["prior_study"] = False
+            return beginner_schedule_reply(new_flow)
+
         if contains_any(t, ["تم", "اخترت", "اختارت", "جاهز", "جاهزة"]):
             return pdf_102_reply(flow_data)
 
@@ -326,6 +385,16 @@ def route_message(state: str, text: str, session: dict):
     # WAITING_PDF_102_CONFIRMATION
     # =========================
     if state == WAITING_PDF_102_CONFIRMATION:
+        if asks_about_prior_study_case(t):
+            new_flow = dict(flow_data)
+            new_flow["prior_study"] = True
+            return placement_test_reply(new_flow)
+
+        if asks_about_beginner_case(t):
+            new_flow = dict(flow_data)
+            new_flow["prior_study"] = False
+            return beginner_schedule_reply(new_flow)
+
         if contains_any(t, ["تم", "خلصت", "قريت", "اطلعت", "جاهز", "جاهزة"]):
             return payment_methods_reply(flow_data)
 
@@ -345,6 +414,16 @@ def route_message(state: str, text: str, session: dict):
     # WAITING_PLACEMENT_TEST_CONFIRMATION
     # =========================
     if state == WAITING_PLACEMENT_TEST_CONFIRMATION:
+        if asks_about_beginner_case(t):
+            new_flow = dict(flow_data)
+            new_flow["prior_study"] = False
+            return beginner_schedule_reply(new_flow)
+
+        if asks_about_prior_study_case(t):
+            new_flow = dict(flow_data)
+            new_flow["prior_study"] = True
+            return placement_test_reply(new_flow)
+
         if contains_any(t, ["تم", "اخترت", "اختارت", "جاهز", "جاهزة"]):
             return create_sales_handoff_result(
                 flow_data=flow_data,
@@ -474,7 +553,6 @@ def route_message(state: str, text: str, session: dict):
                 "reply": "العفو، تحت أمرك في أي وقت."
             }
 
-        # لو المستخدم عايز يبدأ من جديد
         if wants_restart(t) or wants_new_topic(t):
             return {
                 "next_state": WAITING_MAIN_MENU,
@@ -482,52 +560,44 @@ def route_message(state: str, text: str, session: dict):
                 "reply": "تمام، نبدأ من جديد. إنت طالب جديد ولا طالب حالي؟"
             }
 
-        # لو سأل عن كورسات الأطفال
         if is_children(t):
             new_flow = {}
             new_flow["customer_type"] = "new"
             new_flow["audience"] = "children"
             return ask_prior_study_reply(new_flow)
 
-        # لو سأل عن كورسات الكبار
         if is_adults(t):
             new_flow = {}
             new_flow["customer_type"] = "new"
             new_flow["audience"] = "adults"
             return ask_prior_study_reply(new_flow)
 
-        # لو قال طالب جديد
         if is_new_student(t):
             new_flow = {}
             new_flow["customer_type"] = "new"
             return ask_audience_reply(new_flow)
 
-        # لو قال طالب حالي
         if is_current_student(t):
             new_flow = {}
             new_flow["customer_type"] = "current"
             return current_student_menu_reply(new_flow)
 
-        # لو سأل بشكل عام عن الكورسات
-        if contains_any(t, ["كورسات", "الكورسات", "ايه الكورسات", "ما هي الكورسات", "الأنواع", "الانواع"]):
+        if wants_courses_info(t):
             return {
                 "next_state": WAITING_USER_TYPE,
                 "flow_data": {},
                 "reply": "تمام، أقدر أساعدك في كورسات الطلبة الجدد أو الحاليين. إنت طالب جديد ولا طالب حالي؟"
             }
 
-        # لو عايز خدمة العملاء
         if is_support_request(t):
             return create_support_handoff_result(
                 flow_data=flow_data,
                 summary=f"استفسار إضافي بعد handoff: {text}"
             )
 
-        # لو شكوى
         if is_complaint(t):
             return create_complaint_flow_result(flow_data)
 
-        # default
         return {
             "next_state": HANDOFF_DONE,
             "flow_data": flow_data,
@@ -535,4 +605,6 @@ def route_message(state: str, text: str, session: dict):
                 "تم تسجيل طلبك بالفعل، وفريقنا هيتواصل معاك قريب.\n"
                 "ولو تحب تبدأ استفسار جديد، اكتب: ابدأ من جديد."
             )
-        }
+        }
+
+    return unknown_reply(flow_data)