Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-# Static menu data
+# Menu data
 menu_data = [
     {"name": "Veg Burger", "category": "VEGAN",
      "image": "https://upload.wikimedia.org/wikipedia/commons/6/6e/Veggie_burger.jpg",
@@ -22,65 +22,65 @@ menu_data = [
 
 # Function to filter dishes based on category
 def filter_dishes(category):
-    return [dish["name"] for dish in menu_data if category == "ALL" or dish["category"] == category]
+    if category == "ALL":
+        return [dish["name"] for dish in menu_data]
+    return [dish["name"] for dish in menu_data if dish["category"] == category]
 
-# Function to get dish details
+# Function to display dish details
 def get_dish_details(dish_name):
     fallback_image = "https://via.placeholder.com/300x200.png?text=No+Image+Available"
+    fallback_text = "Information not available"
     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"
+    return fallback_image, fallback_text, fallback_text, fallback_text, fallback_text
 
 # Main Gradio app
-def create_ui():
-    with gr.Blocks() as demo:
-        gr.HTML("<h1 style='text-align: center;'>Biryani Hub Menu</h1>")
+with gr.Blocks() as demo:
+    gr.HTML("<h1 style='text-align: center;'>Biryani Hub Menu</h1>")
 
-        # Category Selection Buttons
-        gr.Markdown("### Select a Category to View Available Dishes")
-        with gr.Row():
-            btn_all = gr.Button("ALL")
-            btn_vegan = gr.Button("VEGAN")
-            btn_halal = gr.Button("HALAL")
+    # Buttons for 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")
 
-        # Dropdown for Dish Selection
-        gr.Markdown("### Available Dishes")
-        dish_dropdown = gr.Dropdown(choices=filter_dishes("ALL"), label="Select a Dish", interactive=True)
+    # Dropdown for dish selection
+    gr.Markdown("### Available Dishes")
+    dish_dropdown = gr.Dropdown(choices=[], label="Select a Dish", interactive=True)
 
-        # Dish Details Section
-        gr.Markdown("### Dish Details")
-        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 = gr.Textbox(label="Spice Level", interactive=False)
-        dish_portion = gr.Textbox(label="Portion Size", interactive=False)
+    # Dish details
+    gr.Markdown("### Dish Details")
+    dish_image = gr.Image(label="Dish Image")
+    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)
 
-        # Function to update dropdown list based on category
-        def update_dropdown(category):
-            dishes = filter_dishes(category)
-            return gr.update(choices=dishes, value=None), "Select a dish to view details", "", "", "", ""
+    # Function to update dropdown list based on category
+    def update_dropdown(category):
+        dishes = filter_dishes(category)
+        return gr.update(choices=dishes, value=None)
 
-        # Connect category buttons to dropdown update
-        btn_all.click(fn=lambda: update_dropdown("ALL"), outputs=[dish_dropdown, dish_description, dish_price, dish_spice, dish_portion, dish_image])
-        btn_vegan.click(fn=lambda: update_dropdown("VEGAN"), outputs=[dish_dropdown, dish_description, dish_price, dish_spice, dish_portion, dish_image])
-        btn_halal.click(fn=lambda: update_dropdown("HALAL"), outputs=[dish_dropdown, dish_description, dish_price, dish_spice, dish_portion, dish_image])
+    # Function to update dish details
+    def update_dish_details(dish_name):
+        return get_dish_details(dish_name)
 
-        # Update dish details when a dish is selected
-        def show_dish_details(dish_name):
-            image, description, price, spice_level, portion_size = get_dish_details(dish_name)
-            return image, description, price, spice_level, portion_size
+    # Button events for category selection
+    btn_all.click(fn=lambda: update_dropdown("ALL"), outputs=dish_dropdown)
+    btn_vegan.click(fn=lambda: update_dropdown("VEGAN"), outputs=dish_dropdown)
+    btn_halal.click(fn=lambda: update_dropdown("HALAL"), outputs=dish_dropdown)
 
-        dish_dropdown.change(fn=show_dish_details, inputs=dish_dropdown, outputs=[dish_image, dish_description, dish_price, dish_spice, dish_portion])
-
-    return demo
+    # Dish dropdown event for displaying details
+    dish_dropdown.change(fn=update_dish_details, inputs=dish_dropdown,
+                         outputs=[dish_image, dish_description, dish_price, dish_spice_level, dish_portion_size])
 
 # Launch the app
-if __name__ == "__main__":
-    create_ui().launch()
+demo.launch()