Update app.py

app.py

@@ -1,65 +1,79 @@
 import gradio as gr
 
-# Define the menu categories and dishes
-menu_items = [
-    'Roti', 'Butter Naan', 'Plain Naan', 'Garlic Naan', 'Chilly Naan',  # Breads
-    'Paneer Butter Masala', 'Kadai Paneer', 'Dum Ka Paneer', 'Mixed Veg Curry', 'Dal Makhani',  # Veg Curries
-    'Chicken Curry', 'Mutton Curry', 'Fish Curry', 'Prawns Curry',  # Non-Veg Curries
-    'Veg Biryani', 'Paneer Biryani', 'Chicken Biryani', 'Mutton Biryani', 'Fish Biryani', 'Prawns Biryani'  # Biryani
-]
+# Define menu categories
+breads = ['Roti', 'Butter Naan', 'Plain Naan', 'Garlic Naan', 'Chilly Naan']
+veg_curries = ['Paneer Butter Masala', 'Kadai Paneer', 'Dum Ka Paneer', 'Mixed Veg Curry', 'Dal Makhani']
+non_veg_curries = ['Chicken Curry', 'Mutton Curry', 'Fish Curry', 'Prawns Curry']
+biryani = ['Veg Biryani', 'Paneer Biryani', 'Chicken Biryani', 'Mutton Biryani', 'Fish Biryani', 'Prawns Biryani']
 
-# Pairing suggestions based on selected items
+# Function to provide suggestions based on selection
 def suggest_pairings(selected_items):
-    suggestions = []
+    # Create a set for selected items for easier matching
+    selected_set = set(selected_items)
     
-    # Suggestions based on bread selection
-    if any(bread in selected_items for bread in ['Roti', 'Butter Naan', 'Plain Naan', 'Garlic Naan', 'Chilly Naan']):
-        suggestions.append('Pair with Veg or Non-Veg Curry, and Biryani for a complete meal')
-    
-    # Suggestions based on curry selection
-    if any(curry in selected_items for curry in ['Paneer Butter Masala', 'Kadai Paneer', 'Dum Ka Paneer', 'Mixed Veg Curry', 'Dal Makhani']):
-        suggestions.append('Veg curries go well with rice-based dishes like Biryani')
-    
-    if any(curry in selected_items for curry in ['Chicken Curry', 'Mutton Curry', 'Fish Curry', 'Prawns Curry']):
-        suggestions.append('Non-Veg curries pair excellently with Biryani and Bread')
-    
-    # Suggestions based on Biryani selection
-    if any(biryani in selected_items for biryani in ['Veg Biryani', 'Paneer Biryani', 'Chicken Biryani', 'Mutton Biryani', 'Fish Biryani', 'Prawns Biryani']):
-        suggestions.append('Biryani pairs well with raita, naan, or a curry')
-    
-    # Customize the suggestion list
-    if not suggestions:
-        suggestions.append('Select more dishes to get suggestions!')
-    
-    return '\n'.join(suggestions)
+    # Initial suggestions
+    breads_suggested = []
+    veg_curries_suggested = []
+    non_veg_curries_suggested = []
+    biryani_suggested = []
 
-# Function for customizing selected items
-def customize_selection(selected_items):
-    # Display selected items and provide an option to add custom ingredients
-    selected_items_str = ", ".join(selected_items)
-    return f"You've selected: {selected_items_str}. You can customize your dish by adding ingredients below."
+    # Suggest items based on selection logic
+    if 'Roti' in selected_set or 'Butter Naan' in selected_set or 'Plain Naan' in selected_set or 'Garlic Naan' in selected_set or 'Chilly Naan' in selected_set:
+        veg_curries_suggested = veg_curries
+        non_veg_curries_suggested = non_veg_curries
+    elif 'Paneer Butter Masala' in selected_set or 'Kadai Paneer' in selected_set or 'Dum Ka Paneer' in selected_set or 'Mixed Veg Curry' in selected_set or 'Dal Makhani' in selected_set:
+        breads_suggested = breads
+        biryani_suggested = biryani
+    elif 'Chicken Curry' in selected_set or 'Mutton Curry' in selected_set or 'Fish Curry' in selected_set or 'Prawns Curry' in selected_set:
+        breads_suggested = breads
+        biryani_suggested = biryani
+    elif 'Veg Biryani' in selected_set or 'Paneer Biryani' in selected_set or 'Chicken Biryani' in selected_set or 'Mutton Biryani' in selected_set or 'Fish Biryani' in selected_set or 'Prawns Biryani' in selected_set:
+        breads_suggested = breads
+        veg_curries_suggested = veg_curries
+        non_veg_curries_suggested = non_veg_curries
 
-# Create the Gradio Interface
-def create_app():
-    # Input: Single selection box to select multiple items
-    menu_input = gr.CheckboxGroup(label="Select Items from Menu", choices=menu_items)
+    # If nothing selected, suggest everything
+    if not selected_items:
+        breads_suggested = breads
+        veg_curries_suggested = veg_curries
+        non_veg_curries_suggested = non_veg_curries
+        biryani_suggested = biryani
 
-    # Output: Pairing suggestions and customization box
-    pairing_output = gr.Textbox(label="Pairing Suggestions")
-    customization_output = gr.Textbox(label="Customization Box", interactive=True)
+    return breads_suggested, veg_curries_suggested, non_veg_curries_suggested, biryani_suggested
 
-    # Set up the Gradio Interface
-    interface = gr.Interface(
-        fn=lambda selected_items: (suggest_pairings(selected_items), customize_selection(selected_items)),
-        inputs=menu_input,
-        outputs=[pairing_output, customization_output],
-        live=True
-    )
+# Function to handle user input and generate suggestions
+def menu_selection(selected_items, custom_ingredient):
+    # Get suggestions based on selected items
+    breads_suggested, veg_curries_suggested, non_veg_curries_suggested, biryani_suggested = suggest_pairings(selected_items)
+    
+    # Combine suggestions into one string for easy display
+    suggestions = f"Suggested Pairings:\n\nBreads: {', '.join(breads_suggested)}\nVeg Curries: {', '.join(veg_curries_suggested)}\nNon-Veg Curries: {', '.join(non_veg_curries_suggested)}\nBiryani: {', '.join(biryani_suggested)}"
+    
+    return suggestions, f"Custom Ingredient: {custom_ingredient}"
 
-    return interface
+# Gradio UI
+def create_ui():
+    # Define the menu with all items for selection
+    menu_items = breads + veg_curries + non_veg_curries + biryani
+    
+    # Gradio interface with the selection box and custom input box
+    with gr.Blocks() as demo:
+        with gr.Row():
+            menu_box = gr.CheckboxGroup(label="Select Menu Items", choices=menu_items, type="value")
+            custom_box = gr.Textbox(label="Customize Dish (Add Ingredients)", placeholder="e.g., Extra Cheese, Spicy", lines=2)
+        
+        with gr.Row():
+            next_button = gr.Button("Next")
+
+        suggestions_output = gr.Textbox(label="Suggestions Based on Your Selection", interactive=False)
+        
+        # Define the event when "Next" is clicked
+        next_button.click(fn=menu_selection, inputs=[menu_box, custom_box], outputs=[suggestions_output, custom_box])
+    
+    return demo
 
-# Launch the Gradio interface
-interface = create_app()
-interface.launch()
+# Run the app
+demo = create_ui()
+demo.launch()