Update app.py

app.py

@@ -103,32 +103,42 @@ if 'key0' in st.session_state:
     
     if topic == 'Vulnerability':
 
-        ### 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']
+        col1, col2 = st.columns([1,1])
 
-        # 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
-        )
+        with col1:
+            
+            ### Table 
+            st.table(st.session_state['key0'])
         
-        #Show plot
-        st.plotly_chart(fig, use_container_width=True)
+        with col2: 
+            
+            ### 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)
         
         ### Table 
-        st.table(st.session_state['key0'])
+        #st.table(st.session_state['key0'])
 
        # vulnerability_analysis.vulnerability_display()
     # elif topic == 'Action':