Update app.py

app.py

@@ -57,6 +57,7 @@ class ChatState:
         self.step = "initial"
         self.category = ""
         self.ingredient = ""
+        self.nutrition = ""
 
 chat_state = ChatState()
 
@@ -75,7 +76,7 @@ def process_message(message, history):
             return f"Great! What main ingredient would you like? Available options: {', '.join(VEG_INGREDIENTS)}"
         elif "non-vegetarian" in message.lower():
             return f"Great! What type of meat would you prefer? Available options: {', '.join(NONVEG_TYPES)}"
-        return "Please select either Vegetarian, Non-Vegetarian"
+        return "Please select either Vegetarian or Non-Vegetarian."
     
     elif chat_state.step == "ingredient":
         chat_state.ingredient = message.lower()
@@ -83,12 +84,12 @@ def process_message(message, history):
         return f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"
     
     elif chat_state.step == "nutrition":
-        nutrition = message.lower().replace(" ", "")
+        chat_state.nutrition = message.lower().replace(" ", "")
         category = "nonveg" if "non" in chat_state.category else "veg"
         ingredient = chat_state.ingredient
         
         try:
-            food_items = FOOD_DATABASE[category][ingredient][nutrition]
+            food_items = FOOD_DATABASE[category][ingredient][chat_state.nutrition]
             response = f"Here are some {message} {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']}"
@@ -100,6 +101,7 @@ def process_message(message, history):
 
     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([])