Update app.py

app.py

@@ -252,6 +252,22 @@ async def log_order_tracking(order_id: str, status: str, message: str = None):
         session.add(tracking_entry)
         await session.commit()
 
+from fuzzywuzzy import fuzz
+
+def match_dish(user_input: str, threshold: int = 80) -> str:
+    best_match = None
+    best_score = 0
+    for item in menu_items:
+        dish_name = item["name"]
+        score = fuzz.ratio(user_input.lower(), dish_name.lower())
+        if score > best_score:
+            best_score = score
+            best_match = dish_name
+    if best_score >= threshold:
+        return best_match
+    return None
+
+
 def calculate_shipping_cost(address: str) -> int:
     address_lower = address.lower()
     for area, cost in TOWN_SHIPPING_COSTS.items():
@@ -380,13 +396,9 @@ async def process_order_flow(user_id: str, message: str) -> str:
         user_state[user_id] = state
         return "Sure! What dish would you like to order?"
 
+    # If no state exists or not in order flow, try to detect a dish using fuzzy matching.
     if not state or state.flow != "order":
-        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
+        found_dish = match_dish(message)
         if found_dish:
             state = ConversationState()
             state.flow = "order"
@@ -417,7 +429,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
                     state.step = 5
                     return (f"Thanks! Your phone number is recorded as: {phone_number}.\n"
                             f"Your delivery address is: {address}.\n"
-                            f"Your delivery cost is N{shipping_cost}. Would you like to add any extras such as sides or drinks? (yes/no)")
+                            f"Your delivery cost is N{shipping_cost}. Would you like to add extras (yes/no)?")
                 elif phone_match:
                     state.data["phone_number"] = phone_match.group(1)
                     asyncio.create_task(update_user_profile(user_id, phone_number))
@@ -428,16 +440,13 @@ async def process_order_flow(user_id: str, message: str) -> str:
             else:
                 state.step = 2
                 return f"You selected {found_dish}. How many servings would you like?"
-
+    
+    # If state exists and we're already in order flow:
     if state and state.flow == "order":
         state.update_last_active()
         if state.step == 1:
-            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
+            # Use fuzzy matching on the message in case of typos.
+            found_dish = match_dish(message)
             numbers = re.findall(r'\d+', message)
             if found_dish:
                 state.data["dish"] = found_dish
@@ -454,7 +463,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
                     return f"You selected {found_dish}. How many servings would you like?"
             else:
                 return "I couldn't identify the dish. Please type the dish name from our menu."
-
+    
         if state.step == 2:
             numbers = re.findall(r'\d+', message)
             if not numbers:
@@ -465,7 +474,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
             state.data["quantity"] = quantity
             state.step = 3
             return f"Got it. {quantity} serving(s) of {state.data.get('dish')}. Please provide your phone number and delivery address."
-
+    
         if state.step == 3:
             phone_pattern = r'(\+?\d{10,15})'
             phone_match = re.search(phone_pattern, message)
@@ -484,7 +493,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
                 state.step = 5
                 return (f"Thanks! Your phone number is recorded as: {phone_number}.\n"
                         f"Your delivery address is: {address}.\n"
-                        f"Your delivery cost is N{shipping_cost}. Would you like to add any extras such as sides or drinks? (yes/no)")
+                        f"Your delivery cost is N{shipping_cost}. Would you like extras (yes/no)?")
             elif phone_match:
                 state.data["phone_number"] = phone_match.group(1)
                 asyncio.create_task(update_user_profile(user_id, phone_number))
@@ -492,7 +501,8 @@ async def process_order_flow(user_id: str, message: str) -> str:
             else:
                 return ("Please provide both your phone number and delivery address. "
                         "For example: '09162409591, 1, Iyana Isashi, Isashi, Ojo, Lagos'.")
-
+    
+        # Steps 4, 5, 6, and 7 remain unchanged.
         if state.step == 4:
             state.data["address"] = message
             asyncio.create_task(update_user_profile(user_id, address=message))
@@ -500,12 +510,12 @@ async def process_order_flow(user_id: str, message: str) -> str:
             state.data["shipping_cost"] = shipping_cost
             state.step = 5
             return (f"Thanks. Your delivery address is recorded as: {message}.\n"
-                    f"Your delivery cost is N{shipping_cost}. Would you like to add any extras such as sides or drinks? (yes/no)")
-
+                    f"Your delivery cost is N{shipping_cost}. Would you like extras (yes/no)?")
+    
         if state.step == 5:
             if message.lower() in ["yes", "y"]:
                 state.step = 6
-                return "Please list the extras you would like to add (e.g., drinks, sides, etc.)."
+                return "Please list the extras you'd like (e.g., drinks, sides)."
             elif message.lower() in ["no", "n"]:
                 state.data["extras"] = ""
                 state.step = 7
@@ -523,8 +533,8 @@ async def process_order_flow(user_id: str, message: str) -> str:
                            f"Extras: None\nConfirm order? (yes/no)")
                 return summary
             else:
-                return "Please respond with 'yes' or 'no'. Would you like to add any extras to your order? (yes/no)"
-
+                return "Please respond with 'yes' or 'no' regarding extras."
+    
         if state.step == 6:
             state.data["extras"] = message
             state.step = 7
@@ -542,7 +552,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
                        f"Total Price: N{total_price}\n"
                        f"Extras: {extras}\nConfirm order? (yes/no)")
             return summary
-
+    
         if state.step == 7:
             if message.lower() in ["yes", "y"]:
                 order_id = f"ORD-{int(time.time())}"
@@ -552,7 +562,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
                 shipping_cost = state.data.get("shipping_cost", 0)
                 total_price = (quantity * price_per_serving) + shipping_cost
                 state.data["price"] = str(total_price)
-
+    
                 async def save_order():
                     async with async_session() as session:
                         order = Order(
@@ -569,7 +579,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
                         await session.commit()
                 asyncio.create_task(save_order())
                 asyncio.create_task(log_order_tracking(order_id, "Order Placed", "Order placed and awaiting payment."))
-
+    
                 async def notify_management_order(order_details: dict):
                     message_body = (
                         f"New Order Received:\n"
@@ -591,7 +601,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
                     "address": state.data.get("address", "")
                 }
                 asyncio.create_task(notify_management_order(order_details))
-
+    
                 email = "customer@example.com"
                 payment_data = create_paystack_payment_link(email, total_price * 100, order_id)
                 dish_name = state.data.get("dish", "")