Update app.py

app.py

@@ -106,9 +106,10 @@ if 'key0' in st.session_state:
 
     # Assign dataframe a name
     df_vul = st.session_state['key0']
-
+    st.write(df_vul)
+    
     col1, col2 = st.columns([1,1])
-
+    
     with col1:
         
         # Header
@@ -128,13 +129,27 @@ if 'key0' in st.session_state:
                 their respective labels in the table below.</div>""", unsafe_allow_html=True)
 
     with col2:
-        st.table(df_vul[df_vul['Vulnerability Label'] != 'Other'])
-    #     ### Pie chart
+        
+       
+        # ### Pie chart
                     
-    #     ## Create a df that stores all the labels
-    #     #df_labels = 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
+        # group_counts = {}
+
+        # # Iterate through each sublist
+        # for index, row in df_vul:
+            
+        #     # Iterate through each group in the sublist
+        #     for group in row['Vulnerability Label']:
+                
+        #         # Update the count in the dictionary
+        #         group_counts[group] = group_counts.get(group, 0) + 1
+
+        # st.write(group_counts)
 
-    #     # Count how often each label appears in the "Vulnerability Labels" column
     #     label_counts = df_vul['Vulnerability Label'].value_counts().reset_index()
     #     label_counts.columns = ['Label', 'Count']
 
@@ -154,7 +169,7 @@ if 'key0' in st.session_state:
     #     st.plotly_chart(fig, use_container_width=True)
 
     # ### Table 
-        #st.table(df_vul[df_vul['Vulnerability Label'] != 'Other'])
+    st.table(df_vul[df_vul['Vulnerability Label'] != 'Other'])
 
    # vulnerability_analysis.vulnerability_display()
 # elif topic == 'Action':