Update app.py

app.py

@@ -77,7 +77,7 @@ examples = [
 def get_docs(input_query, country = [], vulnerability_cat = []):
   if not country:
     country = "All Countries"
-  if len(vulnerability_cat) == 0:
+  if not vulnerability_cat:
     if country == "All Countries":
       filters = None
     else:
@@ -141,7 +141,7 @@ def get_prompt(docs, input_query):
 
 def run_query(input_text, country, model_sel):
     # docs = get_docs(input_text, country)
-    docs = get_docs(input_text, country=country,vulnerability_cat=vulnerability_options)
+    docs = get_docs(input_text, country=country,vulnerability_cat=vulnerabilities_cat)
     # st.write('Selected country: ', country) # Debugging country
     if model_sel == "chatGPT":
         # res = pipe.run(query=input_text, documents=docs)
@@ -164,7 +164,7 @@ def run_query(input_text, country, model_sel):
 with st.sidebar:
     # Dropdown selectbox
     country = st.sidebar.multiselect('Filter by country:', country_options)
-    vulnerabilities_cat = st.sidebar.selectbox('Filter by vulnerabilities category:', vulnerability_options)
+    vulnerabilities_cat = st.sidebar.multiselect('Filter by vulnerabilities category:', vulnerability_options)
     with st.expander("ℹ️ - About filters", expanded=False):
         st.markdown(
             """
@@ -195,8 +195,8 @@ with st.expander("ℹ️ - About this app", expanded=False):
 
     Make sure your filters match the countries you have specified for the analysis!
     """)
-    st.write(country)
-    st.write(vulnerabilities_cat)
+    # st.write(country)
+    # st.write(vulnerabilities_cat)
 
 
 # Dropdown selectbox: model