Upload app.py

app.py

@@ -347,9 +347,12 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
             'avatar': AI_AVATAR_ICON,
         })
         
-        # Guardar el historial actualizado
-        joblib.dump(st.session_state.messages, f'data/{st.session_state.chat_id}-st_messages')
-        # Salir de la función para no procesar más este mensaje
+        # Actualizar chat en memoria
+        st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
+            'messages': st.session_state.messages,
+            'gemini_history': st.session_state.gemini_history,
+            'title': st.session_state.chat_title
+        })
         st.rerun()
     
     # Implementación de reintentos con retroceso exponencial
@@ -378,29 +381,36 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
             # Actualizar historial
             st.session_state.gemini_history = st.session_state.chat.history
             
-            # Guardar historial
-            joblib.dump(st.session_state.messages, f'data/{st.session_state.chat_id}-st_messages')
-            joblib.dump(st.session_state.gemini_history, f'data/{st.session_state.chat_id}-gemini_messages')
+            # Actualizar chat en memoria
+            st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
+                'messages': st.session_state.messages,
+                'gemini_history': st.session_state.gemini_history,
+                'title': st.session_state.chat_title
+            })
             
-            # Salir del bucle si la solicitud fue exitosa
             break
             
         except Exception as e:
             retry_count += 1
             if retry_count >= max_retries:
-                # Mostrar mensaje de error si se agotan los reintentos
+                error_message = f"Lo siento, estoy experimentando problemas para procesar tu solicitud. Por favor, intenta de nuevo más tarde. Error: {str(e)}"
+                
                 with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
-                    st.error(f"Lo siento, estoy experimentando problemas para procesar tu solicitud. Por favor, intenta de nuevo más tarde. Error: {str(e)}")
+                    st.error(error_message)
                 
                 st.session_state.messages.append({
                     'role': MODEL_ROLE,
-                    'content': f"Lo siento, estoy experimentando problemas para procesar tu solicitud. Por favor, intenta de nuevo más tarde. Error: {str(e)}",
+                    'content': error_message,
                     'avatar': AI_AVATAR_ICON,
                 })
                 
-                joblib.dump(st.session_state.messages, f'data/{st.session_state.chat_id}-st_messages')
+                # Actualizar chat en memoria con el error
+                st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
+                    'messages': st.session_state.messages,
+                    'gemini_history': st.session_state.gemini_history,
+                    'title': st.session_state.chat_title
+                })
             else:
-                # Esperar antes de reintentar (retroceso exponencial)
                 wait_time = (2 ** retry_count) + (time.time() % 1)
                 time.sleep(wait_time)