Update app.py

app.py

@@ -236,6 +236,76 @@ 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")
+
+#     # 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']
+#             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', 'Type']).size().reset_index(name='Count')
+
+#         # 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")
+
+#         # Drinks Section
+#         if drinks:
+#             st.sidebar.subheader("Drinks")
+#             for item in drinks:
+#                 # Add a unique key to each checkbox to avoid duplicate widget IDs
+#                 item_visibility[item] = st.sidebar.checkbox(item, value=True, key=f"checkbox_{item}_Drink")
+
+#         # Food Section
+#         if foods:
+#             st.sidebar.subheader("Food")
+#             for item in foods:
+#                 # Add a unique key to each checkbox to avoid duplicate widget IDs
+#                 item_visibility[item] = st.sidebar.checkbox(item, value=True, key=f"checkbox_{item}_Food")
+
+#         # 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=filtered_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")
@@ -247,20 +317,27 @@ elif menu == "Graph":
     # Prepare data for plotting
     if st.session_state.history:
         history_data = []
+        user_data = {}  # Store user-specific 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():
+                user = row['Name']
                 for drink in row['Drinks'].split(', '):
-                    history_data.append({'Date': date, 'Item': drink, 'Type': 'Drink'})
+                    history_data.append({'Date': date, 'Item': drink, 'Type': 'Drink', 'User': user})
                 for food in row['Food'].split(', '):
-                    history_data.append({'Date': date, 'Item': food, 'Type': 'Food'})
+                    history_data.append({'Date': date, 'Item': food, 'Type': 'Food', 'User': user})
+                
+                # Append user data for selection
+                if user not in user_data:
+                    user_data[user] = True  # Initialize all users as visible by default
 
         # 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', 'Type']).size().reset_index(name='Count')
+        item_counts = history_df.groupby(['Date', 'Item', 'Type', 'User']).size().reset_index(name='Count')
 
         # Separate items into Drinks and Food, and sort them alphabetically
         drinks = sorted(item_counts[item_counts['Type'] == 'Drink']['Item'].unique())
@@ -269,7 +346,7 @@ elif menu == "Graph":
         # Create a dictionary to store the checkbox values for each item
         item_visibility = {}
 
-        # Create interactive checkboxes for Drinks and Food in the sidebar
+        # Create interactive checkboxes for Drinks, Food, and Users in the sidebar
         st.sidebar.header("Select Items to Display")
 
         # Drinks Section
@@ -286,14 +363,24 @@ elif menu == "Graph":
                 # Add a unique key to each checkbox to avoid duplicate widget IDs
                 item_visibility[item] = st.sidebar.checkbox(item, value=True, key=f"checkbox_{item}_Food")
 
-        # Filter the data based on selected items
+        # User Section: Create a checkbox for each user to toggle their visibility
+        st.sidebar.subheader("Users")
+        for user in user_data.keys():
+            user_data[user] = st.sidebar.checkbox(user, value=True, key=f"checkbox_user_{user}")
+
+        # Filter the data based on selected items and users
         selected_items = [item for item, visible in item_visibility.items() if visible]
-        filtered_item_counts = item_counts[item_counts['Item'].isin(selected_items)]
+        selected_users = [user for user, visible in user_data.items() if visible]
+        filtered_item_counts = item_counts[item_counts['Item'].isin(selected_items) & item_counts['User'].isin(selected_users)]
 
         # Create a line plot for each selected item over time
         plt.figure(figsize=(12, 6))
         sns.lineplot(data=filtered_item_counts, x='Date', y='Count', hue='Item', marker='o')
 
+        # Customize the plot to show integer y-axis labels
+        y_max = filtered_item_counts['Count'].max() + 1
+        plt.yticks(range(0, y_max))  # Show only integer labels on the y-axis
+
         # Customize the plot
         plt.xticks(rotation=45)
         plt.title('Item Selections Over Time')