Update app.py

app.py

@@ -1,67 +1,59 @@
 import gradio as gr
 
-# Define menu items
+# 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']
 
-# Combine everything into one picklist for user selection
-menu_items = breads + veg_curries + non_veg_curries + biryani
-
-# Function to generate suggestions based on selected items
-def generate_suggestions(selected_items):
-    suggestions = []
-    
-    # If bread is selected, suggest curries (veg and non-veg)
-    if any(item in selected_items for item in breads):
-        suggestions.extend(veg_curries + non_veg_curries)
-        
-    # If any curry is selected, suggest breads and biryani
-    if any(item in selected_items for item in veg_curries + non_veg_curries):
-        suggestions.extend(breads + biryani)
-        
-    # If any biryani is selected, suggest breads and curries
-    if any(item in selected_items for item in biryani):
-        suggestions.extend(breads + veg_curries + non_veg_curries)
-    
-    # Remove already selected items from the suggestions
-    suggestions = [item for item in suggestions if item not in selected_items]
+# Define the logic for suggestions
+def get_suggestions(selected_items):
+    # If bread is selected, suggest curries and biryanis
+    if any(bread in selected_items for bread in breads):
+        suggestions = veg_curries + non_veg_curries
+        remaining_category = biryani  # suggest biryanis if bread is selected
+    # If curries are selected, suggest bread and biryanis
+    elif any(curry in selected_items for curry in veg_curries + non_veg_curries):
+        suggestions = breads  # suggest breads if curries are selected
+        remaining_category = biryani  # suggest biryanis if curries are selected
+    # If biryanis are selected, suggest breads and curries
+    elif any(bir in selected_items for bir in biryani):
+        suggestions = breads + veg_curries + non_veg_curries  # suggest breads and curries if biryanis are selected
+        remaining_category = []  # no remaining category if biryanis are selected
+    else:
+        suggestions = []
+        remaining_category = []
+
+    return suggestions, remaining_category
+
+# Define the customization options box
+def customize_selections(selected_items, custom_ingredients):
+    return f"Selected Items: {', '.join(selected_items)}\nCustom Ingredients: {custom_ingredients}"
+
+# Gradio interface
+def select_menu_and_customize(menu_items, custom_ingredients):
+    suggestions, remaining_category = get_suggestions(menu_items)
     
-    return suggestions
-
-# Function for customizing the selected dishes (optional ingredients)
-def customize_dishes(selected_items, custom_ingredients):
-    return f"Selected dishes: {', '.join(selected_items)}\nCustom ingredients: {custom_ingredients}"
-
-# Define Gradio interface
-def create_interface():
-    with gr.Blocks() as demo:
-        with gr.Row():
-            menu = gr.CheckboxGroup(choices=menu_items, label="Select Menu Items", interactive=True)
-            suggestion_box = gr.Dropdown(label="Suggestions", interactive=True)
-            custom_box = gr.Textbox(label="Add Custom Ingredients", placeholder="e.g., extra cheese, spicy", lines=2)
-            next_button = gr.Button("Next")
-            output_box = gr.Textbox(label="Final Selection", placeholder="Your final selection will appear here")
-
-        # Function to handle the logic when the user selects items
-        def handle_selection(selected_items, custom_ingredients):
-            # Get pairing suggestions based on selected items
-            suggestions = generate_suggestions(selected_items)
-            suggestion_box.update(choices=suggestions)
-            # Add custom ingredients to the selection
-            final_selection = customize_dishes(selected_items, custom_ingredients)
-            output_box.update(value=final_selection)
-            return suggestions, final_selection
-        
-        # Set up the interaction logic
-        menu.change(fn=handle_selection, inputs=[menu, custom_box], outputs=[suggestion_box, output_box])
-        next_button.click(fn=handle_selection, inputs=[menu, custom_box], outputs=output_box)
-
-    return demo
-
-# Create the interface
-demo = create_interface()
-
-# Launch the Gradio app
-demo.launch()
+    # Return suggestions for the next selection step
+    return suggestions, remaining_category, customize_selections(menu_items, custom_ingredients)
+
+# Create the Gradio interface
+menu_options = breads + veg_curries + non_veg_curries + biryani
+
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=select_menu_and_customize,
+    inputs=[
+        gr.CheckboxGroup(choices=menu_options, label="Select your menu items"),
+        gr.Textbox(placeholder="Add any custom ingredients here", label="Custom Ingredients")
+    ],
+    outputs=[
+        gr.CheckboxGroup(label="Suggestions based on your selections"),
+        gr.Textbox(label="Remaining options"),
+        gr.Textbox(label="Final Order Summary")
+    ],
+    live=True,  # This ensures suggestions update as the user makes selections
+)
+
+# Launch the app
+interface.launch()