Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-# Food database
+# Corrected Food Database with consistent food item names
 FOOD_DATABASE = {
     "veg": {
         "paneer": {
@@ -48,8 +48,8 @@ FOOD_DATABASE = {
     }
 }
 
-VEG_INGREDIENTS = ["Paneer", "Mushroom", "Aloo", "Brinjal", "Palak"]
-NONVEG_TYPES = ["Chicken", "Fish", "Mutton", "Pork"]
+VEG_INGREDIENTS = ["Paneer", "Mushroom", "Aloo"]
+NONVEG_TYPES = ["Chicken", "Fish", "Mutton"]
 NUTRITION_OPTIONS = ["Low Calorie", "High Protein", "Low Carb", "Balanced", "Gluten Free"]
 
 class ChatState:
@@ -66,7 +66,7 @@ def process_message(message, history):
     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 "Great! Please select your preferred category: Veg, nonveg."
         return "I'm here to help you customize your food. Would you like to proceed? (Yes/No)"
     
     # Step 2: Category selection
@@ -74,17 +74,30 @@ def process_message(message, history):
         chat_state.category = message.lower()
         chat_state.step = "ingredient"
         
-        if "vegetarian" in message.lower():
+        if "veg" in message.lower():
             return f"Great! What main ingredient would you like? Available options: {', '.join(VEG_INGREDIENTS)}"
-        elif "non-vegetarian" in message.lower():
+        elif "nonveg" in message.lower():
             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().strip()  # Ensure no extra spaces or case issues
+        # Get the list of available food items for this ingredient
+        category = "nonveg" if "non" in chat_state.category else "veg"
+        
+        # Check if the ingredient exists in the food database
+        if chat_state.ingredient not in FOOD_DATABASE[category]:
+            chat_state.step = "initial"  # Reset for new conversation
+            return f"Sorry, I don't have any data for {message}. Please select another ingredient."
+
+        # List available food items based on ingredient
+        food_items = FOOD_DATABASE[category][chat_state.ingredient]
+        food_item_names = [item['name'] for nutrition in food_items for item in food_items[nutrition]]
+        
+        # Show available food items
         chat_state.step = "nutrition"
-        return f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"
+        return f"Great! Here are the available food items for {message}:\n{', '.join(food_item_names)}\nNow, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"
     
     # Step 4: Nutrition preference
     elif chat_state.step == "nutrition":
@@ -94,14 +107,6 @@ def process_message(message, history):
         category = "nonveg" if "non" in chat_state.category else "veg"
         ingredient = chat_state.ingredient
 
-        # Debugging: Print the state before fetching food items
-        print(f"Category: {category}, Ingredient: {ingredient}, Nutrition: {nutrition}")
-
-        # 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:
             # Access food items based on category, ingredient, and nutrition preference
@@ -120,29 +125,3 @@ def process_message(message, history):
             return f"Sorry, I don't have any dishes for this combination."
 
     return "I'm not sure how to help with that. Would you like to start over?"
-
-# Gradio interface for the chatbot
-def create_gradio_interface():
-    with gr.Blocks() as demo:
-        chatbot = gr.Chatbot([])
-        msg = gr.Textbox(show_label=False, placeholder="Enter text and press enter")
-        clear = gr.Button("Clear")
-
-        def user(user_message, history):
-            return "", history + [[user_message, None]]
-        
-        def bot(history):
-            user_message = history[-1][0]
-            bot_message = process_message(user_message, history)
-            history[-1][1] = bot_message
-            return history
-
-        msg.submit(user, [msg, chatbot], [msg, chatbot]).then(bot, chatbot, chatbot)
-        clear.click(lambda: None, None, chatbot)
-
-    return demo
-
-# Launch the Gradio app
-if __name__ == "__main__":
-    demo = create_gradio_interface()
-    demo.launch(share=True)