Update app.py

app.py

@@ -1,130 +1,83 @@
 import gradio as gr
 
-# Food database (same as your provided code)
-FOOD_DATABASE = {
-    "veg": {
-        "paneer": {
-            "lowCalorie": [
-                {"name": "Grilled Paneer Tikka", "description": "Marinated and grilled paneer cubes with spices", "calories": 180, "protein": 14},
-                {"name": "Paneer Bhurji", "description": "Scrambled paneer with vegetables", "calories": 200, "protein": 16}
-            ],
-            "highProtein": [
-                {"name": "Protein Paneer Bowl", "description": "High protein paneer with quinoa", "calories": 320, "protein": 24},
-                {"name": "Paneer Steak", "description": "Thick cut paneer steak with herbs", "calories": 280, "protein": 22}
-            ]
-        },
-        "mushroom": {
-            "lowCalorie": [
-                {"name": "Grilled Mushroom Caps", "description": "Herb-stuffed mushroom caps", "calories": 120, "protein": 8},
-                {"name": "Mushroom Soup", "description": "Creamy mushroom soup", "calories": 150, "protein": 6}
-            ]
-        },
-        "aloo": {
-            "lowCalorie": [
-                {"name": "Baked Potato Wedges", "description": "Spiced and baked potato wedges", "calories": 160, "protein": 4},
-                {"name": "Aloo Tikki", "description": "Spiced potato patties", "calories": 180, "protein": 3}
-            ]
-        }
-    },
-    "nonveg": {
-        "chicken": {
-            "highProtein": [
-                {"name": "Grilled Chicken Breast", "description": "Herb-marinated grilled chicken", "calories": 250, "protein": 30},
-                {"name": "Chicken Tikka", "description": "Tandoori spiced chicken pieces", "calories": 280, "protein": 32}
-            ]
-        },
-        "fish": {
-            "lowCalorie": [
-                {"name": "Grilled Salmon", "description": "Lemon herb grilled salmon", "calories": 220, "protein": 25},
-                {"name": "Steamed Fish", "description": "Ginger-garlic steamed fish", "calories": 180, "protein": 22}
-            ]
-        },
-        "mutton": {
-            "balanced": [
-                {"name": "Mutton Curry", "description": "Traditional spiced mutton curry", "calories": 350, "protein": 28},
-                {"name": "Grilled Lamb Chops", "description": "Herb-crusted lamb chops", "calories": 380, "protein": 32}
-            ]
-        }
-    }
-}
-
-VEG_INGREDIENTS = ["Paneer", "Mushroom", "Aloo", "Brinjal", "Palak"]
-NONVEG_TYPES = ["Chicken", "Fish", "Mutton", "Pork"]
-NUTRITION_OPTIONS = ["Low Calorie", "High Protein", "Low Carb", "Balanced", "Gluten Free"]
+# Function that handles the user input and provides responses
+def chatbot(input_text, state):
+    # Initializing state if None
+    if state is None:
+        state = {"step": 1}
 
-class ChatState:
-    def __init__(self):
-        self.step = "initial"
-        self.category = ""
-        self.ingredient = ""
+    # Step 1: Ask what the user wants to customize
+    if state["step"] == 1:
+        state["step"] = 2
+        return "Hey, what would you like to customize?", state
 
-chat_state = ChatState()
+    # Step 2: Ask if the user wants veg or nonveg
+    if state["step"] == 2:
+        state["step"] = 3
+        return "Would you like to customize a veg or nonveg dish?", state
 
