Update app.py
Browse files
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
|