Update app.py

app.py

@@ -64,7 +64,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: Vegetarian or Non-Vegetarian.", ["Vegetarian", "Non-Vegetarian"]
         return "I'm here to help you customize your food. Would you like to proceed? (Yes/No)"
     
     elif chat_state.step == "category":
@@ -72,10 +72,10 @@ def process_message(message, history):
         chat_state.step = "ingredient"
         
         if "vegetarian" in message.lower():
-            return f"Great! What main ingredient would you like? Available options: {', '.join(VEG_INGREDIENTS)}"
+            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 f"Great! What type of meat would you prefer? Available options: {', '.join(NONVEG_TYPES)}", []
+        return "Please select either Vegetarian or Non-Vegetarian."
     
     elif chat_state.step == "ingredient":
         chat_state.ingredient = message.lower()
@@ -83,9 +83,9 @@ def process_message(message, history):
         
         # Check if category is vegetarian or non-vegetarian
         if "veg" in chat_state.category:
-            return f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"
+            return f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}", []
         elif "nonveg" in chat_state.category:
-            return f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"
+            return f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}", []
         return "I'm not sure how to help with that. Would you like to start over?"
     
     elif chat_state.step == "nutrition":
@@ -99,10 +99,10 @@ 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 response
+            return response, []
         except KeyError:
             chat_state.step = "initial"  # Reset for new conversation
-            return f"Sorry, I don't have any dishes for this combination."
+            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?"
 
@@ -112,32 +112,22 @@ def create_gradio_interface():
         msg = gr.Textbox(show_label=False, placeholder="Enter text and press enter")
         clear = gr.Button("Clear")
         
-        # Buttons for category selection
-        veg_button = gr.Button("Vegetarian")
-        nonveg_button = gr.Button("Non-Vegetarian")
-
+        # Buttons for category selection will appear in the chat stream
         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)
+            bot_message, buttons = process_message(user_message, history)
             history[-1][1] = bot_message
+            
+            # Display buttons for category selection after the initial message
+            if buttons:
+                button_elements = [gr.Button(button) for button in buttons]
+                return history + [[bot_message, button_elements]]
+            
             return history
 
-        # Function for category selection buttons
-        def select_category(category):
-            chat_state.step = "category"
-            chat_state.category = category.lower()
-            if "veg" in category.lower():
-                return f"Great! What main ingredient would you like? Available options: {', '.join(VEG_INGREDIENTS)}"
-            elif "non-vegetarian" in category.lower():
-                return f"Great! What type of meat would you prefer? Available options: {', '.join(NONVEG_TYPES)}"
-
-        # Wire buttons to category selection
-        veg_button.click(select_category, inputs=[veg_button], outputs=[chatbot])
-        nonveg_button.click(select_category, inputs=[nonveg_button], outputs=[chatbot])
-
         msg.submit(user, [msg, chatbot], [msg, chatbot]).then(bot, chatbot, chatbot)
         clear.click(lambda: None, None, chatbot)