Update app.py

app.py

@@ -244,6 +244,7 @@ elif menu == "History":
 #     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']
 #             for index, row in record['Summary'].iterrows():
 #                 for drink in row['Drinks'].split(', '):
@@ -257,9 +258,24 @@ elif menu == "History":
 #         # Count occurrences of each item per date
 #         item_counts = history_df.groupby(['Date', 'Item']).size().reset_index(name='Count')
 
-#         # Create a line plot for each item over time
+#         # Get a list of unique items (Drinks and Food)
+#         unique_items = item_counts['Item'].unique()
+
+#         # Create a dictionary to store the checkbox values for each item
+#         item_visibility = {}
+        
+#         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)
+
+#         # Filter the data based on selected items
+#         selected_items = [item for item, visible in item_visibility.items() if visible]
+#         filtered_item_counts = item_counts[item_counts['Item'].isin(selected_items)]
+
+#         # Create a line plot for each selected item over time
 #         plt.figure(figsize=(12, 6))
-#         sns.lineplot(data=item_counts, x='Date', y='Count', hue='Item', marker='o')
+#         sns.lineplot(data=filtered_item_counts, x='Date', y='Count', hue='Item', marker='o')
 
 #         # Customize the plot
 #         plt.xticks(rotation=45)
@@ -277,12 +293,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'})
@@ -324,4 +344,4 @@ elif menu == "Graph":
         # Display the plot
         st.pyplot(plt.gcf())
     else:
-        st.write("No historical data available to plot.")
+        st.write("No historical data available to plot.")