Upload app.py

app.py

@@ -56,28 +56,28 @@ with st.sidebar:
     with col2:
         st.image("assets/robocopy_logo.png", width=300)
     
-    st.write('# Past Chats')
+    st.write('# Chats Anteriores')
     if st.session_state.get('chat_id') is None:
         st.session_state.chat_id = st.selectbox(
-            label='Pick a past chat',
+            label='Selecciona un chat anterior',
             options=[new_chat_id] + list(past_chats.keys()),
-            format_func=lambda x: past_chats.get(x, 'New Chat'),
+            format_func=lambda x: past_chats.get(x, 'Nuevo Chat'),
             placeholder='_',
         )
     else:
         # This will happen the first time AI response comes in
         st.session_state.chat_id = st.selectbox(
-            label='Pick a past chat',
+            label='Selecciona un chat anterior',
             options=[new_chat_id, st.session_state.chat_id] + list(past_chats.keys()),
             index=1,
-            format_func=lambda x: past_chats.get(x, 'New Chat' if x != st.session_state.chat_id else st.session_state.chat_title),
+            format_func=lambda x: past_chats.get(x, 'Nuevo Chat' if x != st.session_state.chat_id else st.session_state.chat_title),
             placeholder='_',
         )
     # Save new chats after a message has been sent to AI
     # TODO: Give user a chance to name chat
-    st.session_state.chat_title = f'ChatSession-{st.session_state.chat_id}'
+    st.session_state.chat_title = f'SesiónChat-{st.session_state.chat_id}'
 
-st.write('# Chat with Gemini')
+st.write('# Chatea con Gemini')
 
 # Chat history (allows to ask multiple questions)
 try:
@@ -109,8 +109,36 @@ for message in st.session_state.messages:
 if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):  # Mensaje más amigable
     # Save this as a chat for later
     if st.session_state.chat_id not in past_chats.keys():
-        past_chats[st.session_state.chat_id] = st.session_state.chat_title
-        joblib.dump(past_chats, 'data/past_chats_list')
+        # Es una nueva conversación, generemos un título basado en el primer mensaje
+        # Primero, guardamos un título temporal
+        temp_title = f'SesiónChat-{st.session_state.chat_id}'
+        past_chats[st.session_state.chat_id] = temp_title
+        
+        # Generamos un título basado en el contenido del mensaje
+        try:
+            # Usamos el mismo modelo para generar un título corto
+            title_generator = genai.GenerativeModel('gemini-2.0-flash')
+            title_response = title_generator.generate_content(
+                f"Genera un título corto (máximo 5 palabras) que describa de qué trata esta consulta, sin usar comillas ni puntuación: '{prompt}'")
+            
+            # Obtenemos el título generado
+            generated_title = title_response.text.strip()
+            
+            # Actualizamos el título en past_chats
+            if generated_title:
+                st.session_state.chat_title = generated_title
+                past_chats[st.session_state.chat_id] = generated_title
+            else:
+                st.session_state.chat_title = temp_title
+        except Exception as e:
+            print(f"Error al generar título: {e}")
+            st.session_state.chat_title = temp_title
+    else:
+        # Ya existe esta conversación, usamos el título guardado
+        st.session_state.chat_title = past_chats[st.session_state.chat_id]
+    
+    joblib.dump(past_chats, 'data/past_chats_list')
+    
     # Display user message in chat message container
     with st.chat_message('user', avatar=USER_AVATAR_ICON):  # Añade el avatar del usuario
         st.markdown(prompt)