Update modules/ui.py

modules/ui.py

@@ -447,12 +447,14 @@ def display_chatbot_interface(lang_code):
 
     # Mostrar mensajes existentes
     with chat_container:
-        for message in st.session_state.messages:
-            st.session_state.message_id += 1
-            if message["role"] == "user":
-                st.text_area("You:", value=message["content"], height=50, key=f"user_msg_{st.session_state.message_id}", disabled=True)
-            else:
-                st.text_area("Assistant:", value=message["content"], height=50, key=f"assistant_msg_{st.session_state.message_id}", disabled=True)
+        assistant_message = st.empty()
+        full_response = ""
+        for char in response:
+            full_response += char
+            assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_{st.session_state.message_id}_typing_{st.session_state.message_counter}", disabled=True)
+            st.session_state.message_counter += 1
+            time.sleep(0.01)
+        assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_{st.session_state.message_id}", disabled=True)
 
     # Área de entrada del usuario
     user_input = st.text_input(t['input_placeholder'], key="user_input")