Upload 10 files

app.py

@@ -108,11 +108,10 @@ for message in st.session_state.messages:
         st.markdown(message['content'])
 
 # Mensaje inicial del sistema si es un chat nuevo
-# Configuración inicial del chat
 if not st.session_state.messages:
     system_prompt = get_puv_system_prompt()
     
-    # Mostrar la carátula inicial solo si no hay mensajes
+    # Mostrar la carátula inicial
     st.markdown("""
         <div style='text-align: center; padding: 1rem 0;'>
             <h1 style='font-size: 24px; margin-bottom: 0.5rem;'>💎 RoboCopy PUV Creator</h1>
@@ -123,7 +122,7 @@ if not st.session_state.messages:
         </div>
     """, unsafe_allow_html=True)
     
-    # Mostrar los ejemplos solo si no hay mensajes
+    # Mostrar los ejemplos
     ejemplos = [
         {"texto": "¿Cuál es la mejor fórmula para mi PUV? 🤔", "prompt": "Ayúdame a elegir la mejor fórmula para mi Propuesta de Valor Única"},
         {"texto": "¿Cómo hacer mi PUV más impactante? 💫", "prompt": "Dame consejos para hacer mi Propuesta de Valor más persuasiva y memorable"},
@@ -147,6 +146,31 @@ if not st.session_state.messages:
     # Enviar el system prompt como primer mensaje
     st.session_state.chat.send_message(system_prompt)
 
+    # Solo añadir el mensaje inicial si no es un ejemplo
+    if not st.session_state.get('prompt'):
+        st.session_state.messages.append(
+            dict(
+                role=MODEL_ROLE,
+                content="""
+                Hola, soy RoboCopy tu asistente especializado en crear Propuestas de Valor Únicas.
+                
+                Para ayudarte a crear PUVs efectivas, necesito conocer:
+                
+                1. ¿Qué producto o servicio ofreces?
+                2. ¿A quién va dirigido? (describe tu público objetivo)
+                3. ¿Qué fórmula prefieres usar? Puedo ofrecerte:
+                   - Tradicional: Clara y directa
+                   - Anti-tradicional: Innovadora y disruptiva
+                   - Contrato Imposible: Audaz y sorprendente
+                   - Reto Ridículo: Humorística y relatable
+                4. ¿Cuántos ejemplos de PUVs necesitas?
+                
+                ¿Empezamos con tu producto o servicio?
+                """,
+                avatar=AI_AVATAR_ICON,
+            )
+        )
+
 # Add system message to chat history
 st.session_state.messages.append(
     dict(
@@ -378,6 +402,12 @@ if prompt := st.chat_input('Describe tu producto/servicio y audiencia objetivo..
 if 'prompt' in st.session_state and st.session_state.prompt:  # Verificamos que el prompt no esté vacío
     prompt = st.session_state.prompt
     
+    # Guardar el chat para después si es nuevo
+    if st.session_state.chat_id not in past_chats.keys():
+        temp_title = f'SesiónChat-{st.session_state.chat_id}'
+        past_chats[st.session_state.chat_id] = temp_title
+        joblib.dump(past_chats, 'data/past_chats_list')
+    
     # Mostrar el mensaje del usuario
     with st.chat_message('user', avatar=USER_AVATAR_ICON):
         st.markdown(prompt)
@@ -389,8 +419,8 @@ if 'prompt' in st.session_state and st.session_state.prompt:  # Verificamos que
         'avatar': USER_AVATAR_ICON
     })
     
-    # Procesar la respuesta del modelo solo si el prompt tiene contenido
-    if prompt.strip():  # Verificación adicional después de eliminar espacios
+    # Procesar la respuesta del modelo
+    if prompt.strip():
         with st.chat_message(MODEL_ROLE, avatar=AI_AVATAR_ICON):
             try:
                 response = st.session_state.chat.send_message(prompt)
@@ -402,10 +432,18 @@ if 'prompt' in st.session_state and st.session_state.prompt:  # Verificamos que
                     'content': response.text,
                     'avatar': AI_AVATAR_ICON
                 })
+                
+                # Guardar el historial actualizado
+                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',
+                )
             except ValueError as e:
                 st.error("Error: El mensaje no puede estar vacío. Por favor, escribe algo.")
-    else:
-        st.warning("Por favor, ingresa un mensaje antes de enviar.")
     
     # Limpiar el prompt del estado de la sesión
     st.session_state.prompt = None