Update app.py

app.py

@@ -10,24 +10,15 @@ def load_data(file_url):
     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
+def generate_menu(data, season, sub_category_pref, outings):
     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]
-
     meals_needed = 14 - outings  # Main Courses (14) minus outings
 
     menu = []
-    sub_categories = ['Sabzi', 'Rice', 'Daal', 'Meat', 'Chicken', 'Fish']
-
-    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'))
+        menu.extend(sub_category_dishes.sample(min(freq, len(sub_category_dishes))).to_dict('records'))
 
     menu = menu[:meals_needed]
 
@@ -39,37 +30,19 @@ def generate_menu(data, season, dietary_restriction, outings):
 
     menu_df = pd.DataFrame(menu)
 
-    # Function to assign weekdays to dishes
-    def assign_weekdays(menu):
-        weekdays = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
-
-        menu_with_weekdays = []
-        for i, dish in enumerate(menu):
-            if isinstance(dish, dict):
-                dish_with_day = dish.copy()  # Copy the dish dictionary
-            else:
-                dish_with_day = {"Name": dish}  # If it's just a string, create a dict with "Name" key
-
-            dish_with_day["Weekday"] = weekdays[i % 7]  # Assign the weekday
-            menu_with_weekdays.append(dish_with_day)
-
-        return pd.DataFrame(menu_with_weekdays)
-
     lunch_menu = menu_df.iloc[:len(menu_df)//2]
     dinner_menu = menu_df.iloc[len(menu_df)//2:]
 
     total_cost = menu_df["Cost per 4 persons"].sum()
 
-    lunch_menu = assign_weekdays(lunch_menu)
-    dinner_menu = assign_weekdays(dinner_menu)
-
-    lunch_menu = lunch_menu[["Name", "Weekday"]]
-    dinner_menu = dinner_menu[["Name", "Weekday"]]
+    lunch_menu = lunch_menu[["Name"]]
+    dinner_menu = dinner_menu[["Name"]]
 
     return lunch_menu, dinner_menu, total_cost
 
-# Function to calculate BMI and recommend diet
+# 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)
 
@@ -82,6 +55,15 @@ def bmi_insights(weight, height_cm):
     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."
 
+# 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."
+    )
+
 # Main workflow
 def main():
     st.title("Roz Roz Ka Masla..... Aaj Kya Pakayen!!! ")
@@ -91,44 +73,69 @@ def main():
     image = Image.open("food_image.jpg")  # Ensure you have an image in the same folder
     st.sidebar.image(image, use_container_width=True)
 
-    # User input for BMI insights
-    st.sidebar.header("Enter your details for BMI insights:")
-    weight = st.sidebar.number_input("Enter your weight (kg)", min_value=30, max_value=200, value=70)
-    height_cm = st.sidebar.number_input("Enter your height (cm)", min_value=90, max_value=250, value=175)
+    # User input for BMI insights or Infant's weekly diet
+    st.sidebar.header("Select your category:")
+    category = st.sidebar.radio("Are you an Adult or Infant?", ["Adult", "Infant"])
+
+    if category == "Infant":
+        st.sidebar.header("Infant Weekly Diet Recommendations")
+        diet_recommendation = infant_diet_recommendation()
+        st.sidebar.write(diet_recommendation)
+        return
 
-    if st.sidebar.button("Get BMI Insights"):
-        bmi_message = bmi_insights(weight, height_cm)
-        st.sidebar.success(bmi_message)
+    if category == "Adult":
+        # User input for BMI insights (only for Adults)
+        st.sidebar.header("Enter your details for BMI insights:")
+        weight = st.sidebar.number_input("Enter your weight (kg)", min_value=30, max_value=200, value=70)
+        height_cm = st.sidebar.number_input("Enter your height (cm)", min_value=90, max_value=250, value=175)
+
+        if st.sidebar.button("Get BMI Insights"):
+            bmi_message = bmi_insights(weight, height_cm)
+            st.sidebar.success(bmi_message)
 
     # Google Drive file URL
     file_url = 'https://drive.google.com/uc?id=1BJFao3C6p8k3_KjLfmDo5KbNMrTs7MRo'  # Replace with actual file link
     data = load_data(file_url)
 
-    # 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"])
+    # Main menu generation section (for Adults)
+    if category == "Adult":
+        season = st.selectbox("Select the current season", ["Summer", "Winter"])
+
+        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)
+        }
 
-    outings = st.number_input("Enter the number of outings this week", min_value=0, max_value=14, 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
 
-    # Generate Menu when button is clicked
-    if st.button("Generate Menu"):
-        lunch_menu, dinner_menu, total_cost = generate_menu(data, season, dietary_restriction, outings)
+        if st.button("Generate Menu"):
+            lunch_menu, dinner_menu, total_cost = generate_menu(data, season, sub_category_pref, outings)
 
-        st.header("Weekly Menu")
-        st.subheader("Lunch Menu")
-        st.write(lunch_menu)
+            st.header("Weekly Menu")
+            st.subheader("Lunch Menu")
+            st.write(lunch_menu)
 
-        st.subheader("Dinner Menu")
-        st.write(dinner_menu)
+            st.subheader("Dinner Menu")
+            st.write(dinner_menu)
 
-        st.subheader(f"Total Expenditure for the Week: PKR {total_cost}")
+            st.subheader(f"Total Expenditure for the Week: PKR {total_cost}")
 
-        # 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)
+            # 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'")
+            st.success("Weekly menu saved as 'weekly_menu_split.xlsx'")
 
 if __name__ == "__main__":
     main()