Update app.py

app.py

@@ -58,6 +58,8 @@ def process_json():
     if (content_type == 'application/json'):
         requestQuery = request.get_json()
         response= chain.run(requestQuery['query'])
+        print(response)
+        pretty_print_docs(response['source_documents'])
         print("Ques:>>>>"+requestQuery['query']+"\n Ans:>>>"+response)
         return jsonify(botMessage=response); 
     else:
@@ -106,7 +108,7 @@ def file_Upload():
         texts = text_splitter.split_documents(documents)
         embeddings = OpenAIEmbeddings()
         vectordb = Chroma.from_documents(texts,embeddings)
-        chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever())
+        chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever(),return_source_documents=True)
         
         return render_template("index.html")
 
@@ -118,5 +120,8 @@ def KBUpload():
 def aiassist():
     return render_template("index.html")
 
+def pretty_print_docs(docs):
+    print(f"\n{'-' * 100}\n".join([f"Document {i+1}:\n\n" + d.page_content for i, d in enumerate(docs)]))
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0',  port=int(os.environ.get('PORT', 7860)))