Update app.py

app.py

@@ -103,54 +103,30 @@ if 'key0' in st.session_state:
     
     if topic == 'Vulnerability':
 
-        # Display charts
-        col1, col2 = st.columns([1,1])
-
-        # Pie chart
-        with col1:
-            
-            # 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 = 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_labels = df_labels.merge(label_counts, on='Label', how='left')
-
-            # Configure graph
-            fig = px.bar(df_labels,
-                    x="Label", 
-                    y="Count",   
-            #         color='Country',
-                    title='Label Counts',
-                    hover_name="Count",
-            #        color_discrete_sequence=px.colors.qualitative.Plotly
-                    )
-            
-            #Show plot
-            st.plotly_chart(fig, use_container_width=True)
-
-
-        # Bar cart
-        with col2: 
-            
-            # Configure graph
-            fig = px.pie(df_labels,
-                    names="Label", 
-                    values="Count",   
-            #         color='Country',
-                    title='Label Counts',
-                    hover_name="Count",
-            #        color_discrete_sequence=px.colors.qualitative.Plotly
-                    )
-            
-            #Show plot
-            st.plotly_chart(fig, use_container_width=True)
-           
+        ### 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 = 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_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)
         
-        # Display the table 
+        ### Table 
         st.table(st.session_state['key0'])
 
        # vulnerability_analysis.vulnerability_display()