Update app.py

app.py

@@ -99,3 +99,71 @@ def process_message(message, history):
         food_items = FOOD_DATABASE[category][chat_state.ingredient]
         food_item_names = [item['name'] for nutrition in food_items for item in food_items[nutrition]]
         
+        # Show available food items and move to nutrition selection
+        chat_state.step = "nutrition"
+        return f"Great! Here are the available food items for {message}:\n{', '.join(food_item_names)}\nNow, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"
+    
+    # Step 4: Nutrition preference
+    elif chat_state.step == "nutrition":
+        # Sanitize and normalize input (remove spaces, handle case)
+        nutrition = message.lower().replace(" ", "").strip()  # Remove spaces and make lowercase
+        chat_state.nutrition = nutrition
+        category = "nonveg" if "non" in chat_state.category else "veg"
+        ingredient = chat_state.ingredient
+
+        # Try to fetch the food items from the correct category, ingredient, and nutrition preference
+        try:
+            # Access food items based on category, ingredient, and nutrition preference
+            food_items = FOOD_DATABASE[category][ingredient][nutrition]
+            if not food_items:
+                return f"Sorry, no {nutrition.replace('_', ' ').title()} dishes are available for {ingredient}."
+
+            response = f"Here are some {nutrition.replace('_', ' ').title()} {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?"
+
+# Gradio interface for the chatbot
+def create_gradio_interface():
+    with gr.Blocks() as demo:
+        category_btns = gr.Radio(choices=["Vegetarian", "Non-Vegetarian"], label="Select Category")
+        ingredient_btns = gr.Radio(choices=VEG_INGREDIENTS + NONVEG_TYPES, label="Select Ingredient")
+        nutrition_btns = gr.Radio(choices=NUTRITION_OPTIONS, label="Select Nutrition")
+        chatbot = gr.Chatbot([])
+        msg = gr.Textbox(show_label=False, placeholder="Enter text and press enter")
+        clear = gr.Button("Clear")
+
+        # Function for user message input and history handling
+        def user(user_message, history):
+            return "", history + [[user_message, None]]
+        
+        # Function to handle bot's response
+        def bot(history):
+            user_message = history[-1][0]
+            bot_message = process_message(user_message, history)
+            history[-1][1] = bot_message
+            return history
+
+        # Handle category selection and pass to the next steps
+        category_btns.change(lambda category: process_message(category, []), category_btns, chatbot)
+        ingredient_btns.change(lambda ingredient: process_message(ingredient, []), ingredient_btns, chatbot)
+        nutrition_btns.change(lambda nutrition: process_message(nutrition, []), nutrition_btns, chatbot)
+
+        msg.submit(user, [msg, chatbot], [msg, chatbot]).then(bot, chatbot, chatbot)
+        clear.click(lambda: None, None, chatbot)
+
+    return demo
+
+# Start the Gradio interface and Flask app together
+if __name__ == "__main__":
+    demo = create_gradio_interface()
+    demo.launch(share=True, inline=True)
+    # Run Flask app
+    app.run(debug=True, use_reloader=False)