Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 # Extended menu data with dish details
 menu_data = [
     {"name": "Veg Burger", "category": "VEGAN",
-     "image": "pizza_main.jpg",
+     "image": "https://upload.wikimedia.org/wikipedia/commons/6/6e/Veggie_burger.jpg",
      "description": "A delicious vegan burger with plant-based patty, lettuce, and tomato.",
      "spice_level": "Medium", "portion_size": "Regular"},
     {"name": "Chicken Biryani", "category": "HALAL",
@@ -24,6 +24,28 @@ menu_data = [
      "spice_level": "Mild", "portion_size": "Small"}
 ]
 
+# Define available dishes and their portion sizes with image paths
+dish_data = {
+    "Veg Burger": {
+        "main": "https://upload.wikimedia.org/wikipedia/commons/6/6e/Veggie_burger.jpg",
+        "small": "https://via.placeholder.com/150?text=Veg+Burger+Small",
+        "medium": "https://via.placeholder.com/150?text=Veg+Burger+Medium",
+        "large": "https://via.placeholder.com/150?text=Veg+Burger+Large",
+    },
+    "Chicken Biryani": {
+        "main": "https://upload.wikimedia.org/wikipedia/commons/7/79/Chicken_Biryani.jpg",
+        "small": "https://via.placeholder.com/150?text=Chicken+Biryani+Small",
+        "medium": "https://via.placeholder.com/150?text=Chicken+Biryani+Medium",
+        "large": "https://via.placeholder.com/150?text=Chicken+Biryani+Large",
+    },
+    "Paneer Butter Masala": {
+        "main": "https://upload.wikimedia.org/wikipedia/commons/3/3d/Paneer_Butter_Masala.jpg",
+        "small": "https://via.placeholder.com/150?text=Paneer+Small",
+        "medium": "https://via.placeholder.com/150?text=Paneer+Medium",
+        "large": "https://via.placeholder.com/150?text=Paneer+Large",
+    },
+}
+
 spice_options = ["🌶 Mild", "🌶🌶 Medium", "🌶🌶🌶 High"]
 portion_options = ["🧆 Small", "🍽 Medium", "🍛 Large"]
 
@@ -52,10 +74,16 @@ def display_dishes(category):
 
 # Function to fetch details of the selected dish
 def get_dish_details(dish_name):
-    for dish in menu_data:
-        if dish["name"] == dish_name:
-            return dish["image"], dish["name"], dish["description"], "🌶 Medium", "🍽 Medium"
-    return "https://via.placeholder.com/300x200", "Unknown Dish", "No description available.", "N/A", "N/A"
+    if dish_name in dish_data:
+        main_image = dish_data[dish_name]["main"]
+        return main_image, dish_name, "Select a portion size below:"
+    return "https://via.placeholder.com/300x200", "Unknown Dish", "No description available."
+
+# Function to display the portion size image
+def display_portion(dish_name, portion):
+    if dish_name in dish_data:
+        return dish_data[dish_name].get(portion, None)
+    return None
 
 # Function to simulate placing an order
 def place_order(dish_name, spice_level, portion_size):
@@ -84,7 +112,8 @@ with gr.Blocks() as demo:
             dish_name = gr.Textbox(label="Dish Name", interactive=False)
             dish_description = gr.Textbox(label="Description", interactive=False)
             spice_dropdown = gr.Radio(label="Select Spice Level", choices=spice_options, value="🌶 Medium")
-            portion_dropdown = gr.Radio(label="Select Portion Size", choices=portion_options, value="🍽 Medium")
+            portion_dropdown = gr.Radio(label="Select Portion Size", choices=["small", "medium", "large"], value="medium")
+            portion_image = gr.Image(label="Portion Size Image")
 
     # Place Order Button
     place_order_btn = gr.Button("Place Order")
@@ -100,13 +129,16 @@ with gr.Blocks() as demo:
 
     # Event: Show dish details when a dish is selected
     def show_dish_details(dish_name):
-        image, name, description, spice, portion = get_dish_details(dish_name)
-        return image, name, description, spice, portion
+        image, name, description = get_dish_details(dish_name)
+        return image, name, description
 
     # Hidden input to receive dish name from JavaScript
     selected_dish_name = gr.Textbox(visible=False)
     selected_dish_name.change(fn=show_dish_details, inputs=selected_dish_name,
-                              outputs=[dish_image, dish_name, dish_description, spice_dropdown, portion_dropdown])
+                              outputs=[dish_image, dish_name, dish_description])
+
+    # Event: Display portion size image
+    portion_dropdown.change(fn=display_portion, inputs=[selected_dish_name, portion_dropdown], outputs=portion_image)
 
     # Event: Place order button
     place_order_btn.click(fn=place_order,