Upload app.py

app.py

@@ -298,10 +298,9 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
     is_first_message = len(st.session_state.messages) == 0
     
     # Guardar información del chat
-    if st.session_state.chat_id not in past_chats.keys():
+    if st.session_state.current_chat_id not in st.session_state.chats_in_memory:
         # Es una nueva conversación, generamos un título basado en el primer mensaje
-        temp_title = f'SesiónChat-{st.session_state.chat_id}'
-        past_chats[st.session_state.chat_id] = temp_title
+        temp_title = f'SesiónChat-{st.session_state.current_chat_id}'
         
         # Generamos un título basado en el contenido del mensaje
         try:
@@ -313,17 +312,18 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
             
             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')
+        
+        # Guardar en memoria
+        st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
+            'messages': st.session_state.messages,
+            'gemini_history': st.session_state.gemini_history,
+            'title': st.session_state.chat_title
+        }
     
     # Mostrar mensaje del usuario
     with st.chat_message('user', avatar=USER_AVATAR_ICON):