nagasurendra commited on
Commit
f0ce3bc
·
verified ·
1 Parent(s): c05d8d3

Create menu.py

Browse files
Files changed (1) hide show
  1. routes/menu.py +10 -0
routes/menu.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ from utils.file_initializer import MENU_FILE
3
+
4
+ def load_menu(preferences=None, allergies=None):
5
+ menu = pd.read_excel(MENU_FILE)
6
+ if preferences:
7
+ menu = menu[menu["Preferences"].str.contains(preferences, na=False)]
8
+ if allergies:
9
+ menu = menu[~menu["Allergens"].str.contains(allergies, na=False)]
10
+ return menu