Update app.py

app.py

@@ -78,7 +78,6 @@ def main():
 
         obj_retriever = obj_index.as_retriever(similarity_top_k=6)
 
-
         context = """You are an agent designed to answer scientific queries over a set of given documents.
         Please always use the tools provided to answer a question. Do not rely on prior knowledge.
         """
@@ -94,18 +93,13 @@ def main():
 
         if user_prompt:
             st.session_state.history.append({"user": user_prompt})
-            placeholder = st.empty()
-            placeholder.markdown("### Query Response:\nProcessing...")
 
             with st.spinner("Processing..."):
-                response_stream = agent.query(user_prompt, stream=True)
-                response_text = ""
-
-                for chunk in response_stream:
-                    response_text += chunk
-                    placeholder.markdown(f"### Query Response:\n{response_text}")
+                response = agent.query(user_prompt)
+                st.session_state.history.append({"agent": response})
 
-                st.session_state.history.append({"agent": response_text})
+            # Display the latest response
+            st.markdown(f"### Query Response:\n{response}")
 
         # Display chat history
         if st.session_state.history: