Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 import gdown
 from PIL import Image
+import random
 
 # Function to load data from Google Drive
 def load_data(file_url):
@@ -10,7 +11,7 @@ def load_data(file_url):
     return pd.read_excel(file_path, engine='openpyxl')
 
 # Function to generate the weekly menu
-def generate_menu(data, season, sub_category_pref, outings):
+def generate_menu(data, season, sub_category_pref, outings, dietary_restrictions):
     season_data = data[(data['Season'] == season) | (data['Season'] == 'All Seasons')]
 
     meals_needed = 14 - outings  # Main Courses (14) minus outings
@@ -22,6 +23,14 @@ def generate_menu(data, season, sub_category_pref, outings):
 
     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)
 
@@ -42,23 +51,39 @@ def generate_menu(data, season, sub_category_pref, outings):
 
 # Function to calculate BMI and recommend diet
 def bmi_insights(weight, height_cm):
-    # Convert height from cm to meters
     height_m = height_cm / 100
     bmi = weight / (height_m ** 2)
 
     if bmi < 18.5:
-        return f"Your BMI is {bmi:.2f}. You are underweight. A diet rich in proteins and healthy fats would be beneficial. Consider incorporating more calories and nutritious snacks."
+        return f"Your BMI is {bmi:.2f}. You are underweight. A diet rich in proteins and healthy fats would be beneficial."
     elif 18.5 <= bmi < 24.9:
-        return f"Your BMI is {bmi:.2f}. You are in the normal weight range. Maintain a balanced diet, including fruits, vegetables, whole grains, and lean proteins."
+        return f"Your BMI is {bmi:.2f}. You are in the normal weight range. Maintain a balanced diet."
     elif 25 <= bmi < 29.9:
-        return f"Your BMI is {bmi:.2f}. You are overweight. A diet with fewer carbs and more lean proteins can help manage your weight. Consider reducing sugar intake."
+        return f"Your BMI is {bmi:.2f}. You are overweight. Consider reducing sugar intake."
     else:
-        return f"Your BMI is {bmi:.2f}. You are obese. A calorie-restricted diet with a focus on whole foods, vegetables, and lean proteins is recommended. Consult a health professional for personalized guidance."
+        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))
 
 # Main workflow
 def main():
     st.title("Roz Roz Ka Masla..... Aaj Kya Pakayen!!! ")
-    st.title("Now generate weekly Menu with BMI Insights")
+    st.title("Now generate weekly Menu with BMI Insights and more features")
     
     # Add food image at the top right of the app
     image = Image.open("food_image.jpg")  # Ensure you have an image in the same folder
@@ -82,6 +107,9 @@ def main():
 
     outings = st.number_input("Enter the number of outings this week", min_value=0, max_value=14, value=0)
 
+    # Dietary restrictions
+    dietary_restrictions = st.selectbox("Select any dietary restrictions", ["None", "Diabetic", "Gluten-Free", "Low-Sodium"])
+
     st.header("How many dishes from each sub-category would you like?")
     sub_category_pref = {
         'Sabzi': st.number_input("Number of Sabzi dishes", min_value=0, value=0),
@@ -98,7 +126,7 @@ def main():
         return
 
     if st.button("Generate Menu"):
-        lunch_menu, dinner_menu, total_cost = generate_menu(data, season, sub_category_pref, outings)
+        lunch_menu, dinner_menu, total_cost = generate_menu(data, season, sub_category_pref, outings, dietary_restrictions)
 
         st.header("Weekly Menu")
         st.subheader("Lunch Menu")
@@ -109,14 +137,27 @@ def main():
 
         st.subheader(f"Total Expenditure for the Week: PKR {total_cost}")
 
-               # Save menu to Excel
+        # Save menu to Excel
         with pd.ExcelWriter("weekly_menu_split.xlsx") as writer:
             lunch_menu.to_excel(writer, sheet_name="Lunch", index=False)
             dinner_menu.to_excel(writer, sheet_name="Dinner", index=False)
 
         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()
+    main()