Update engine/flow_router.py

engine/flow_router.py

@@ -474,16 +474,65 @@ 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,
+                "flow_data": {},
+                "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, ["كورسات", "الكورسات", "ايه الكورسات", "ما هي الكورسات", "الأنواع", "الانواع"]):
+            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,
-            "reply": "تم تسجيل طلبك بالفعل، وفريقنا هيتواصل معاك قريب. لو عندك استفسار إضافي ابعته هنا."
-        }
-
-    return unknown_reply(flow_data)
+            "reply": (
+                "تم تسجيل طلبك بالفعل، وفريقنا هيتواصل معاك قريب.\n"
+                "ولو تحب تبدأ استفسار جديد، اكتب: ابدأ من جديد."
+            )
+        }