Update app.py

app.py

@@ -87,39 +87,17 @@ 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
-    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_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)
 
-    # Apply weekdays to the menu
-    full_menu_with_weekdays = assign_weekdays(menu_df)
+    lunch_menu = menu_df.iloc[:len(menu_df)//2]
+    dinner_menu = menu_df.iloc[len(menu_df)//2:]
 
-    lunch_menu = full_menu_with_weekdays[:len(full_menu_with_weekdays)//2]
-    dinner_menu = full_menu_with_weekdays[len(full_menu_with_weekdays)//2:]
-
-    # 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_df, dinner_menu_df, total_cost, shopping_list
+    lunch_menu = lunch_menu[["Name"]]
+    dinner_menu = dinner_menu[["Name"]]
+
+    return lunch_menu, dinner_menu, total_cost, shopping_list
 
 # Function to calculate BMI and recommend diet
 def bmi_insights(weight, height_cm):
@@ -182,12 +160,10 @@ def main():
 
         st.header("Weekly Menu")
         st.subheader("Lunch Menu")
-        for index, row in lunch_menu.iterrows():
-            st.write(f"{row['Weekday']}: {row['Name']}")
+        st.write(lunch_menu)
 
         st.subheader("Dinner Menu")
-        for index, row in dinner_menu.iterrows():
-            st.write(f"{row['Weekday']}: {row['Name']}")
+        st.write(dinner_menu)
 
         st.subheader(f"Total Expenditure for the Week: PKR {total_cost}")