Update app.py

app.py

@@ -108,15 +108,15 @@ if 'key0' in st.session_state:
         # Pie chart
         with col1:
             
-            # Create a df that stores how often the labels appear
-            df_count = pd.DataFrame(list(label_dict.items()), columns=['Label ID', 'Label'])
+            # Create a df that stores all the labels
+            df_labels = pd.DataFrame(list(label_dict.items()), columns=['Label ID', 'Label'])
 
             # Count how often each label appears in the "Vulnerability Labels" column
             label_counts = st.session_state['key0']['Vulnerability Label'].value_counts().reset_index()
             label_counts.columns = ['Label', 'Count']
 
             # Merge the label counts with the df_label DataFrame
-            df_label = df_label.merge(label_counts, on='Label', how='left')
+            df_labels = df_labels.merge(label_counts, on='Label', how='left')
         
             
             fig = px.bar(df_label,