response was disapeering!

app.py

@@ -109,5 +109,15 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, How can I help you today?"
             stream=True
         )
 
-        response = st.write_stream(output)
-    st.session_state.messages.append({"role": "assistant", "content": response})
+        # Create a placeholder for the streaming response
+        message_placeholder = st.empty()
+        full_response = ""
+
+        # Stream the response and accumulate it
+        for chunk in output:
+            full_response += chunk
+            message_placeholder.markdown(full_response + "▌")
+        
+        # Display final response and store it
+        message_placeholder.markdown(full_response)
+        st.session_state.messages.append({"role": "assistant", "content": full_response})