Update modules/ui.py

modules/ui.py

@@ -440,15 +440,16 @@ def display_chatbot_interface(lang_code):
             st.write(user_input)
 
         # Generar y mostrar respuesta del chatbot
-        with st.chat_message("assistant"):
-            with st.spinner("Pensando..."):
-                try:
-                    response = get_chatbot_response(st.session_state.chatbot, user_input, lang_code)
-                except Exception as e:
-                    st.error(f"Error al generar la respuesta del chatbot: {str(e)}")
-                    response = "Lo siento, hubo un error al procesar tu solicitud."
-                st.write(response)
-
+        with chat_container: 
+            with st.chat_message("assistant"):
+                message_placeholder = st.empty()  
+                full_response = ""
+
+                for chunk in get_chatbot_response(st.session_state.chatbot, prompt, lang_code):
+                    full_response += chunk
+                    message_placeholder.markdown(full_response + "▌")
+                message_placeholder.markdown(full_response)
+                
         # Agregar respuesta del asistente a los mensajes
         st.session_state.messages.append({"role": "assistant", "content": response})