Update app.py

app.py

@@ -2,6 +2,7 @@ import appStore.vulnerability_analysis as vulnerability_analysis
 import appStore.doc_processing as processing
 from utils.uploadAndExample import add_upload
 import streamlit as st
+from utils.vulnerability_classifier import label_dict
 
 st.set_page_config(page_title = 'Vulnerability Analysis', 
                    initial_sidebar_state='expanded', layout="wide") 
@@ -105,11 +106,23 @@ if 'key0' in st.session_state:
 
         # Pie chart
         with col1:
+            
             print(type(st.session_state['key0']))
+
+            # Create a df that stores how often the labels appear
+            df_count = 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 Labels'].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')
+            st.write(df_label)
+            #bar_data = st.session_state['key0']
             
-            bar_data = st.session_state['key0']
             
-            # fig = px.bar(chart3_data,
+            # fig = px.bar(st.session_state['key0'],
             #         x="Year", 
             #         y="Value",   
             #         color='Country',