Update app.py

app.py

@@ -10,23 +10,23 @@ 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, bmi_category):
+def generate_menu(data, season, outings, bmi_category):
     # Filter season data
     season_data = data[(data['Season'] == season) | (data['Season'] == 'All Seasons')]
     
     # Adjust the menu based on BMI category
     if bmi_category == "Underweight":
         # Suggest high-calorie dishes like meat, rice, and calorie-dense vegetables
-        preferred_dishes = ['Meat', 'Rice', 'Sabzi', 'Chicken']
+        preferred_dishes = ['Meat', 'Rice', 'Sabzi', 'Chicken', 'Fish']
     elif bmi_category == "Normal Weight":
         # Balanced meals with a variety of dishes
-        preferred_dishes = ['Sabzi', 'Rice', 'Daal', 'Chicken', 'Fish']
+        preferred_dishes = ['Sabzi', 'Rice', 'Daal', 'Chicken', 'Fish', 'Meat']
     elif bmi_category == "Overweight":
         # Focus on lean proteins and vegetables, limiting carbs
-        preferred_dishes = ['Sabzi', 'Chicken', 'Fish', 'Meat']
+        preferred_dishes = ['Sabzi', 'Chicken', 'Fish', 'Lean Meat', 'Daal']
     else:  # Obese category
         # Focus on lean proteins, vegetables, and smaller portions of carbs
-        preferred_dishes = ['Sabzi', 'Daal', 'Chicken', 'Fish']
+        preferred_dishes = ['Sabzi', 'Daal', 'Chicken', 'Fish', 'Vegetables']
 
     # Select dishes based on the BMI category and preferences
     menu = []
@@ -43,13 +43,17 @@ def generate_menu(data, season, sub_category_pref, outings, bmi_category):
     side_dishes = data[data['Course'] == 'Side Dish'].sample(2).to_dict('records')
     menu.extend(side_dishes)
 
-    # Generate the menu and calculate total cost
+    # Create DataFrame for the menu
     menu_df = pd.DataFrame(menu)
+
+    # Split the menu into lunch and dinner
     lunch_menu = menu_df.iloc[:len(menu_df)//2]
     dinner_menu = menu_df.iloc[len(menu_df)//2:]
 
+    # Calculate the total cost
     total_cost = menu_df["Cost per 4 persons"].sum()
 
+    # Only include the names of the dishes for lunch and dinner menus
     lunch_menu = lunch_menu[["Name"]]
     dinner_menu = dinner_menu[["Name"]]
 
@@ -57,25 +61,23 @@ def generate_menu(data, season, sub_category_pref, outings, bmi_category):
 
 # Function to calculate BMI and recommend diet for adults
 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.", "Underweight"
+        return f"Your BMI is {bmi:.2f}. You are underweight. A diet rich in proteins and healthy fats would be beneficial.", "Underweight"
     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.", "Normal Weight"
+        return f"Your BMI is {bmi:.2f}. You are in the normal weight range. Maintain a balanced diet.", "Normal Weight"
     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.", "Overweight"
+        return f"Your BMI is {bmi:.2f}. You are overweight. A diet with fewer carbs and more lean proteins can help manage your weight.", "Overweight"
     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.", "Obese"
+        return f"Your BMI is {bmi:.2f}. You are obese. A calorie-restricted diet with a focus on whole foods and lean proteins is recommended.", "Obese"
 
 # Function to provide general diet recommendation for infants
 def infant_diet_recommendation():
     return (
         "For infants (up to 12 months), breast milk or formula should be the primary source of nutrition. "
         "Around 6 months, you can start introducing solid foods, including pureed fruits, vegetables, and cereals. "
-        "Ensure a variety of textures as they grow and introduce protein-rich foods, like pureed meat or lentils. "
         "Consult with your pediatrician for more personalized advice."
     )
 
@@ -108,6 +110,17 @@ def main():
             bmi_message, bmi_category = bmi_insights(weight, height_cm)
             st.sidebar.success(bmi_message)
 
+        # Dropdown for selecting medical condition(s)
+        medical_condition = st.sidebar.selectbox(
+            "Select any medical condition", [
+                "None", "Diabetes", "Celiac Disease", "Hypertension", "Heart Disease", "Obesity", 
+                "Cholesterol Disorders", "IBS", "GERD", "Peptic Ulcers", "Gout", "Lactose Intolerance", 
+                "Kidney Disease", "Hypothyroidism", "Anemia", "Food Allergies", "Acid Reflux", 
+                "Crohn's Disease", "Osteoporosis", "Autoimmune Diseases", "Migraines", "PCOS", 
+                "CFS", "Mental Health"
+            ]
+        )
+
     # Google Drive file URL
     file_url = 'https://drive.google.com/uc?id=1BJFao3C6p8k3_KjLfmDo5KbNMrTs7MRo'  # Replace with actual file link
     data = load_data(file_url)
@@ -118,23 +131,8 @@ def main():
 
         outings = st.number_input("Enter the number of outings this week", min_value=0, max_value=14, value=0)
 
-        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
-
         if st.button("Generate Menu"):
-            lunch_menu, dinner_menu, total_cost = generate_menu(data, season, sub_category_pref, outings, bmi_category)
+            lunch_menu, dinner_menu, total_cost = generate_menu(data, season, outings, bmi_category)
 
             st.header("Weekly Menu")
             st.subheader("Lunch Menu")