Update app.py

app.py

@@ -93,15 +93,36 @@ def get_refs(res):
     if ref_id in ref_ids:
       print("Ref. "+str(ref_id)+" ["+doc['meta']['country']+" "+doc['meta']['document_name']+"]: "+doc['content'])
 
+# def run_query(input_text):
+#     docs = get_docs(input_text)
+#     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))
+
 def run_query(input_text):
     docs = get_docs(input_text)
+    st.write('Debug: Documents', docs) # Debugging output
+
     res = pipe.run(query=input_text, documents=docs)
+    st.write('Debug: Results', res) # Debugging output
+
     output = res["results"][0]
     st.write('Response')
-    st.success(output)
+    st.success(str(output))
+
     references = get_refs(res)
+    st.write('Debug: References', references) # Debugging output
+
     st.write('References')
-    st.success(str(references))
+    if references:
+        st.success(str(references))
+    else:
+        st.warning('No references found.')
+
 
 
 # Setup retriever, pulling from local faiss datastore