Update app.py

app.py

@@ -483,7 +483,6 @@ if prompt and not st.session_state.chat_active:
 
     # Generate AI response
     with st.chat_message("assistant"):
-        message_placeholder = st.text()
         full_response = ""
         
         try:
@@ -497,7 +496,7 @@ if prompt and not st.session_state.chat_active:
             
             # Display the response
 
-            message_placeholder.markdown(full_response)
+            st.write(full_response)
             follow_up_output = agent_executor.invoke({
                 "input": f"Extract 1-2 follow-up questions from the following text: {full_response}",
                 "chat_history": st.session_state.chat_history
@@ -510,7 +509,7 @@ if prompt and not st.session_state.chat_active:
         except Exception as e:
             logger.error(f"Error generating response: {e}")
             full_response = "I apologize, but an error occurred while generating the response. Please try again."
-            message_placeholder.markdown(full_response)
+            st.write(full_response)
 
     # Add AI response to chat history
     st.session_state.chat_history.append({"role": "assistant", "content": full_response})