Upload app.py

app.py

@@ -52,7 +52,8 @@ new_chat_id = f'{time.time()}'
 # Configuración de avatares e identificadores
 MODEL_ROLE = 'ai'
 AI_AVATAR_ICON = '🤖'
-USER_AVATAR_ICON = '👤'
+USER_AVATAR_ICON = '👤'  # Aseguramos que el emoji sea visible
+AI_AVATAR_ICON = '🤖'
 
 # Función para manejar títulos de chat de manera unificada
 def get_chat_title(messages):
@@ -180,8 +181,9 @@ if not st.session_state.gemini_history:
     st.session_state.gemini_history = st.session_state.chat.history
 
 # Mostrar mensajes del historial
+# En la sección de mensajes, modificar:
 for message in st.session_state.messages:
-    with st.chat_message(name=message['role'], avatar=message.get('avatar')):
+    with st.chat_message(name=message['role'], avatar=message.get('avatar', USER_AVATAR_ICON if message['role'] == 'user' else AI_AVATAR_ICON)):
         st.markdown(message['content'])
 
 # React to user input
@@ -317,7 +319,14 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
     process_model_response(prompt)  # Mantener dentro del bloque
     
     # En la sección de ejemplos principal agregar:
-    if st.session_state.show_examples and not st.session_state.messages:
+    if 'show_examples' not in st.session_state:
+        st.session_state.show_examples = True  # Forzar muestra inicial
+    
+    if st.session_state.show_examples:
+        # Añadir mensajes de ejemplo con avatares visibles
+        add_message('ai', WELCOME_MESSAGE, AI_AVATAR_ICON)
+        add_message('ai', "¿Quieres que te muestre algunos ejemplos de PUVs?", AI_AVATAR_ICON)
+        
         # Usar contenedor para ancho completo
         main_container = st.container()