Update app.py

app.py

@@ -293,6 +293,30 @@ def process_order_flow(user_id: str, message: str) -> str:
         user_state[user_id] = state
         return "Sure! What dish would you like to order?"
 
+    # --- New Logic: Parse Dish and Quantity in a Single Message ---
+    if not state or state.flow != "order":
+        # Check if the message contains a dish and quantity
+        dish_candidates = [item["name"] for item in menu_items]
+        found_dish = None
+        for dish in dish_candidates:
+            if dish.lower() in message.lower():
+                found_dish = dish
+                break
+        numbers = re.findall(r'\d+', message)
+        if found_dish and numbers:
+            quantity = int(numbers[0])
+            if quantity <= 0:
+                return "Please enter a valid quantity (e.g., 1, 2, 3)."
+            # Initialize state and skip to step 3 (phone number)
+            state = ConversationState()
+            state.flow = "order"
+            state.step = 3
+            state.data["dish"] = found_dish
+            state.data["quantity"] = quantity
+            state.update_last_active()
+            user_state[user_id] = state
+            return f"You selected {found_dish} with {quantity} serving(s). Please provide your phone number for order updates."
+
     if state and state.flow == "order":
         state.update_last_active()
         # --- Step 1: Expecting Dish Selection (and optionally quantity) ---