Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import pandas as pd
 from datetime import datetime
 import os
+import matplotlib.pyplot as plt
+import seaborn as sns
 from huggingface_hub import HfApi, upload_file, list_repo_files, hf_hub_download
 
 # Configuration for Hugging Face Repository
@@ -105,7 +107,7 @@ if "history" not in st.session_state:
     st.session_state.current_selections = load_current_selections().to_dict(orient="records")
 
 # Sidebar for navigating through different views
-menu = st.sidebar.selectbox("Select View", ["Poll", "Current", "History"])
+menu = st.sidebar.selectbox("Select View", ["Poll", "Current", "History", "Graph"])
 
 # Function to reset the current selections after submission
 def reset_selections():
@@ -233,3 +235,40 @@ elif menu == "History":
             st.table(record["Summary"])
     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.")