improve retrieved document parsing

app.py

@@ -101,7 +101,7 @@ def retriever(query: str, selected_retriever: str):
     """Simulate a 'retriever' step, searching for relevant information."""
     with st.chat_message("assistant"):
         placeholder = st.empty()
-        text=""
+        text = ""
         
         if selected_retriever == "BM25 Retriever":
             message = "Using the BM25 retriever to search for documents related to your query..."
@@ -156,6 +156,10 @@ def retriever(query: str, selected_retriever: str):
 
         display_retrieved_documents = f" ||| Retrieved Documents: \n{retrieved_document_titles}"
 
+    with st.chat_message("assistant"):
+        placeholder = st.empty()
+        text = ""
+        
         for chunk in display_retrieved_documents.split():
             if chunk == "|||":
                 text += chunk + "\n"
@@ -173,7 +177,7 @@ def reasoner(query: str, documents: list[str], selected_reasoner: str, llm_clien
     """Simulate a 'reasoner' step, thinking about how to answer."""
     with st.chat_message("assistant"):
         placeholder = st.empty()
-        text=""
+        text = ""
 
         if selected_reasoner == "Claude Sonnet":
             message = "Using Claude Sonnet to reason and verify the claim..."