Update app.py

app.py

@@ -10,27 +10,40 @@ def load_data(file_url):
     gdown.download(file_url, file_path, quiet=False)
     return pd.read_excel(file_path, engine='openpyxl')
 
-# Function to generate the weekly menu
-def generate_menu(data, season, sub_category_pref, outings, dietary_restrictions):
+# Function to generate the weekly menu based on BMI, dietary restriction, and pantry sync
+def generate_menu(data, season, sub_category_pref, outings, dietary_restrictions, pantry_ingredients):
     season_data = data[(data['Season'] == season) | (data['Season'] == 'All Seasons')]
 
     meals_needed = 14 - outings  # Main Courses (14) minus outings
 
     menu = []
+    shopping_list = set()  # Track missing ingredients
+
     for sub_category, freq in sub_category_pref.items():
         sub_category_dishes = season_data[season_data['Sub-Category'] == sub_category]
-        menu.extend(sub_category_dishes.sample(min(freq, len(sub_category_dishes))).to_dict('records'))
+        
+        # Filter dishes based on dietary restrictions
+        if dietary_restrictions == 'Diabetic':
+            sub_category_dishes = sub_category_dishes[~sub_category_dishes['Ingredients'].str.contains('sugar', case=False, na=False)]
+        elif dietary_restrictions == 'Gluten-Free':
+            sub_category_dishes = sub_category_dishes[~sub_category_dishes['Ingredients'].str.contains('gluten', case=False, na=False)]
+        elif dietary_restrictions == 'Low-Sodium':
+            sub_category_dishes = sub_category_dishes[~sub_category_dishes['Ingredients'].str.contains('sodium', case=False, na=False)]
+
+        # Sample the dishes based on the user preference
+        selected_dishes = sub_category_dishes.sample(min(freq, len(sub_category_dishes)))
+
+        # Check the ingredients for the pantry and add missing items to the shopping list
+        for dish in selected_dishes.to_dict('records'):
+            ingredients = dish.get('Ingredients', '').split(',')
+            for ingredient in ingredients:
+                if ingredient.strip().lower() not in pantry_ingredients:
+                    shopping_list.add(ingredient.strip().lower())
+
+            menu.append(dish)
 
     menu = menu[:meals_needed]
 
-    # Filter out dishes based on dietary restrictions
-    if dietary_restrictions == 'Diabetic':
-        menu = [dish for dish in menu if 'sugar' not in dish['Ingredients'].lower()]
-    elif dietary_restrictions == 'Gluten-Free':
-        menu = [dish for dish in menu if 'gluten' not in dish['Ingredients'].lower()]
-    elif dietary_restrictions == 'Low-Sodium':
-        menu = [dish for dish in menu if 'sodium' not in dish['Ingredients'].lower()]
-
     outing_dishes = data[data['Home Made / Outing'] == 'Outing'].sample(outings).to_dict('records')
     menu.extend(outing_dishes)
 
@@ -47,7 +60,7 @@ def generate_menu(data, season, sub_category_pref, outings, dietary_restrictions
     lunch_menu = lunch_menu[["Name"]]
     dinner_menu = dinner_menu[["Name"]]
 
-    return lunch_menu, dinner_menu, total_cost
+    return lunch_menu, dinner_menu, total_cost, shopping_list
 
 # Function to calculate BMI and recommend diet
 def bmi_insights(weight, height_cm):
@@ -64,21 +77,8 @@ def bmi_insights(weight, height_cm):
         return f"Your BMI is {bmi:.2f}. You are obese. A calorie-restricted diet with lean proteins is recommended."
 
 # Function for Smart Inventory Management (Pantry Sync)
-def pantry_sync(available_ingredients, data):
-    # Suggest meals based on ingredients available in the pantry
-    suggested_dishes = []
-    for dish in data.to_dict('records'):
-        ingredients = dish.get('Ingredients', '').split(',')
-        if all(ingredient in available_ingredients for ingredient in ingredients):
-            suggested_dishes.append(dish)
-    return pd.DataFrame(suggested_dishes)
-
-# Function for Shopping List Generator
-def generate_shopping_list(menu):
-    ingredients = []
-    for dish in menu:
-        ingredients.extend(dish['Ingredients'].split(','))
-    return sorted(set(ingredients))
+def pantry_sync(available_ingredients):
+    return set(available_ingredients)
 
 # Main workflow
 def main():
@@ -125,8 +125,15 @@ def main():
         st.warning("The total number of main courses exceeds 14! Please adjust your preferences.")
         return
 
+    # Pantry Sync
+    pantry_ingredients_input = st.text_area("Enter available ingredients in your pantry (comma separated)")
+    if pantry_ingredients_input:
+        pantry_ingredients = pantry_sync(pantry_ingredients_input.split(','))
+    else:
+        pantry_ingredients = set()
+
     if st.button("Generate Menu"):
-        lunch_menu, dinner_menu, total_cost = generate_menu(data, season, sub_category_pref, outings, dietary_restrictions)
+        lunch_menu, dinner_menu, total_cost, shopping_list = generate_menu(data, season, sub_category_pref, outings, dietary_restrictions, pantry_ingredients)
 
         st.header("Weekly Menu")
         st.subheader("Lunch Menu")
@@ -137,6 +144,11 @@ def main():
 
         st.subheader(f"Total Expenditure for the Week: PKR {total_cost}")
 
+        # Shopping list for missing ingredients
+        if shopping_list:
+            st.subheader("Shopping List for Missing Ingredients")
+            st.write(sorted(shopping_list))
+
         # Save menu to Excel
         with pd.ExcelWriter("weekly_menu_split.xlsx") as writer:
             lunch_menu.to_excel(writer, sheet_name="Lunch", index=False)
@@ -144,20 +156,5 @@ def main():
 
         st.success("Weekly menu saved as 'weekly_menu_split.xlsx'")
 
-    # Pantry sync section
-    st.header("Smart Pantry Sync")
-    available_ingredients = st.text_area("Enter available ingredients in your pantry (comma separated)")
-    if available_ingredients:
-        available_ingredients = [item.strip() for item in available_ingredients.split(',')]
-        pantry_suggested_dishes = pantry_sync(available_ingredients, data)
-        st.subheader("Suggested Dishes Based on Pantry Ingredients")
-        st.write(pantry_suggested_dishes)
-
-    # Shopping list section
-    st.header("Generate Shopping List")
-    if st.button("Generate Shopping List"):
-        shopping_list = generate_shopping_list(lunch_menu.to_dict('records') + dinner_menu.to_dict('records'))
-        st.write("Shopping List:", shopping_list)
-
 if __name__ == "__main__":
     main()