Update app.py

app.py

@@ -87,36 +87,39 @@ 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)
 
-    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 = []
+
+        # Create menu for the week with each dish assigned to a weekday
         for i, dish in enumerate(menu):
             if isinstance(dish, dict):
-                dish_with_day = dish.copy()  # Copy the dish dictionary
+                dish_name = dish["Name"]
             else:
-                dish_with_day = {"Name": dish}  # If it's just a string, create a dict with "Name" key
+                dish_name = dish  # If it's just a string, take the name directly
 
-            dish_with_day["Weekday"] = weekdays[i % 7]  # Assign the weekday
+            dish_with_day = {"Name": dish_name, "Weekday": weekdays[i % 7]}
             menu_with_weekdays.append(dish_with_day)
 
-        return pd.DataFrame(menu_with_weekdays)
+        return menu_with_weekdays
 
-    lunch_menu = menu_df.iloc[:len(menu_df)//2]
-    dinner_menu = menu_df.iloc[len(menu_df)//2:]
+    menu_df = pd.DataFrame(menu)
 
-    total_cost = menu_df["Cost per 4 persons"].sum()
+    # Apply weekdays to the menu
+    full_menu_with_weekdays = assign_weekdays(menu_df)
 
-    lunch_menu = assign_weekdays(lunch_menu)
-    dinner_menu = assign_weekdays(dinner_menu)
+    lunch_menu = full_menu_with_weekdays[:len(full_menu_with_weekdays)//2]
+    dinner_menu = full_menu_with_weekdays[len(full_menu_with_weekdays)//2:]
 
-    lunch_menu = lunch_menu[["Name", "Weekday"]]
-    dinner_menu = dinner_menu[["Name", "Weekday"]]
+    # Convert to DataFrame for display
+    lunch_menu_df = pd.DataFrame(lunch_menu)
+    dinner_menu_df = pd.DataFrame(dinner_menu)
+
+    # Calculate total cost (assuming it's available in the data)
+    total_cost = menu_df["Cost per 4 persons"].sum()
 
-    return lunch_menu, dinner_menu, total_cost, shopping_list
+    return lunch_menu_df, dinner_menu_df, total_cost, shopping_list
 
 # Function to calculate BMI and recommend diet
 def bmi_insights(weight, height_cm):
@@ -179,10 +182,12 @@ def main():
 
         st.header("Weekly Menu")
         st.subheader("Lunch Menu")
-        st.write(lunch_menu)
+        for index, row in lunch_menu.iterrows():
+            st.write(f"{row['Weekday']}: {row['Name']}")
 
         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}")