Update app.py

app.py

@@ -129,31 +129,31 @@ if 'key0' in st.session_state:
 
     with col2:
         
-        ### Pie chart
+    #     ### Pie chart
                     
-        # 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 = 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')
-
-        # Configure graph
-        fig = px.pie(df_labels,
-                names="Label", 
-                values="Count",
-                title='Label Counts',
-                hover_name="Count",
-                color_discrete_sequence=px.colors.qualitative.Plotly
-        )
+    #     ## 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 = 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')
+
+    #     # Configure graph
+    #     fig = px.pie(df_labels,
+    #             names="Label", 
+    #             values="Count",
+    #             title='Label Counts',
+    #             hover_name="Count",
+    #             color_discrete_sequence=px.colors.qualitative.Plotly
+    #     )
         
-        #Show plot
-        st.plotly_chart(fig, use_container_width=True)
+    #     #Show plot
+    #     st.plotly_chart(fig, use_container_width=True)
 
-    ### Table 
+    # ### Table 
     st.table(df_vul[df_vul['Vulnerability Label'] != 'Other'])
 
    # vulnerability_analysis.vulnerability_display()