Update app.py

app.py

@@ -64,76 +64,41 @@ 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 or Non-Vegetarian.", ["Vegetarian", "Non-Vegetarian"]
-        return "I'm here to help you customize your food. Would you like to proceed? (Yes/No)"
+            return {
+                'role': 'bot',
+                'content': "Great! Please select your preferred category: Vegetarian or Non-Vegetarian."
+            }, ["Vegetarian", "Non-Vegetarian"]
+        
+        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 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 f"Great! What type of meat would you prefer? Available options: {', '.join(NONVEG_TYPES)}", []
-        return "Please select either Vegetarian or Non-Vegetarian."
+            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."
+        }, []
     
     elif chat_state.step == "ingredient":
         chat_state.ingredient = message.lower()
         chat_state.step = "nutrition"
         
-        # 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)}", []
-        elif "nonveg" in chat_state.category:
-            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":
-        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]
-            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']}"
-            chat_state.step = "initial"  # Reset for new conversation
-            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 "I'm not sure how to help with that. Would you like to start over?"
-
-def create_gradio_interface():
-    with gr.Blocks() as demo:
-        chatbot = gr.Chatbot([], type='messages')  # Set the type to 'messages' to avoid warning
-        msg = gr.Textbox(show_label=False, placeholder="Enter text and press enter")
-        clear = gr.Button("Clear")
-        
-        # 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, 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
-
-        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)
+            return {
+                'role': 'bot',
+                'content': f"Perfect! Now, select your nutrition preference: {