Update app.py

app.py

@@ -129,32 +129,31 @@ if 'key0' in st.session_state:
                 their respective labels in the table below.</div>""", unsafe_allow_html=True)
 
     with col2:
-        st.write("")
-        # ### Pie chart
+        
+        ### Bar chart
                     
         # # Create a df that stores all the labels
-        # df_labels = pd.DataFrame(list(label_dict.items()), columns=['Label ID', 'Label'])
+        df_labels = pd.DataFrame(list(label_dict.items()), columns=['Label ID', 'Label'])
 
-        # # Count how often each label appears in the "Vulnerability Labels" column
-        # group_counts = {}
+        # Count how often each label appears in the "Vulnerability Labels" column
+        group_counts = {}
 
-        # # Iterate through each sublist
-        # for index, row in df_vul:
+        # Iterate through each sublist
+        for index, row in df_vul:
             
-        #     # Iterate through each group in the sublist
-        #     for group in row['Vulnerability Label']:
+            # Iterate through each group in the sublist
+            for sublist in row['Vulnerability Label']:
                 
-        #         # Update the count in the dictionary
-        #         group_counts[group] = group_counts.get(group, 0) + 1
-
-        # st.write(group_counts)
+                # Update the count in the dictionary
+                group_counts[sublist] = group_counts.get(sublist, 0) + 1
 
-    #     label_counts = df_vul['Vulnerability Label'].value_counts().reset_index()
-    #     label_counts.columns = ['Label', 'Count']
-
-    #     # Merge the label counts with the df_label DataFrame
-    #     df_labels = df_labels.merge(label_counts, on='Label', how='left')
+        # Create a new dataframe from group_counts
+        df_label_count = pd.DataFrame(list(group_counts.items()), columns=['Label', 'Count'])
 
+        # Merge the label counts with the df_label DataFrame
+        df_label_count = df_labels.merge(df_label_counts, on='Label', how='left')
+        st.write("df_label_count")
+        
     #     # Configure graph
     #     fig = px.pie(df_labels,
     #             names="Label",