Upload app.py

app.py

@@ -11,22 +11,6 @@ from session_state import SessionState
 # Inicializar el estado de la sesión
 state = SessionState()
 
-# Función para aplicar efecto de desvanecimiento
-def apply_fade_in_effect():
-    """Aplica un efecto de desvanecimiento para el texto"""
-    st.markdown("""
-        <style>
-        @keyframes fadeIn {
-            from { opacity: 0; }
-            to { opacity: 1; }
-        }
-        .fade-in-text {
-            animation: fadeIn 3s ease-in;  /* Aumentado de 1.5s a 3s */
-            opacity: 1;
-        }
-        </style>
-    """, unsafe_allow_html=True)
-
 # Función para detectar saludos y generar respuestas personalizadas
 def is_greeting(text):
     """Detecta si el texto es un saludo simple"""
@@ -47,13 +31,28 @@ def process_message(prompt, is_example=False):
         st.markdown(prompt)
     
     state.add_message('user', prompt, USER_AVATAR_ICON)
-    enhanced_prompt = get_enhanced_prompt(prompt, is_example)
     
-    # Aplicar el efecto de desvanecimiento
-    apply_fade_in_effect()
+    # Obtener el prompt mejorado primero
+    enhanced_prompt = get_enhanced_prompt(prompt, is_example)
     
-    # Procesar respuesta del modelo con el nuevo efecto
-    process_model_response(enhanced_prompt)
+    # Mover la respuesta del modelo después del mensaje del usuario
+    with st.chat_message(MODEL_ROLE, avatar=AI_AVATAR_ICON):
+        try:
+            message_placeholder = st.empty()
+            typing_indicator = st.empty()
+            typing_indicator.markdown("*Generando respuesta...*")
+            
+            response = state.send_message(enhanced_prompt)
+            full_response = stream_response(response, message_placeholder, typing_indicator)
+            
+            if full_response:
+                state.add_message(MODEL_ROLE, full_response, AI_AVATAR_ICON)
+                state.gemini_history = state.chat.history
+                state.save_chat_history()
+                
+        except Exception as e:
+            st.error(f"Error en el streaming: {str(e)}")
+            return
 
 def handle_chat_title(prompt):
     """Maneja la lógica del título del chat"""
@@ -82,17 +81,13 @@ def process_model_response(enhanced_prompt):
             typing_indicator = st.empty()
             typing_indicator.markdown("*Generando respuesta...*")
             
-            # Aplicar el efecto de desvanecimiento
-            apply_fade_in_effect()
-            
             response = state.send_message(enhanced_prompt)
-            # Usar la nueva función con efecto de desvanecimiento
-            full_response = stream_response_with_fade(response, message_placeholder, typing_indicator)
+            full_response = stream_response(response, message_placeholder, typing_indicator)
             
-            if full_response:
-                state.add_message(MODEL_ROLE, full_response, AI_AVATAR_ICON)
-                state.gemini_history = state.chat.history
-                state.save_chat_history()
+            # Actualizar historial
+            state.add_message(role=MODEL_ROLE, content=full_response, avatar=AI_AVATAR_ICON)
+            state.gemini_history = state.chat.history
+            state.save_chat_history()
             
         except Exception as e:
             st.error(f"Error: {str(e)}")
@@ -105,7 +100,7 @@ def stream_response(response, message_placeholder, typing_indicator):
             if chunk.text:
                 for ch in chunk.text:
                     full_response += ch
-                    time.sleep(0.1)
+                    time.sleep(0.01)
                     typing_indicator.markdown("*Generando respuesta...*")
                     message_placeholder.markdown(full_response + '▌')
     except Exception as e:
@@ -116,33 +111,6 @@ def stream_response(response, message_placeholder, typing_indicator):
     message_placeholder.markdown(full_response)
     return full_response
 
-def stream_response_with_fade(response, message_placeholder, typing_indicator):
-    """Maneja el streaming de la respuesta con efecto de desvanecimiento"""
-    full_response = ''
-    try:
-        for chunk in response:
-            if chunk.text:
-                # Añadir el texto del chunk actual
-                full_response += chunk.text
-                typing_indicator.markdown("*Generando respuesta...*")
-                
-                # Aplicar el efecto de desvanecimiento con un ID único para forzar la re-renderización
-                unique_id = int(time.time() * 1000)  # Timestamp en milisegundos
-                message_placeholder.markdown(
-                    f'<div id="fade-{unique_id}" class="fade-in-text">{full_response}</div>', 
-                    unsafe_allow_html=True
-                )
-                
-                # Esperar más tiempo entre actualizaciones
-                time.sleep(1.5)  # Aumentado de 0.8 a 1.5 segundos
-        
-        typing_indicator.empty()
-        message_placeholder.markdown(f'<div class="fade-in-text">{full_response}</div>', unsafe_allow_html=True)
-        return full_response
-    except Exception as e:
-        st.error(f"Error en el streaming: {str(e)}")
-        return ''
-
 # Función para cargar CSS personalizado
 def load_css(file_path):
     with open(file_path) as f: