Update app.py

app.py

@@ -1,59 +1,72 @@
+menu = {
+    "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"]
+}
+def suggest_pairings(selected_dishes):
+    # Default suggestions
+    suggested_pairings = {
+        "veg_curries": [],
+        "non_veg_curries": [],
+        "biryani": [],
+        "breads": [],
+    }
+    
+    # If bread is selected
+    if any(bread in selected_dishes for bread in menu['breads']):
+        suggested_pairings["veg_curries"].extend(menu["veg_curries"])
+        suggested_pairings["non_veg_curries"].extend(menu["non_veg_curries"])
+
+    # If biryani is selected
+    if any(biryani in selected_dishes for biryani in menu["biryani"]):
+        suggested_pairings["veg_curries"].extend(menu["veg_curries"])
+        suggested_pairings["non_veg_curries"].extend(menu["non_veg_curries"])
+
+    # If only curries (veg or non-veg) are selected, suggest breads and biryanis
+    if any(curry in selected_dishes for curry in menu["veg_curries"] + menu["non_veg_curries"]):
+        suggested_pairings["breads"].extend(menu["breads"])
+        suggested_pairings["biryani"].extend(menu["biryani"])
+
+    return suggested_pairings
 import gradio as gr
 
-# 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']
-
-# 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)
+# Function to handle dish selection and show pairings
+def dish_pairing(selected_bread, selected_curry, selected_biryani):
+    selected_dishes = [selected_bread, selected_curry, selected_biryani]
+    
+    # Get the pairings based on selected dishes
+    pairings = suggest_pairings(selected_dishes)
+    
+    # Create a response with the pairings
+    pairing_output = f"Suggested Pairings:\n\n"
+    
+    if pairings["veg_curries"]:
+        pairing_output += f"Veg Curries: {', '.join(pairings['veg_curries'])}\n"
+    if pairings["non_veg_curries"]:
+        pairing_output += f"Non-Veg Curries: {', '.join(pairings['non_veg_curries'])}\n"
+    if pairings["biryani"]:
+        pairing_output += f"Biryani: {', '.join(pairings['biryani'])}\n"
+    if pairings["breads"]:
+        pairing_output += f"Breads: {', '.join(pairings['breads'])}\n"
+
+    return pairing_output
+
+
+# Create Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Dish Pairing App")
     
-    # 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()
+    with gr.Row():
+        bread_input = gr.Radio(menu["breads"], label="Select a Bread", type="value")
+        curry_input = gr.Radio(menu["veg_curries"] + menu["non_veg_curries"], label="Select a Curry", type="value")
+        biryani_input = gr.Radio(menu["biryani"], label="Select a Biryani", type="value")
+
+    output = gr.Textbox(label="Suggested Pairings")
+    
+    # Button to trigger pairing suggestions
+    submit_btn = gr.Button("Show Pairings")
+    
+    submit_btn.click(dish_pairing, inputs=[bread_input, curry_input, biryani_input], outputs=output)
+
+demo.launch()