Update app.py

app.py

@@ -14,16 +14,14 @@ def filter_menu(preference):
     # Load menu data
     menu_data = load_menu()
 
-    # 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}")
-
-    # Filter data based on preference
-    if preference != "All":
-        filtered_data = menu_data[menu_data["Category"].str.contains(preference, case=False, na=False)]
-    else:
+    # Define filter conditions
+    if preference == "Halal/Non-Veg":
+        filtered_data = menu_data[menu_data["Ingredients"].str.contains("Chicken|Mutton|Fish|Prawns|Goat", case=False, na=False)]
+    elif preference == "Vegetarian":
+        filtered_data = menu_data[~menu_data["Ingredients"].str.contains("Chicken|Mutton|Fish|Prawns|Goat", case=False, na=False)]
+    elif preference == "Guilt-Free":
+        filtered_data = menu_data[menu_data["Description"].str.contains(r"Fat: ([0-9]|10)g", case=False, na=False)]
+    else:  # Default to "All"
         filtered_data = menu_data
 
     # Generate HTML for the menu
@@ -46,13 +44,12 @@ def filter_menu(preference):
 
 # Gradio app definition
 def app():
-    with gr.Blocks(title="Dynamic Menu with Preferences") as demo:
-        # Add a title
+    with gr.Blocks(title="Dynamic Menu with Filters") as demo:
         gr.Markdown("## Dynamic Menu with Preferences")
 
         # Radio button for selecting preference
         selected_preference = gr.Radio(
-            choices=["All", "Vegan", "Halal", "Guilt-Free"],
+            choices=["All", "Vegetarian", "Halal/Non-Veg", "Guilt-Free"],
             value="All",
             label="Choose a Preference",
         )
@@ -63,9 +60,13 @@ def app():
         # Define interactivity
         selected_preference.change(filter_menu, inputs=[selected_preference], outputs=[menu_output])
 
+        # Layout
+        gr.Row([selected_preference])
+        gr.Row(menu_output)
+
     return demo
 
 # Run the app
 if __name__ == "__main__":
     demo = app()
-    demo.launch()
+    demo.launch()