Upload app.py

app.py

@@ -222,12 +222,11 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
     
     # Si es el primer mensaje, mostrar el mensaje de bienvenida
     if is_first_message:
-        with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
-            st.markdown(WELCOME_MESSAGE)
+        mensaje_mostrado = mostrar_con_efecto_escritura(WELCOME_MESSAGE, velocidad=0.05)
         
         st.session_state.messages.append({
             'role': MODEL_ROLE,
-            'content': WELCOME_MESSAGE,
+            'content': mensaje_mostrado,
             'avatar': AI_AVATAR_ICON,
         })
         
@@ -244,30 +243,18 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
             # Enviar mensaje al modelo
             response = st.session_state.chat.send_message(prompt, stream=True)
             
-            # Mostrar respuesta del asistente
-            with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
-                message_placeholder = st.empty()
-                full_response = ''
-                
-                # Indicador de escritura
-                typing_indicator = st.empty()
-                typing_indicator.markdown("*Estoy analizando tu información...*")
-                
-                # Mostrar respuesta por fragmentos
-                for chunk in response:
-                    for ch in chunk.text.split(' '):
-                        full_response += ch + ' '
-                        time.sleep(0.1)
-                        message_placeholder.write(full_response + '▌')
-                
-                # Eliminar indicador y mostrar respuesta completa
-                typing_indicator.empty()
-                message_placeholder.write(full_response)
+            # Procesar la respuesta completa
+            full_text = ""
+            for chunk in response:
+                full_text += chunk.text
+            
+            # Mostrar respuesta del asistente con efecto de escritura
+            mensaje_mostrado = mostrar_con_efecto_escritura(full_text)
             
             # Añadir respuesta al historial
             st.session_state.messages.append({
                 'role': MODEL_ROLE,
-                'content': st.session_state.chat.history[-1].parts[0].text,
+                'content': mensaje_mostrado,
                 'avatar': AI_AVATAR_ICON,
             })