Update app.py

app.py

@@ -373,22 +373,26 @@ elif menu == "Graph":
         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')
-        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())
+        # Check if there is data to display
+        if not filtered_item_counts.empty:
+            # 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 y-axis to show only integer labels
+            y_max = max(filtered_item_counts['Count'].max() + 1, 1)  # Set y_max to at least 1 to avoid errors
+            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')
+            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 data to display. Please select at least one user and one item.")
     else:
         st.write("No historical data available to plot.")