-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."}], []
-        
-        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)}"}], []
-        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": "Please select either Vegetarian or Non-Vegetarian."}], []
-    
-    elif chat_state.step == "ingredient":
-        chat_state.ingredient = message.lower()
-        chat_state.step = "nutrition"
-        
-        return [{"role": "bot", "content": f"Perfect! Now, select your nutrition preference: {', '.join(NUTRITION_OPTIONS)}"}], []
-    
-    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 {nutrition} {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 [{"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."}], []
+    # Step 3: Ask for nutritional selection (based on veg or nonveg)
+    if state["step"] == 3:
+        if input_text.lower() in ["veg", "nonveg"]:
+            state["step"] = 4
+            return f"Great! You've selected {input_text}. Now, what are your nutritional preferences?", state
+        return "Please select either 'veg' or 'nonveg'.", state
 
-    return [{"role": "bot", "content": "I'm not sure how to help with that. Would you like to start over?"}], []
+    # Step 4: Show ingredients based on veg or nonveg
+    if state["step"] == 4:
+        if input_text.lower() == "veg":
+            state["step"] = 5
+            return "Please choose an ingredient: Paneer, Gobi, Babycorn, Spinach, Mushroom, Tofu, Corn, Peas, Broccoli, Cauliflower", state
+        elif input_text.lower() == "nonveg":
+            state["step"] = 6
+            return "Please choose an ingredient: Fish, Chicken, Mutton, Prawns, Duck, Turkey, Crab, Lobster, Shrimp, Squid", state
+        return "Please select either 'veg' or 'nonveg'.", state
 
-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")
-        
-        def user(user_message, history):
-            return "", history + [{"role": "user", "content": user_message}]
+    # Step 5: Show dish items based on ingredient selection for veg
+    if state["step"] == 5:
+        veg_items = {
+            "paneer": ["Paneer Tikka", "Paneer Butter Masala", "Paneer Lababdar", "Paneer Makhani", "Palak Paneer", "Paneer Pakora", "Paneer Bhurji", "Shahi Paneer", "Kadai Paneer", "Paneer Kofta"],
+            "gobi": ["Gobi Manchurian", "Aloo Gobi", "Gobi Paratha", "Gobi Fry", "Gobi Masala", "Gobi Do Pyaza", "Gobi Curry", "Gobi Pulao", "Gobi Pakora", "Gobi Stir Fry"],
+            "babycorn": ["Babycorn Fry", "Babycorn Manchurian", "Babycorn Pakora", "Babycorn Masala", "Babycorn Pulao", "Babycorn Stir Fry", "Babycorn Biryani", "Babycorn Curry", "Babycorn Tikka", "Babycorn Chutney"],
+            "spinach": ["Palak Paneer", "Spinach Soup", "Spinach Paratha", "Palak Chaat", "Spinach Curry", "Spinach Pulao", "Spinach Biryani", "Spinach and Cheese Toast", "Spinach Frittata", "Spinach Pakora"],
+            "mushroom": ["Mushroom Masala", "Mushroom Soup", "Mushroom Biryani", "Mushroom Curry", "Mushroom Stir Fry", "Mushroom Tikka", "Mushroom Paratha", "Mushroom Pulao", "Mushroom Manchurian", "Mushroom Pakora"],
+            "tofu": ["Tofu Stir Fry", "Tofu Curry", "Tofu Tikka", "Tofu Soup", "Tofu Salad", "Tofu Stir Fried Rice", "Tofu Pulao", "Tofu Pakora", "Tofu Biryani", "Tofu Wrap"],
+            "corn": ["Sweet Corn Soup", "Corn Pakora", "Corn Stir Fry", "Corn Tikki", "Corn Pulao", "Corn Biryani", "Corn Masala", "Corn Fritters", "Corn Salad", "Corn and Peas Curry"],
+            "peas": ["Matar Paneer", "Matar Masala", "Matar Pulao", "Matar Tikki", "Matar Paratha", "Matar Pakora", "Matar Chaat", "Matar Biryani", "Matar Curry", "Matar Stir Fry"],
+            "broccoli": ["Broccoli Stir Fry", "Broccoli Soup", "Broccoli Salad", "Broccoli Pakora", "Broccoli Tikki", "Broccoli Curry", "Broccoli Pulao", "Broccoli Frittata", "Broccoli Manchurian", "Broccoli Cheese Bake"],
+            "cauliflower": ["Gobi Manchurian", "Aloo Gobi", "Gobi Pakora", "Gobi Tikka", "Gobi Masala", "Gobi Pulao", "Aloo Gobi Paratha", "Gobi Curry", "Gobi Stir Fry", "Gobi Cheese Bake"]
+        }
         
-        def bot(history):
-            user_message = history[-1]["content"]
-            bot_message, buttons = process_message(user_message, history)
-            history[-1]["content"] = bot_message[0]["content"]  # Update the message content
-            
-            return history
+        ingredient = input_text.lower()
+        if ingredient in veg_items:
+            return f"Here are some {ingredient} dishes: {', '.join(veg_items[ingredient])}", state
+        return "Please select a valid ingredient from the list.", state
 
-        msg.submit(user, [msg, chatbot], [msg, chatbot]).then(bot, chatbot, chatbot)
-        clear.click(lambda: None, None, chatbot)
+    # Step 6: Show dish items based on ingredient selection for nonveg
+    if state["step"] == 6:
+        nonveg_items = {
+            "fish": ["Fish Curry", "Grilled Fish", "Fish Fry", "Fish Tikka", "Fish Biryani", "Fish Korma", "Fish Masala", "Fish Soup", "Fish and Chips", "Fish Amritsari"],
+            "chicken": ["Chicken Curry", "Chicken Biryani", "Grilled Chicken", "Chicken Tikka", "Chicken Pakora", "Chicken Fry", "Chicken Korma", "Chicken Masala", "Chicken Shawarma", "Chicken Pulao"],
+            "mutton": ["Mutton Rogan Josh", "Mutton Biryani", "Mutton Curry", "Mutton Korma", "Mutton Fry", "Mutton Seekh Kebab", "Mutton Nihari", "Mutton Do Pyaza", "Mutton Masala", "Mutton Kofta"],
+            "prawns": ["Prawn Curry", "Prawn Masala", "Prawn Biryani", "Prawn Fry", "Prawn Tikka", "Prawn Soup", "Prawn Pulao", "Prawn Korma", "Prawn Grill", "Prawn Prawn Pakora"],
+            "duck": ["Duck Curry", "Duck Roast", "Duck Biryani", "Duck Korma", "Duck Fry", "Duck Soup", "Duck Stir Fry", "Duck Tikka", "Duck Pulao", "Duck Shawarma"],
+            "turkey": ["Turkey Roast", "Turkey Biryani", "Turkey Soup", "Turkey Masala", "Turkey Korma", "Turkey Stir Fry", "Turkey Tikka", "Turkey Grill", "Turkey Pulao", "Turkey Tacos"],
+            "crab": ["Crab Curry", "Crab Fry", "Crab Soup", "Crab Masala", "Crab Biryani", "Crab Pulao", "Crab Cakes", "Crab Tikka", "Crab Stir Fry", "Crab Roast"],
+            "lobster": ["Lobster Curry", "Lobster Grilled", "Lobster Soup", "Lobster Biryani", "Lobster Masala", "Lobster Tikka", "Lobster Stir Fry", "Lobster Pulao", "Lobster Shawarma", "Lobster Pakora"],
+            "shrimp": ["Shrimp Biryani", "Shrimp Curry", "Shrimp Tikka", "Shrimp Masala", "Shrimp Fry", "Shrimp Pulao", "Shrimp Soup", "Shrimp Grill", "Shrimp Korma", "Shrimp Pakora"],
+            "squid": ["Squid Curry", "Squid Masala", "Grilled Squid", "Squid Soup", "Squid Stir Fry", "Squid Biryani", "Squid Tikka", "Squid Pakora", "Squid Pulao", "Squid Korma"]
+        }
+        
+        ingredient = input_text.lower()
+        if ingredient in nonveg_items:
+            return f"Here are some {ingredient} dishes: {', '.join(nonveg_items[ingredient])}", state
+        return "Please select a valid ingredient from the list.", state
 
-    return demo
+# Set up the Gradio interface
+iface = gr.Interface(fn=chatbot,
+                     inputs=["text", "state"],
+                     outputs=["text", "state"],
+                     live=True,
+                     allow_flagging="never")
 
-# Launch the Gradio app
-if __name__ == "__main__":
-    demo = create_gradio_interface()
-    demo.launch(share=True)
+iface.launch()