la interface del chat es correcta

app.py

@@ -27,9 +27,11 @@ if prompt := st.chat_input("Pregúntale a Don Simón"):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-with st.chat_message("assistant"):
-    stream = model(prompt,max_new_tokens=50)
-    response = st.write(stream)
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    with st.chat_message("assistant"):
+        stream = model(prompt,max_new_tokens=50)
+        #stream = prompt
+        response = st.write(stream)
+        st.session_state.messages.append({"role": "assistant", "content": stream})
 
 
+print(st.session_state)