Update app.py

app.py

@@ -365,40 +365,77 @@ async def chatbot_response(request: Request, background_tasks: BackgroundTasks):
         sentiment_modifier = "Great to hear from you! "
 
     # Check for specialized commands:
-    if "menu" in user_message.lower():
-        # Return menu with images
-        menu_with_images = []
+    # Check for specialized commands:
+if "menu" in user_message.lower():
+    # Return menu with images and options for selection
+    menu_with_images = []
+    for index, item in enumerate(menu_items, start=1):
+        image_url = google_image_scrape(item["name"])
+        menu_with_images.append({
+            "number": index,  # Add a number for each dish
+            "name": item["name"],
+            "description": item["description"],
+            "price": item["price"],
+            "image_url": image_url
+        })
+    response_payload = {
+        "response": sentiment_modifier + "Here’s our delicious menu:",
+        "menu": menu_with_images,
+        "follow_up": (
+            "To order, type the *number* or *name* of the dish you'd like. "
+            "For example, type '1' or 'Jollof Rice' to order Jollof Rice.\n\n"
+            "You can also ask for nutritional facts by typing, for example, 'Nutritional facts for Jollof Rice'."
+        )
+    }
+    background_tasks.add_task(log_chat_to_db, user_id, "outbound", str(response_payload))
+    return JSONResponse(content=response_payload)
+
+# Handle dish selection for ordering
+if any(item["name"].lower() in user_message.lower() for item in menu_items) or \
+   any(str(index) == user_message.strip() for index, item in enumerate(menu_items, start=1)):
+    # Extract the selected dish
+    selected_dish = None
+    if user_message.strip().isdigit():
+        # User selected by number
+        dish_number = int(user_message.strip())
+        if 1 <= dish_number <= len(menu_items):
+            selected_dish = menu_items[dish_number - 1]["name"]
+    else:
+        # User selected by name
         for item in menu_items:
-            image_url = google_image_scrape(item["name"])
-            menu_with_images.append({
-                "name": item["name"],
-                "description": item["description"],
-                "price": item["price"],
-                "image_url": image_url
-            })
-        response_payload = {
-            "response": sentiment_modifier + "Here’s our delicious menu:",
-            "menu": menu_with_images,
-            "follow_up": "Would you like nutritional facts for any dish? Just type, for example, 'Nutritional facts for Jollof Rice'."
-        }
-        background_tasks.add_task(log_chat_to_db, user_id, "outbound", str(response_payload))
-        return JSONResponse(content=response_payload)
-
-    if "nutritional facts for" in user_message.lower():
-        dish_name = user_message.lower().replace("nutritional facts for", "").strip().title()
-        dish = next((item for item in menu_items if item["name"].lower() == dish_name.lower()), None)
-        if dish:
-            response_text = f"Nutritional facts for {dish['name']}:\n{dish['nutrition']}"
-        else:
-            response_text = f"Sorry, I couldn't find nutritional facts for {dish_name}."
+            if item["name"].lower() in user_message.lower():
+                selected_dish = item["name"]
+                break
+
+    if selected_dish:
+        # Trigger the order flow
+        user_state[user_id] = {"flow": "order", "step": 1, "data": {"dish": selected_dish}, "last_active": datetime.utcnow()}
+        response_text = f"You selected {selected_dish}. How many servings would you like?"
         background_tasks.add_task(log_chat_to_db, user_id, "outbound", response_text)
         return JSONResponse(content={"response": sentiment_modifier + response_text})
+    else:
+        response_text = "Sorry, I couldn't find that dish in the menu. Please try again."
+        background_tasks.add_task(log_chat_to_db, user_id, "outbound", response_text)
+        return JSONResponse(content={"response": sentiment_modifier + response_text})
+
+# Handle nutritional facts request
+if "nutritional facts for" in user_message.lower():
+    dish_name = user_message.lower().replace("nutritional facts for", "").strip().title()
+    dish = next((item for item in menu_items if item["name"].lower() == dish_name.lower()), None)
+    if dish:
+        response_text = f"Nutritional facts for {dish['name']}:\n{dish['nutrition']}"
+    else:
+        response_text = f"Sorry, I couldn't find nutritional facts for {dish_name}."
+    background_tasks.add_task(log_chat_to_db, user_id, "outbound", response_text)
+    return JSONResponse(content={"response": sentiment_modifier + response_text})
+
+# Check if this is an order flow request
+order_response = process_order_flow(user_id, user_message)
+if order_response:
+    background_tasks.add_task(log_chat_to_db, user_id, "outbound", order_response)
+    return JSONResponse(content={"response": sentiment_modifier + order_response})
 
-    # Check if this is an order flow request
-    order_response = process_order_flow(user_id, user_message)
-    if order_response:
-        background_tasks.add_task(log_chat_to_db, user_id, "outbound", order_response)
-        return JSONResponse(content={"response": sentiment_modifier + order_response})
+   
 
     # For context-aware conversation: store conversation context
     conversation_context.setdefault(user_id, []).append({"timestamp": datetime.utcnow().isoformat(), "message": user_message})