Update app.py

app.py

@@ -93,18 +93,19 @@ 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'] + "' \n "
+            result_str += "Ref. " + str(ref_id) + " [" + doc['meta']['country'] + " " + doc['meta']['document_name'] + "]: " + "'" + doc['content'] + "'  " # Two spaces at the end
   return result_str 
 
 def run_query(input_text):
     docs = get_docs(input_text)
+    st.write('Debug: country', country) # Debugging country
     res = pipe.run(query=input_text, documents=docs)
     output = res["results"][0]
     st.write('Response')
     st.success(output)
     references = get_refs(res)
     st.write('References')
-    st.success(str(references))
+    st.markdown(references)
 
 # def run_query(input_text):
 #     docs = get_docs(input_text)