Update app.py

app.py

@@ -64,31 +64,26 @@ 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 {"role": "bot", "content": "Great! Please select your preferred category: Vegetarian or Non-Vegetarian."}, ["Vegetarian", "Non-Vegetarian"]
+            return [{"role": "bot", "content": "Great! Please select your preferred category: Vegetarian or Non-Vegetarian."}], []
         
-        return {"role": "bot", "content": "I'm here to help you customize your food. Would you like to proceed? (Yes/No)"}, []
+        return [{"role": "bot", "content": "I'm here to help you customize your food. Would you like to proceed? (Yes/No)"}], []
     
     elif chat_state.step == "category":
         chat_state.category = message.lower()
         chat_state.step = "ingredient"
         
         if "vegetarian" in message.lower():
-            return {"role": "bot", "content": f"Great! What main ingredient would you like? Available options: {', '.join(VEG_INGREDIENTS)}"}, []
+            return [{"role": "bot", "content": f"Great! What main ingredient would you like? Available options: {', '.join(VEG_INGREDIENTS)}"}], []
         elif "non-vegetarian" in message.lower():
-            return {"role": "bot", "content": f"Great! What type of meat would you prefer? Available options: {', '.join(NONVEG_TYPES)}"}, []
+            return [{"role": "bot", "content": f"Great! What type of meat would you prefer? Available options: {', '.join(NONVEG_TYPES)}"}], []
         
-        return {"role": "bot", "content": "Please select either Vegetarian or Non-Vegetarian."}, []
+        return [{"role": "bot", "content": "Please select either Vegetarian or Non-Vegetarian."}], []
     
     elif chat_state.step == "ingredient":
         chat_state.ingredient = message.lower()
         chat_state.step = "nutrition"
         
-        if "veg" in chat_state.category:
-            return {"role": "bot", "content": f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"}, []
-        elif "nonveg" in chat_state.category:
-            return {"role": "bot", "content": f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"}, []
-        
-        return {"role": "bot", "content": "I'm not sure how to help with that. Would you like to start over?"}, []
+        return [{"role": "bot", "content": f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"}], []
     
     elif chat_state.step == "nutrition":
         nutrition = message.lower().replace(" ", "")
@@ -101,12 +96,12 @@ def process_message(message, history):
             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 {"role": "bot", "content": response}, []
+            return [{"role": "bot", "content": response}], []
         except KeyError:
             chat_state.step = "initial"  # Reset for new conversation
-            return {"role": "bot", "content": f"Sorry, I don't have any dishes for this combination."}, []
+            return [{"role": "bot", "content": f"Sorry, I don't have any dishes for this combination."}], []
 
-    return {"role": "bot", "content": "I'm not sure how to help with that. Would you like to start over?"}, []
+    return [{"role": "bot", "content": "I'm not sure how to help with that. Would you like to start over?"}], []
 
 def create_gradio_interface():
     with gr.Blocks() as demo:
@@ -120,12 +115,7 @@ def create_gradio_interface():
         def bot(history):
             user_message = history[-1]["content"]
             bot_message, buttons = process_message(user_message, history)
-            history[-1]["content"] = bot_message["content"]
-            
-            # Display buttons for category selection after the initial message
-            if buttons:
-                button_elements = [gr.Button(button) for button in buttons]
-                return history + [{"role": "bot", "content": bot_message["content"]}, button_elements]
+            history[-1]["content"] = bot_message[0]["content"]  # Update the message content
             
             return history