Upload app.py

app.py

@@ -123,6 +123,13 @@ INSTRUCCIONES CRÍTICAS PARA TODAS LAS FÓRMULAS:
 - NO incluyas explicaciones sobre por qué la PUV es efectiva
 - SOLO presenta la PUV, nada más
 
+COMPORTAMIENTO COMO AGENTE:
+- Al final de CADA respuesta, SIEMPRE pregunta al usuario qué más le gustaría hacer
+- Ofrece opciones específicas como: mejorar las PUVs, refinarlas, cambiar de fórmula, o crear nuevas para otro producto/servicio
+- Actúa como un agente proactivo que busca satisfacer al usuario
+- Muestra entusiasmo por ayudar y ofrece sugerencias concretas para continuar
+- Mantén un tono conversacional y amigable en todo momento
+
 IMPORTANTE: Cuando el usuario pregunte por tus funciones o capacidades, responde siempre en primera persona singular (yo hago, yo creo, yo analizo) de forma breve y concisa. No menciones que eres un equipo de expertos, sino un asistente especializado en crear PUVs.
 """
 
@@ -199,46 +206,45 @@ with st.sidebar:
         st.image("assets/robocopy_logo.png", width=300)
     
     st.write('# Chats Anteriores')
-    if st.session_state.get('chat_id') is None:
-        st.session_state.chat_id = st.selectbox(
+    if st.session_state.get('current_chat_id') is None:
+        st.session_state.current_chat_id = st.selectbox(
             label='Selecciona un chat anterior',
-            options=[new_chat_id] + list(past_chats.keys()),
-            format_func=lambda x: past_chats.get(x, 'Nuevo Chat'),
+            options=[new_chat_id] + list(st.session_state.chats_in_memory.keys()),
+            format_func=lambda x: st.session_state.chats_in_memory.get(x, 'Nuevo Chat'),
             placeholder='_',
         )
     else:
-        # This will happen the first time AI response comes in
-        st.session_state.chat_id = st.selectbox(
+        st.session_state.current_chat_id = st.selectbox(
             label='Selecciona un chat anterior',
-            options=[new_chat_id, st.session_state.chat_id] + list(past_chats.keys()),
+            options=[new_chat_id, st.session_state.current_chat_id] + list(st.session_state.chats_in_memory.keys()),
             index=1,
-            format_func=lambda x: past_chats.get(x, 'Nuevo Chat' if x != st.session_state.chat_id else st.session_state.chat_title),
+            format_func=lambda x: st.session_state.chats_in_memory.get(x, 'Nuevo Chat' if x != st.session_state.current_chat_id else st.session_state.chat_title),
             placeholder='_',
         )
     
-    # Inicializar el título de chat si es necesario
-    if st.session_state.get('chat_title') is None:
-        if st.session_state.chat_id in past_chats:
-            st.session_state.chat_title = past_chats[st.session_state.chat_id]
-        else:
+    # Botón para borrar el historial
+    if st.button('🗑️ Borrar Historial de Chat Actual'):
+        if st.session_state.current_chat_id in st.session_state.chats_in_memory:
+            del st.session_state.chats_in_memory[st.session_state.current_chat_id]
+            st.session_state.messages = []
+            st.session_state.gemini_history = []
             st.session_state.chat_title = 'Nuevo Chat'
-   
-st.write('# Creador de Propuestas Únicas de Valor')
+            st.rerun()
 
 # Inicializar mensajes y cargar historial
 if not st.session_state.get('messages'):
     st.session_state.messages = []
 
-# Cargar historial de chat si existe
-try:
-    st.session_state.messages = joblib.load(f'data/{st.session_state.chat_id}-st_messages')
-    st.session_state.gemini_history = joblib.load(f'data/{st.session_state.chat_id}-gemini_messages')
-    print('old cache')
-except:
-    if not st.session_state.get('messages'):
-        st.session_state.messages = []
+# Cargar historial de chat si existe en memoria
+if st.session_state.current_chat_id in st.session_state.chats_in_memory:
+    chat_data = st.session_state.chats_in_memory[st.session_state.current_chat_id]
+    st.session_state.messages = chat_data.get('messages', [])
+    st.session_state.gemini_history = chat_data.get('gemini_history', [])
+    st.session_state.chat_title = chat_data.get('title', 'Nuevo Chat')
+else:
+    st.session_state.messages = []
     st.session_state.gemini_history = []
-    print('new_cache made')
+    st.session_state.chat_title = 'Nuevo Chat'
 
 # Configuración del modelo
 model = genai.GenerativeModel(model_name='gemini-2.0-flash')