Update app.py

app.py

@@ -35,8 +35,10 @@ extras = [
 # Filter dishes by category
 def filter_dishes(category):
     if category == "ALL":
-        return menu_data
-    return [dish for dish in menu_data if dish["category"] == category]
+        filtered_menu = menu_data
+    else:
+        filtered_menu = [dish for dish in menu_data if dish["category"] == category]
+    return generate_dish_html(filtered_menu)
 
 # Generate dish cards with "Add" button
 def display_dishes(category):