Update app.py

app.py

@@ -138,20 +138,12 @@ def main():
                 # Get response
                 response = qa_chain.invoke({'query': prompt})
                 
-                # Extract result and sources
+                # Extract result only (no sources)
                 result = response["result"]
-                source_documents = response["source_documents"]
                 
-                # Format source documents
-                source_docs_text = "\n\n**Sources:**\n"
-                for i, doc in enumerate(source_documents, 1):
-                    source_docs_text += f"{i}. Page {doc.metadata.get('page', 'N/A')}: {doc.page_content[:100]}...\n\n"
-                
-                # Display result and sources
-                result_to_show = f"{result}\n{source_docs_text}"
-                
-                st.chat_message('assistant').markdown(result_to_show)
-                st.session_state.messages.append({'role': 'assistant', 'content': result_to_show})
+                # Display only the result
+                st.chat_message('assistant').markdown(result)
+                st.session_state.messages.append({'role': 'assistant', 'content': result})
                 
         except Exception as e:
             error_message = f"Error: {str(e)}"
@@ -160,4 +152,4 @@ def main():
             st.session_state.messages.append({'role': 'assistant', 'content': error_message})
 
 if __name__ == "__main__":
-    main()
+    main()