leavoigt commited on
Commit
c7320de
1 Parent(s): c91ed20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -35,6 +35,7 @@ install_packages()
35
 
36
 
37
  import appStore.vulnerability_analysis as vulnerability_analysis
 
38
  import appStore.doc_processing as processing
39
  from utils.uploadAndExample import add_upload
40
  from utils.vulnerability_classifier import label_dict
@@ -126,7 +127,7 @@ with st.expander("ℹ️ - About this app", expanded=False):
126
  st.write("")
127
 
128
  # Define the apps used
129
- apps = [processing.app, vulnerability_analysis.app]
130
 
131
  multiplier_val =1/len(apps)
132
  if st.button("Analyze Document"):
@@ -140,7 +141,7 @@ if 'key0' in st.session_state:
140
  with st.sidebar:
141
  topic = st.radio(
142
  "Which category you want to explore?",
143
- (['Vulnerability']))
144
 
145
  if topic == 'Vulnerability':
146
 
@@ -167,6 +168,7 @@ if 'key0' in st.session_state:
167
  their respective labels in the table below.</div>""", unsafe_allow_html=True)
168
 
169
  with col2:
 
170
  ### Pie chart
171
 
172
  # Create a df that stores all the labels
 
35
 
36
 
37
  import appStore.vulnerability_analysis as vulnerability_analysis
38
+ import appStore.target as target_extraction
39
  import appStore.doc_processing as processing
40
  from utils.uploadAndExample import add_upload
41
  from utils.vulnerability_classifier import label_dict
 
127
  st.write("")
128
 
129
  # Define the apps used
130
+ apps = [processing.app, vulnerability_analysis.app, target_extraction.app]
131
 
132
  multiplier_val =1/len(apps)
133
  if st.button("Analyze Document"):
 
141
  with st.sidebar:
142
  topic = st.radio(
143
  "Which category you want to explore?",
144
+ (['Vulnerability', 'Concrete targets/actions']))
145
 
146
  if topic == 'Vulnerability':
147
 
 
168
  their respective labels in the table below.</div>""", unsafe_allow_html=True)
169
 
170
  with col2:
171
+
172
  ### Pie chart
173
 
174
  # Create a df that stores all the labels