Update app.py

app.py

@@ -88,9 +88,12 @@ if st.session_state.selected_model != model_option:
 if st.button("Limpiar Chat"):
     st.session_state.messages = []
 
+# Cargar la imagen del avatar del asistente
+assistant_avatar = "botm.png"
+
 # Mostrar mensajes de chat del historial en la aplicación
 for message in st.session_state.messages:
-    avatar = "images/assistant_avatar.png" if message["role"] == "assistant" else "images/user_avatar.png"
+    avatar = assistant_avatar if message["role"] == "assistant" else "🧑‍💻"
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
@@ -109,7 +112,7 @@ private_instruction = (
 if prompt := st.chat_input("Escribe tu mensaje aquí..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    with st.chat_message("user", avatar="images/user_avatar.png"):  
+    with st.chat_message("user", avatar="🧑‍💻"):  
         st.markdown(prompt)
 
     # Preparar los mensajes para la API, incluyendo la instrucción privada
@@ -130,7 +133,7 @@ if prompt := st.chat_input("Escribe tu mensaje aquí..."):
         )
 
         # Usar la función generadora con st.write_stream
-        with st.chat_message("assistant", avatar="images/assistant_avatar.png"):
+        with st.chat_message("assistant", avatar=assistant_avatar):
             chat_responses_generator = generate_chat_responses(chat_completion)
             full_response = st.write_stream(chat_responses_generator)