Update app.py

app.py

@@ -74,7 +74,8 @@ def main(message: str):
         qa_chain = ConversationalRetrievalChain.from_llm(
             ChatOpenAI(temperature=0),
             vectorstore.as_retriever(),
-            memory=memory
+            memory=memory,
+            return_source_documents=True  # This enables returning the source of the answer
         )
 
         cl.Message(content=f"Selected paper: {selected_paper.title}\nLink: {selected_paper.entry_id}\nYou can now ask questions about this paper. Type 'new search' when you want to search for a different paper.").send()
@@ -90,7 +91,9 @@ def main(message: str):
         else:
             # Answer questions about the selected paper
             response = qa_chain({"question": message})
-            cl.Message(content=response["answer"]).send()
+            answer = response["answer"]
+            sources = "\n".join([f"- {doc.metadata['source']}" for doc in response["source_documents"]])
+            cl.Message(content=f"{answer}\n\nSources:\n{sources}").send()
 
 if __name__ == "__main__":
-    cl.run()
+    cl.run()