Update app.py

app.py

@@ -1,74 +1,112 @@
 import gradio as gr
 
-# Define available dishes and their portion sizes with image paths
+# Define available dishes and their portion sizes with image paths and categories
 dish_data = {
     "Pizza": {
-        "main": "pizza_main.jpg",
-        "small": "pizza_small.jpg",
-        "medium": "pizza_medium.jpg",
-        "large": "pizza_large.jpg",
+        "main": "images/pizza_main.jpg",
+        "small": "images/pizza_small.jpg",
+        "medium": "images/pizza_medium.jpg",
+        "large": "images/pizza_large.jpg",
+        "category": "All",
     },
-    "Burger": {
-        "main": "images/burger_main.jpg",
-        "small": "images/burger_small.jpg",
-        "medium": "images/burger_medium.jpg",
-        "large": "images/burger_large.jpg",
+    "Vegan Salad": {
+        "main": "images/vegan_salad_main.jpg",
+        "small": "images/vegan_salad_small.jpg",
+        "medium": "images/vegan_salad_medium.jpg",
+        "large": "images/vegan_salad_large.jpg",
+        "category": "Vegan",
     },
-    "Pasta": {
-        "main": "images/pasta_main.jpg",
-        "small": "images/pasta_small.jpg",
-        "medium": "images/pasta_medium.jpg",
-        "large": "images/pasta_large.jpg",
+    "Halal Chicken": {
+        "main": "images/halal_chicken_main.jpg",
+        "small": "images/halal_chicken_small.jpg",
+        "medium": "images/halal_chicken_medium.jpg",
+        "large": "images/halal_chicken_large.jpg",
+        "category": "Halal",
+    },
+    "Allergenic Soup": {
+        "main": "images/allergenic_soup_main.jpg",
+        "small": "images/allergenic_soup_small.jpg",
+        "medium": "images/allergenic_soup_medium.jpg",
+        "large": "images/allergenic_soup_large.jpg",
+        "category": "Allergenic",
     },
 }
 
-# Function to handle displaying the main dish image and enabling portion buttons
+def filter_dishes(category):
+    return {dish: data for dish, data in dish_data.items() if category == "All" or data["category"] == category}
+
 def display_dish(dish):
     if dish not in dish_data:
-        return None, dish, gr.update(visible=False)
+        return None, dish, gr.update(visible=False), gr.update(visible=False)
     main_image = dish_data[dish]["main"]
-    return main_image, dish, gr.update(visible=True)
+    return main_image, dish, gr.update(visible=True), gr.update(visible=True)
 
-# Function to handle displaying the portion size image
 def display_portion(dish, portion):
     if dish not in dish_data:
         return None
     portion_image = dish_data[dish].get(portion, None)
     return portion_image
 
+def spice_level_selection(spice_level):
+    return f"Selected spice level: {spice_level}"
+
 # Define the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Food Ordering System")
 
-    # Section to display dishes as clickable buttons
-    with gr.Row():
-        selected_dish = gr.State()  # Store the selected dish
-        main_image = gr.Image(label="Dish Image")
-        dish_name_display = gr.Text(label="Dish Name")
-        portion_row = gr.Row(visible=False)  # Portion size buttons section
-        portion_image = gr.Image(label="Portion Size Image")
+    # Category selection
+    category_dropdown = gr.Dropdown(
+        choices=["All", "Vegan", "Halal", "Allergenic"],
+        value="All",
+        label="Select Category"
+    )
+
+    # Dynamic dish buttons
+    dish_buttons = gr.Row()
+
+    def update_dish_buttons(category):
+        filtered_dishes = filter_dishes(category)
+        buttons = []
+        for dish, data in filtered_dishes.items():
+            buttons.append(gr.Button(value=dish))
+        return buttons
+
+    category_dropdown.change(
+        fn=update_dish_buttons,
+        inputs=category_dropdown,
+        outputs=dish_buttons
+    )
 
-        dish_buttons = []
-        for dish, data in dish_data.items():
-            dish_buttons.append(gr.Button(value=dish, elem_id=f"button_{dish}"))
+    # Section to display the main image of the selected dish
+    selected_dish = gr.State()  # Store the selected dish
+    main_image = gr.Image(label="Dish Image")
+    dish_name_display = gr.Text(label="Dish Name")
+    portion_row = gr.Row(visible=False)  # Portion size buttons section
+    portion_image = gr.Image(label="Portion Size Image")
+    spice_row = gr.Row(visible=False)  # Spice level buttons section
 
-    # Section to display portion size buttons
+    # Portion size buttons
     with portion_row:
         portion_buttons = []
         for portion in ["small", "medium", "large"]:
             portion_buttons.append(gr.Button(value=portion.capitalize()))
 
+    # Spice level buttons
+    with spice_row:
+        spice_buttons = []
+        for spice in ["Mild", "Medium", "Spicy"]:
+            spice_buttons.append(gr.Button(value=spice))
+
     # Link dish buttons to display the selected dish
-    for button, dish in zip(dish_buttons, dish_data.keys()):
-        button.click(
-            fn=display_dish,
-            inputs=[gr.Textbox(value=dish, visible=False)],
-            outputs=[main_image, dish_name_display, portion_row],
-        )
+    def create_dish_click_handler(dish):
+        return lambda: display_dish(dish)
+
+    for dish in dish_data.keys():
+        button = gr.Button(value=dish, visible=False)
         button.click(
-            fn=lambda x: x,
-            inputs=[gr.Textbox(value=dish, visible=False)],
-            outputs=selected_dish,
+            fn=create_dish_click_handler(dish),
+            inputs=[],
+            outputs=[main_image, dish_name_display, portion_row, spice_row],
         )
 
     # Link portion buttons to display the portion image
@@ -79,5 +117,13 @@ with gr.Blocks() as demo:
             outputs=portion_image,
         )
 
+    # Link spice level buttons to display the spice level
+    for spice_button, spice in zip(spice_buttons, ["Mild", "Medium", "Spicy"]):
+        spice_button.click(
+            fn=spice_level_selection,
+            inputs=[gr.Textbox(value=spice, visible=False)],
+            outputs=dish_name_display
+        )
+
 # Launch the Gradio app
 demo.launch()