Update app.py

app.py

@@ -86,7 +86,7 @@ examples = [
 def get_docs(input_query, country = [], vulnerability_cat = []):
   if not country:
     country = "All Countries"
-  if not vulnerability_cat:
+  if len(vulnerability_cat) == 0:
     if country == "All Countries":
       filters = None
     else:
@@ -96,7 +96,7 @@ def get_docs(input_query, country = [], vulnerability_cat = []):
           filters = {'vulnerability_cat': {'$in': vulnerability_cat}}
     else:
       filters = {'country': {'$in': country},'vulnerability_cat': {'$in': vulnerability_cat}}
-  docs = retriever.retrieve(query=input_query, filters = filters, top_k = 10)
+  docs = retriever.retrieve(query=query, filters = filters, top_k = 10)
   # Break out the key fields and convert to pandas for filtering
   docs = [{**x.meta,"score":x.score,"content":x.content} for x in docs]
   df_docs = pd.DataFrame(docs)