Update app.py

app.py

@@ -240,12 +240,16 @@ elif menu == "History":
 # elif menu == "Graph":
 #     st.title("Breakfast Poll History - Graph View")
 
+#     # Load the history if not already loaded
+#     if not st.session_state.history:
+#         st.session_state.history = load_history()
+
 #     # Prepare data for plotting
 #     if st.session_state.history:
 #         history_data = []
 #         for record in st.session_state.history:
 #             # Extract only the date part (YYYY-MM-DD) for display
-#             date = record['Date']
+#             date = record['Date'].split("_")[0]  # Use only the YYYY-MM-DD portion of the date
 #             for index, row in record['Summary'].iterrows():
 #                 for drink in row['Drinks'].split(', '):
 #                     history_data.append({'Date': date, 'Item': drink, 'Type': 'Drink'})
@@ -313,18 +317,29 @@ elif menu == "Graph":
         history_df = pd.DataFrame(history_data)
         
         # Count occurrences of each item per date
-        item_counts = history_df.groupby(['Date', 'Item']).size().reset_index(name='Count')
+        item_counts = history_df.groupby(['Date', 'Item', 'Type']).size().reset_index(name='Count')
 
-        # Get a list of unique items (Drinks and Food)
-        unique_items = item_counts['Item'].unique()
+        # Separate items into Drinks and Food, and sort them alphabetically
+        drinks = sorted(item_counts[item_counts['Type'] == 'Drink']['Item'].unique())
+        foods = sorted(item_counts[item_counts['Type'] == 'Food']['Item'].unique())
 
         # Create a dictionary to store the checkbox values for each item
         item_visibility = {}
-        
+
+        # Create interactive checkboxes for Drinks and Food in the sidebar
         st.sidebar.header("Select Items to Display")
-        # Create a checkbox for each unique item in the sidebar
-        for item in unique_items:
-            item_visibility[item] = st.sidebar.checkbox(item, value=True)
+
+        # Drinks Section
+        if drinks:
+            st.sidebar.subheader("Drinks")
+            for item in drinks:
+                item_visibility[item] = st.sidebar.checkbox(item, value=True)
+
+        # Food Section
+        if foods:
+            st.sidebar.subheader("Food")
+            for item in foods:
+                item_visibility[item] = st.sidebar.checkbox(item, value=True)
 
         # Filter the data based on selected items
         selected_items = [item for item, visible in item_visibility.items() if visible]