Update streamlit_app.py

streamlit_app.py

@@ -98,16 +98,23 @@ if question and st.session_state.query_count < 50:
         with response_col:
             st.markdown("### 🤖 Assistant Response")
             
-            # Create a placeholder for the streaming response
+            # live placeholder for streaming text
             response_placeholder = st.empty()
-        
-            # Start timing
+            
+            # timing
             start_time = time.time()
-        
-            # Run the chatbot and let it stream directly into Streamlit
-            answer = st.session_state.bot.ask(question, stream_placeholder=response_placeholder)
-        
+            
+            # Reset accumulated text and attach placeholder to the handler
+            st.session_state.bot.stream_handler.current_text = ""
+            st.session_state.bot.stream_handler.placeholder = response_placeholder
+            
+            # Call ask normally (no extra keyword arg)
+            answer = st.session_state.bot.ask(question)
+            
             response_time = time.time() - start_time
+            
+            # Ensure final result is displayed without cursor
+            response_placeholder.markdown(answer)
 
 
         with debug_col: