Update app.py

app.py

@@ -236,6 +236,43 @@ elif menu == "History":
     else:
         st.write("No history records found.")
 
+# # Graph view to display a line chart of item selections over time
+# elif menu == "Graph":
+#     st.title("Breakfast Poll History - Graph View")
+
+#     # Prepare data for plotting
+#     if st.session_state.history:
+#         history_data = []
+#         for record in st.session_state.history:
+#             date = record['Date']
+#             for index, row in record['Summary'].iterrows():
+#                 for drink in row['Drinks'].split(', '):
+#                     history_data.append({'Date': date, 'Item': drink, 'Type': 'Drink'})
+#                 for food in row['Food'].split(', '):
+#                     history_data.append({'Date': date, 'Item': food, 'Type': 'Food'})
+
+#         # Create a DataFrame from history data
+#         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')
+
+#         # Create a line plot for each item over time
+#         plt.figure(figsize=(12, 6))
+#         sns.lineplot(data=item_counts, x='Date', y='Count', hue='Item', marker='o')
+
+#         # Customize the plot
+#         plt.xticks(rotation=45)
+#         plt.title('Item Selections Over Time')
+#         plt.xlabel('Date')
+#         plt.ylabel('Number of Selections')
+#         plt.legend(loc='upper center', bbox_to_anchor=(0.5, -0.15), ncol=3)
+        
+#         # Display the plot
+#         st.pyplot(plt.gcf())
+#     else:
+#         st.write("No historical data available to plot.")
+
 # Graph view to display a line chart of item selections over time
 elif menu == "Graph":
     st.title("Breakfast Poll History - Graph View")
@@ -244,7 +281,8 @@ elif menu == "Graph":
     if st.session_state.history:
         history_data = []
         for record in st.session_state.history:
-            date = record['Date']
+            # Extract only the date part (YYYY-MM-DD) for display
+            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'})
@@ -257,9 +295,24 @@ elif menu == "Graph":
         # 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)
@@ -271,4 +324,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.")