Update app.py

app.py

@@ -9,25 +9,75 @@ 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, dietary_restriction, outings):
-    # Filter data by season and dietary restrictions
+# Function to generate the weekly menu based on BMI, dietary restriction, and pantry sync
+def generate_menu(data, season, outings, dietary_restrictions, bmi, pantry_ingredients):
     season_data = data[(data['Season'] == season) | (data['Season'] == 'All Seasons')]
 
-    # Apply dietary restrictions (e.g., remove gluten or low-sodium)
-    if dietary_restriction == "Gluten-Free":
-        season_data = season_data[season_data['Gluten-Free'] == True]
-    elif dietary_restriction == "Low-Sodium":
-        season_data = season_data[season_data['Sodium'] < 100]
+    # Define the number of dishes based on BMI
+    if bmi < 18.5:  # Underweight
+        sub_category_pref = {
+            'Sabzi': 3,  # More vegetables to increase calorie intake
+            'Rice': 2,   # More rice to increase carbohydrates
+            'Daal': 2,   # Protein-rich dals
+            'Meat': 3,   # More meat for proteins
+            'Chicken': 2,
+            'Fish': 1
+        }
+    elif 18.5 <= bmi < 24.9:  # Normal weight
+        sub_category_pref = {
+            'Sabzi': 2,
+            'Rice': 2,
+            'Daal': 2,
+            'Meat': 2,
+            'Chicken': 2,
+            'Fish': 1
+        }
+    elif 25 <= bmi < 29.9:  # Overweight
+        sub_category_pref = {
+            'Sabzi': 4,  # More vegetables
+            'Rice': 1,   # Limit rice
+            'Daal': 2,   # Protein-rich dals
+            'Meat': 1,   # Limit meat intake
+            'Chicken': 3,
+            'Fish': 1
+        }
+    else:  # Obese
+        sub_category_pref = {
+            'Sabzi': 5,  # More vegetables
+            'Rice': 1,   # Very limited rice
+            'Daal': 2,   # Protein-rich dals
+            'Meat': 1,   # Very limited meat
+            'Chicken': 2,
+            'Fish': 1
+        }
 
     meals_needed = 14 - outings  # Main Courses (14) minus outings
 
     menu = []
-    sub_categories = ['Sabzi', 'Rice', 'Daal', 'Meat', 'Chicken', 'Fish']
+    shopping_list = set()  # Track missing ingredients
 
-    for sub_category in sub_categories:
+    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(1, 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]
 
@@ -66,7 +116,7 @@ def generate_menu(data, season, dietary_restriction, outings):
     lunch_menu = lunch_menu[["Name", "Weekday"]]
     dinner_menu = dinner_menu[["Name", "Weekday"]]
 
-    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):
@@ -74,18 +124,22 @@ def bmi_insights(weight, height_cm):
     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):
+    return set(available_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
@@ -106,13 +160,22 @@ def main():
 
     # Main menu generation section
     season = st.selectbox("Select the current season", ["Summer", "Winter"])
-    dietary_restriction = st.selectbox("Select your dietary restriction", ["None", "Gluten-Free", "Low-Sodium"])
 
     outings = st.number_input("Enter the number of outings this week", min_value=0, max_value=14, value=0)
 
-    # Generate Menu when button is clicked
+    # Dietary restrictions
+    dietary_restrictions = st.selectbox("Select any dietary restrictions", ["None", "Diabetic", "Gluten-Free", "Low-Sodium"])
+
+    # 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, dietary_restriction, outings)
+        bmi = weight / (height_cm / 100) ** 2  # BMI calculation
+        lunch_menu, dinner_menu, total_cost, shopping_list = generate_menu(data, season, outings, dietary_restrictions, bmi, pantry_ingredients)
 
         st.header("Weekly Menu")
         st.subheader("Lunch Menu")
@@ -123,6 +186,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)