Upload app.py

app.py

@@ -31,28 +31,13 @@ def process_message(prompt, is_example=False):
         st.markdown(prompt)
     
     state.add_message('user', prompt, USER_AVATAR_ICON)
-    
-    # Obtener el prompt mejorado primero
     enhanced_prompt = get_enhanced_prompt(prompt, is_example)
     
-    # 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
+    # Aplicar el efecto de desvanecimiento
+    apply_fade_in_effect()
+    
+    # Procesar respuesta del modelo con el nuevo efecto
+    process_model_response(enhanced_prompt)
 
 def handle_chat_title(prompt):
     """Maneja la lógica del título del chat"""
@@ -81,13 +66,17 @@ 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)
-            full_response = stream_response(response, message_placeholder, typing_indicator)
+            # Usar la nueva función con efecto de desvanecimiento
+            full_response = stream_response_with_fade(response, message_placeholder, typing_indicator)
             
-            # 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()
+            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: {str(e)}")
@@ -277,3 +266,35 @@ if state.has_prompt():
     process_message(prompt, is_example=True)
     # Limpiar el prompt
     state.clear_prompt()
+
+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 0.5s ease-in;
+        }
+        </style>
+    """, unsafe_allow_html=True)
+
+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:
+                full_response += chunk.text
+                typing_indicator.markdown("*Generando respuesta...*")
+                message_placeholder.markdown(f'<div class="fade-in-text">{full_response}</div>', unsafe_allow_html=True)
+                time.sleep(0.3)
+        
+        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 ''