Update app.py

app.py

@@ -2,7 +2,6 @@ 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):
@@ -11,9 +10,47 @@ def load_data(file_url):
     return pd.read_excel(file_path, engine='openpyxl')
 
 # 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):
+def generate_menu(data, season, outings, dietary_restrictions, bmi, pantry_ingredients):
     season_data = data[(data['Season'] == season) | (data['Season'] == 'All Seasons')]
 
+    # 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 = []
@@ -110,21 +147,6 @@ def main():
     # 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),
-        'Rice': st.number_input("Number of Rice dishes", min_value=0, value=0),
-        'Daal': st.number_input("Number of Daal dishes", min_value=0, value=0),
-        'Meat': st.number_input("Number of Meat dishes", min_value=0, value=0),
-        'Chicken': st.number_input("Number of Chicken dishes", min_value=0, value=0),
-        'Fish': st.number_input("Number of Fish dishes", min_value=0, value=0)
-    }
-
-    total_main_courses = sum(sub_category_pref.values()) + outings
-    if total_main_courses > 14:
-        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:
@@ -133,7 +155,8 @@ def main():
         pantry_ingredients = set()
 
     if st.button("Generate Menu"):
-        lunch_menu, dinner_menu, total_cost, shopping_list = generate_menu(data, season, sub_category_pref, outings, dietary_restrictions, pantry_ingredients)
+        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")