Update app.py

app.py

@@ -60,16 +60,29 @@ if uploaded_file:
     query = st.text_input("Ask a question:")
     
     if query:
-        result = qa_chain({"query": query})
-        answer = result["result"]
-        st.session_state.conversation_history.append((query, answer))
-        
-        # Display the current answer
-        st.write("**Answer:**", answer)
-        
-        # Display conversation history
-        st.subheader("Conversation History")
-        for idx, (q, a) in enumerate(st.session_state.conversation_history, 1):
-            st.write(f"**Q{idx}:** {q}")
-            st.write(f"**A{idx}:** {a}")
-            
+    # Process the query
+    result = qa_chain({"query": query})
+    answer = result["result"]
+    sources = result["source_documents"]
+    
+    # Append to conversation history
+    st.session_state.conversation_history.append((query, answer, sources))
+    
+    # Display the current answer
+    st.write("**Answer:**", answer)
+    
+    # Display the sources
+    st.subheader("Source Documents")
+    for i, doc in enumerate(sources, start=1):
+        st.write(f"**Source {i}:** {doc.metadata.get('source', 'Unknown Source')}")
+        st.write(doc.page_content[:500])  # Display the first 500 characters of the source content
+    
+    # Display conversation history
+    st.subheader("Conversation History")
+    for idx, (q, a, s) in enumerate(st.session_state.conversation_history, 1):
+        st.write(f"**Q{idx}:** {q}")
+        st.write(f"**A{idx}:** {a}")
+        st.write(f"**Sources for Q{idx}:**")
+        for i, doc in enumerate(s, start=1):
+            st.write(f"**Source {i}:** {doc.metadata.get('source', 'Unknown Source')}")
+            st.write(doc.page_content[:300])  # Show a snippet for brevity