Update app.py

app.py

@@ -108,29 +108,25 @@ if 'key0' in st.session_state:
         # Pie chart
         with col1:
             
-            st.write(type(st.session_state['key0']))
-            st.write(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']
+            # 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')
-            # st.write(df_label)
-            #bar_data = st.session_state['key0']
-            
+            # Merge the label counts with the df_label DataFrame
+            df_label = df_label.merge(label_counts, on='Label', how='left')
+        
             
-            # fig = px.bar(st.session_state['key0'],
-            #         x="Year", 
-            #         y="Value",   
+            fig = px.bar(df_label,
+                    x="Label", 
+                    y="Count",   
             #         color='Country',
-            #         title='Chart 3 - Total Population',
-            #         hover_name="Value",
-            #         color_discrete_sequence=px.colors.qualitative.Plotly
-            #         )
+                    title='Label Count',
+                    hover_name="Value",
+                    color_discrete_sequence=px.colors.qualitative.Plotly
+                    )