Update app.py

app.py

@@ -121,6 +121,54 @@ def process_message(message, history):
         
         try:
             food_items = FOOD_DATABASE[category][ingredient][nutrition]
-            response = f"Here are some {message} {ingredient} dishes for you:\n"
+            response = f"Here are some {nutrition} {ingredient} dishes for you:\n"
             for item in food_items:
-                response += f"\n• {item['name']} ({item
+                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
+            }, []
+        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': "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)