Update app.py

app.py

@@ -46,10 +46,12 @@ examples = [
 
 def get_docs(input_query, country = None):
     # Construct a hacky query to focus the retriever on the target country (see notes below)
-    if country:
-        query = "For the country of "+country+", "+input_query
-    else:
-        query = input_query
+    query = "For the country of "+country+", "+input_query
+    # if country:
+    #     query = "For the country of "+country+", "+input_query
+    # else:
+    #     query = input_query
+    
     # Get top 150 because we want to make sure we have 10 pertaining to the selected country
     # TEMP SOLUTION: not ideal, but FAISS document store doesnt allow metadata filtering. Needs to be tested with the full dataset
     docs = retriever.retrieve(query=query,top_k = 150)
@@ -93,7 +95,7 @@ def get_refs(res):
       doc = res['documents'][i].to_dict()
       ref_id = doc['meta']['ref_id']
       if ref_id in ref_ids:
-             result_str += "Ref. " + str(ref_id) + " [" + doc['meta']['country'] + " " + doc['meta']['document_name'] + "]: " + "'" + doc['content'] + "'<br> <br>" # Add <br> for a line break
+             result_str += "Ref. " + str(ref_id) + " [" + doc['meta']['country'] + " " + doc['meta']['document_name'] + "]: " + "*'" + doc['content'] + "'*<br> <br>" # Add <br> for a line break
   return result_str 
 
 def run_query(input_text):