Update app.py

app.py

@@ -9,86 +9,63 @@ def load_menu():
     except Exception as e:
         raise ValueError(f"Error loading menu file: {e}")
 
-# Function to get unique categories
-def get_categories():
+# Function to filter menu items based on preference
+def filter_menu(preference):
+    # Load menu data
     menu_data = load_menu()
-    return list(menu_data["Category"].unique())
 
-# Function to get dishes for a selected category
-def get_dishes(category):
-    menu_data = load_menu()
-    dishes = menu_data[menu_data["Category"] == category]["Dish Name"].tolist()
-    return dishes
+    # Check for the required columns
+    required_columns = ["Category", "Dish Name", "Price ($)", "Image URL", "Description"]
+    for col in required_columns:
+        if col not in menu_data.columns:
+            raise ValueError(f"Missing column in Excel file: {col}")
 
-# Function to display the menu for a selected dish
-def get_menu_item(dish_name):
-    menu_data = load_menu()
-    item = menu_data[menu_data["Dish Name"] == dish_name]
-    if item.empty:
-        return "No item found for the selected dish."
-    
-    # Extract item details
-    item = item.iloc[0]
-    html_content = f"""
-    <div style="display: flex; align-items: center; border: 1px solid #ddd; border-radius: 8px; padding: 15px; margin-bottom: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
-        <div style="flex: 1; margin-right: 15px;">
-            <h3 style="margin: 0; font-size: 18px;">{item['Dish Name']}</h3>
-            <p style="margin: 5px 0; font-size: 16px; color: #888;">${item['Price ($)']}</p>
-            <p style="margin: 5px 0; font-size: 14px; color: #555;">{item['Description']}</p>
-        </div>
-        <div style="flex-shrink: 0; text-align: center;">
-            <img src="{item['Image URL']}" alt="{item['Dish Name']}" style="width: 100px; height: 100px; border-radius: 8px; object-fit: cover; margin-bottom: 10px;">
-            <button style="background-color: #28a745; color: white; border: none; padding: 8px 15px; font-size: 14px; border-radius: 5px; cursor: pointer;">Add</button>
+    # Filter data based on preference
+    if preference != "All":
+        filtered_data = menu_data[menu_data["Category"].str.contains(preference, case=False, na=False)]
+    else:
+        filtered_data = menu_data
+
+    # Generate HTML for the menu
+    html_content = ""
+    for _, item in filtered_data.iterrows():
+        html_content += f"""
+        <div style="display: flex; align-items: center; border: 1px solid #ddd; border-radius: 8px; padding: 15px; margin-bottom: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
+            <div style="flex: 1; margin-right: 15px;">
+                <h3 style="margin: 0; font-size: 18px;">{item['Dish Name']}</h3>
+                <p style="margin: 5px 0; font-size: 16px; color: #888;">${item['Price ($)']}</p>
+                <p style="margin: 5px 0; font-size: 14px; color: #555;">{item['Description']}</p>
+            </div>
+            <div style="flex-shrink: 0; text-align: center;">
+                <img src="{item['Image URL']}" alt="{item['Dish Name']}" style="width: 100px; height: 100px; border-radius: 8px; object-fit: cover; margin-bottom: 10px;">
+                <button style="background-color: #28a745; color: white; border: none; padding: 8px 15px; font-size: 14px; border-radius: 5px; cursor: pointer;">Add</button>
+            </div>
         </div>
-    </div>
-    """
+        """
     return html_content
 
 # Gradio app definition
 def app():
-    with gr.Blocks(title="Dynamic Menu with Categories") as demo:
-        gr.Markdown("## Dynamic Menu with Categories")
-
-        # Dropdown for selecting a category
-        category_dropdown = gr.Dropdown(
-            choices=get_categories(),
-            value="Appetizer",
-            label="Select a Category",
-        )
+    with gr.Blocks(title="Dynamic Menu with Preferences") as demo:
+        # Add a title
+        gr.Markdown("## Dynamic Menu with Preferences")
 
-        # Dropdown for selecting a dish
-        dish_dropdown = gr.Dropdown(
-            choices=get_dishes("Appetizer"),
-            value="Samosa",
-            label="Select a Dish",
+        # Radio button for selecting preference
+        selected_preference = gr.Radio(
+            choices=["All", "Vegan", "Halal", "Guilt-Free"],
+            value="All",
+            label="Choose a Preference",
         )
 
-        # Output area for the selected menu item
-        menu_output = gr.HTML()
+        # Output area for menu items
+        menu_output = gr.HTML(value=filter_menu("All"))
 
         # Define interactivity
-        def update_dishes(category):
-            return gr.update(choices=get_dishes(category), value=get_dishes(category)[0])
-
-        category_dropdown.change(
-            update_dishes,
-            inputs=[category_dropdown],
-            outputs=[dish_dropdown]
-        )
-
-        dish_dropdown.change(
-            get_menu_item,
-            inputs=[dish_dropdown],
-            outputs=[menu_output]
-        )
-
-        # Layout
-        gr.Row([category_dropdown, dish_dropdown])
-        gr.Row(menu_output)
+        selected_preference.change(filter_menu, inputs=[selected_preference], outputs=[menu_output])
 
     return demo
 
 # Run the app
 if __name__ == "__main__":
     demo = app()
-    demo.launch()
+    demo.launch()