Update app.py

app.py

@@ -1,82 +1,86 @@
 import gradio as gr
 
 # Menu data with all details
-menu_data = {
-    "ALL": [
-        {"name": "Veg Burger", "category": "VEGAN", 
-         "image": "https://upload.wikimedia.org/wikipedia/commons/6/6e/Veggie_burger.jpg",
-         "description": "A delicious vegan burger with plant-based patty, lettuce, and tomato.",
-         "price": "$8.99", "spice_level": "Medium", "portion_size": "Regular"},
-        {"name": "Chicken Biryani", "category": "HALAL", 
-         "image": "https://upload.wikimedia.org/wikipedia/commons/7/79/Chicken_Biryani.jpg",
-         "description": "Spicy chicken biryani with aromatic basmati rice and tender chicken pieces.",
-         "price": "$12.99", "spice_level": "High", "portion_size": "Large"},
-        {"name": "Paneer Butter Masala", "category": "VEGAN", 
-         "image": "https://upload.wikimedia.org/wikipedia/commons/3/3d/Paneer_Butter_Masala.jpg",
-         "description": "Paneer cooked in a rich and creamy tomato-based gravy.",
-         "price": "$10.99", "spice_level": "Mild", "portion_size": "Medium"},
-        {"name": "Beef Steak", "category": "HALAL", 
-         "image": "https://upload.wikimedia.org/wikipedia/commons/d/d5/Beef_Steak.jpg",
-         "description": "Juicy beef steak served with mashed potatoes and grilled vegetables.",
-         "price": "$14.99", "spice_level": "Medium", "portion_size": "Large"}
-    ]
-}
+menu_data = [
+    {"name": "Veg Burger", "category": "VEGAN",
+     "image": "https://upload.wikimedia.org/wikipedia/commons/6/6e/Veggie_burger.jpg",
+     "description": "A delicious vegan burger with plant-based patty, lettuce, and tomato.",
+     "price": "$8.99", "spice_level": "Medium", "portion_size": "Regular"},
+    {"name": "Chicken Biryani", "category": "HALAL",
+     "image": "https://upload.wikimedia.org/wikipedia/commons/7/79/Chicken_Biryani.jpg",
+     "description": "Spicy chicken biryani with aromatic basmati rice and tender chicken pieces.",
+     "price": "$12.99", "spice_level": "High", "portion_size": "Large"},
+    {"name": "Paneer Butter Masala", "category": "VEGAN",
+     "image": "https://upload.wikimedia.org/wikipedia/commons/3/3d/Paneer_Butter_Masala.jpg",
+     "description": "Paneer cooked in a rich and creamy tomato-based gravy.",
+     "price": "$10.99", "spice_level": "Mild", "portion_size": "Medium"},
+    {"name": "Beef Steak", "category": "HALAL",
+     "image": "https://upload.wikimedia.org/wikipedia/commons/d/d5/Beef_Steak.jpg",
+     "description": "Juicy beef steak served with mashed potatoes and grilled vegetables.",
+     "price": "$14.99", "spice_level": "Medium", "portion_size": "Large"}
+]
 
 # Function to filter dishes based on category
 def filter_dishes(category):
-    return [dish["name"] for dish in menu_data["ALL"] if category == "ALL" or dish["category"] == category]
+    return [dish["name"] for dish in menu_data if category == "ALL" or dish["category"] == category]
 
-# Function to display details for a selected dish
+# Function to fetch details for a selected dish
 def get_dish_details(dish_name):
+    # Fallback values for missing or invalid data
     fallback_image = "https://via.placeholder.com/300x200.png?text=No+Image+Available"
-    for dish in menu_data["ALL"]:
+    fallback_text = "Information not available"
+    
+    # Find the selected dish
+    for dish in menu_data:
         if dish["name"] == dish_name:
             return (
                 dish.get("image", fallback_image),
-                dish.get("description", "No description available."),
-                dish.get("price", "N/A"),
-                dish.get("spice_level", "N/A"),
-                dish.get("portion_size", "N/A"),
+                dish.get("description", fallback_text),
+                dish.get("price", fallback_text),
+                dish.get("spice_level", fallback_text),
+                dish.get("portion_size", fallback_text),
             )
-    return fallback_image, "Dish not found.", "N/A", "N/A", "N/A"
+    # If dish is not found, return fallbacks
+    return fallback_image, fallback_text, fallback_text, fallback_text, fallback_text
 
-# Main Gradio app
+# Main Gradio App
 with gr.Blocks() as demo:
     gr.HTML("<h1 style='text-align: center;'>Biryani Hub Menu</h1>")
 
-    # Category Buttons
+    # Category Selection
     gr.Markdown("### Select a Category")
     with gr.Row():
         btn_all = gr.Button("ALL")
         btn_vegan = gr.Button("VEGAN")
         btn_halal = gr.Button("HALAL")
 
-    # Dish Selection (Radio Button)
+    # Dish Selection Radio
     gr.Markdown("### Available Dishes")
     dish_radio = gr.Radio(choices=filter_dishes("ALL"), label="Select a Dish", interactive=True)
 
     # Dish Details
     gr.Markdown("### Dish Details")
-    dish_image = gr.Image(label="Dish Image")
+    dish_image = gr.Image(label="Dish Image", type="pil")
     dish_description = gr.Textbox(label="Description", interactive=False)
     dish_price = gr.Textbox(label="Price", interactive=False)
     dish_spice_level = gr.Textbox(label="Spice Level", interactive=False)
     dish_portion_size = gr.Textbox(label="Portion Size", interactive=False)
 
-    # Event: Update dish list when a category button is clicked
+    # Update the dish list when category buttons are clicked
     def update_dishes(category):
-        return gr.update(choices=filter_dishes(category), value=None)
+        new_dishes = filter_dishes(category)
+        return gr.update(choices=new_dishes, value=None), "", "", "", "", ""
 
-    btn_all.click(fn=lambda: update_dishes("ALL"), outputs=dish_radio)
-    btn_vegan.click(fn=lambda: update_dishes("VEGAN"), outputs=dish_radio)
-    btn_halal.click(fn=lambda: update_dishes("HALAL"), outputs=dish_radio)
+    btn_all.click(fn=lambda: update_dishes("ALL"), outputs=[dish_radio, dish_image, dish_description, dish_price, dish_spice_level, dish_portion_size])
+    btn_vegan.click(fn=lambda: update_dishes("VEGAN"), outputs=[dish_radio, dish_image, dish_description, dish_price, dish_spice_level, dish_portion_size])
+    btn_halal.click(fn=lambda: update_dishes("HALAL"), outputs=[dish_radio, dish_image, dish_description, dish_price, dish_spice_level, dish_portion_size])
 
-    # Event: Show dish details when a dish is selected
+    # Update dish details when a dish is selected
     dish_radio.change(
         fn=get_dish_details,
         inputs=dish_radio,
         outputs=[dish_image, dish_description, dish_price, dish_spice_level, dish_portion_size]
     )
 
-# Launch the app
+# Run the Gradio app
 demo.launch()