Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-# Food database (same as your provided code)
+# Food database
 FOOD_DATABASE = {
     "veg": {
         "paneer": {
@@ -62,12 +62,14 @@ class ChatState:
 chat_state = ChatState()
 
 def process_message(message, history):
+    # Step 1: If initial step (asks if user wants to proceed)
     if chat_state.step == "initial":
         if "yes" in message.lower() or "yeah" in message.lower():
             chat_state.step = "category"
             return "Great! Please select your preferred category: Vegetarian, Non-Vegetarian."
         return "I'm here to help you customize your food. Would you like to proceed? (Yes/No)"
     
+    # Step 2: Category selection
     elif chat_state.step == "category":
         chat_state.category = message.lower()
         chat_state.step = "ingredient"
@@ -78,21 +80,36 @@ def process_message(message, history):
             return f"Great! What type of meat would you prefer? Available options: {', '.join(NONVEG_TYPES)}"
         return "Please select either Vegetarian or Non-Vegetarian."
     
+    # Step 3: Ingredient selection
     elif chat_state.step == "ingredient":
-        chat_state.ingredient = message.lower()
+        chat_state.ingredient = message.lower().strip()  # Ensure no extra spaces or case issues
         chat_state.step = "nutrition"
         return f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"
     
+    # Step 4: Nutrition preference
     elif chat_state.step == "nutrition":
-        chat_state.nutrition = message.lower().replace(" ", "")
+        # Sanitize and normalize input (remove spaces, handle case)
+        nutrition = message.lower().replace(" ", "").strip()  # Remove spaces and make lowercase
+        chat_state.nutrition = nutrition
         category = "nonveg" if "non" in chat_state.category else "veg"
         ingredient = chat_state.ingredient
-        
+
+        # Check if the ingredient exists in the database for the selected category
+        if ingredient not in FOOD_DATABASE[category]:
+            chat_state.step = "initial"  # Reset for new conversation
+            return f"Sorry, I don't have any data for {ingredient}. Please select another ingredient."
+
+        # Try to fetch the food items from the correct category, ingredient, and nutrition preference
         try:
-            food_items = FOOD_DATABASE[category][ingredient][chat_state.nutrition]
-            response = f"Here are some {message} {ingredient} dishes for you:\n"
+            # Access food items based on category, ingredient, and nutrition preference
+            food_items = FOOD_DATABASE[category][ingredient][nutrition]
+            if not food_items:
+                return f"Sorry, no {nutrition.replace('_', ' ').title()} dishes are available for {ingredient}."
+
+            response = f"Here are some {nutrition.replace('_', ' ').title()} {ingredient} dishes for you:\n"
             for item in food_items:
                 response += f"\n• {item['name']} ({item['calories']} cal, {item['protein']}g protein)\n  {item['description']}"
+            
             chat_state.step = "initial"  # Reset for new conversation
             return response
         except KeyError: