Update app.py

app.py

@@ -87,17 +87,42 @@ def generate_menu(data, season, outings, dietary_restrictions, bmi, pantry_ingre
     side_dishes = data[data['Course'] == 'Side Dish'].sample(2).to_dict('records')
     menu.extend(side_dishes)
 
+    # Function to assign weekdays to dishes (for Breakfast, Lunch, and Dinner)
+    def assign_weekdays(menu):
+        weekdays = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
+        menu_with_weekdays = []
+
+        # Assigning the dishes to the weekdays for Breakfast, Lunch, and Dinner
+        for i, dish in enumerate(menu):
+            if isinstance(dish, dict):
+                dish_name = dish["Name"]
+            else:
+                dish_name = dish  # If it's just a string, take the name directly
+
+            dish_with_day = {"Name": dish_name, "Weekday": weekdays[i % 7]}
+            menu_with_weekdays.append(dish_with_day)
+
+        return menu_with_weekdays
+
     menu_df = pd.DataFrame(menu)
 
-    lunch_menu = menu_df.iloc[:len(menu_df)//2]
-    dinner_menu = menu_df.iloc[len(menu_df)//2:]
+    # Apply weekdays to the menu
+    full_menu_with_weekdays = assign_weekdays(menu_df)
 
-    total_cost = menu_df["Cost per 4 persons"].sum()
+    # Split menu for breakfast, lunch, and dinner (assuming 7 for each)
+    breakfast_menu = full_menu_with_weekdays[:7]
+    lunch_menu = full_menu_with_weekdays[7:14]
+    dinner_menu = full_menu_with_weekdays[14:21]
 
-    lunch_menu = lunch_menu[["Name"]]
-    dinner_menu = dinner_menu[["Name"]]
+    # Convert to DataFrame for display
+    breakfast_menu_df = pd.DataFrame(breakfast_menu)
+    lunch_menu_df = pd.DataFrame(lunch_menu)
+    dinner_menu_df = pd.DataFrame(dinner_menu)
 
-    return lunch_menu, dinner_menu, total_cost, shopping_list
+    # Calculate total cost (assuming it's available in the data)
+    total_cost = menu_df["Cost per 4 persons"].sum()
+
+    return breakfast_menu_df, lunch_menu_df, dinner_menu_df, total_cost, shopping_list
 
 # Function to calculate BMI and recommend diet
 def bmi_insights(weight, height_cm):
@@ -156,14 +181,24 @@ def main():
 
     if st.button("Generate Menu"):
         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)
+        breakfast_menu, lunch_menu, dinner_menu, total_cost, shopping_list = generate_menu(data, season, outings, dietary_restrictions, bmi, pantry_ingredients)
 
         st.header("Weekly Menu")
+        
+        # Display Breakfast Menu
+        st.subheader("Breakfast Menu")
+        for index, row in breakfast_menu.iterrows():
+            st.write(f"{row['Weekday']}: {row['Name']}")
+
+        # Display Lunch Menu
         st.subheader("Lunch Menu")
-        st.write(lunch_menu)
+        for index, row in lunch_menu.iterrows():
+            st.write(f"{row['Weekday']}: {row['Name']}")
 
+        # Display Dinner Menu
         st.subheader("Dinner Menu")
-        st.write(dinner_menu)
+        for index, row in dinner_menu.iterrows():
+            st.write(f"{row['Weekday']}: {row['Name']}")
 
         st.subheader(f"Total Expenditure for the Week: PKR {total_cost}")
 
@@ -174,6 +209,7 @@ def main():
 
         # Save menu to Excel
         with pd.ExcelWriter("weekly_menu_split.xlsx") as writer:
+            breakfast_menu.to_excel(writer, sheet_name="Breakfast", index=False)
             lunch_menu.to_excel(writer, sheet_name="Lunch", index=False)
             dinner_menu.to_excel(writer, sheet_name="Dinner", index=False)