Upload app.py

app.py

@@ -105,53 +105,142 @@ for message in st.session_state.messages:
     ):
         st.markdown(message['content'])
 
-# React to user input
-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():
-        # 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
+# Inicializar variables de estado
+if 'show_examples' not in st.session_state:
+    st.session_state.show_examples = True
+if 'selected_persona' not in st.session_state:
+    st.session_state.selected_persona = "estratega"
+
+# Función para seleccionar persona
+def select_puv_persona():
+    st.sidebar.write("## 🧠 Selecciona un Experto")
+    
+    # Inicializar la selección de persona si no existe
+    if 'selected_persona' not in st.session_state:
+        st.session_state.selected_persona = "estratega"
+    
+    # Crear botones para cada persona
+    cols = st.sidebar.columns(len(PUV_PERSONAS))
+    for i, (persona_id, persona) in enumerate(PUV_PERSONAS.items()):
+        with cols[i]:
+            if st.button(f"{persona['emoji']}\n{persona['nombre']}", key=f"btn_{persona_id}"):
+                st.session_state.selected_persona = persona_id
+                st.rerun()
+    
+    # Mostrar descripción de la persona seleccionada
+    persona_actual = PUV_PERSONAS[st.session_state.selected_persona]
+    st.sidebar.markdown(f"**{persona_actual['emoji']} {persona_actual['nombre']}**")
+    st.sidebar.markdown(f"_{persona_actual['descripcion']}_")
+    
+    return st.session_state.selected_persona
+
+# Función para modificar el prompt con la personalidad seleccionada
+def apply_persona_to_prompt(prompt, persona_id):
+    if persona_id in PUV_PERSONAS:
+        persona = PUV_PERSONAS[persona_id]
+        return f"{persona['prompt_prefix']}\n\nConsulta original: {prompt}"
+    return prompt
+
+# Función para manejar mensajes
+def add_message(role, content, avatar=None):
+    message = {
+        'role': role,
+        'content': content
+    }
+    if avatar:
+        message['avatar'] = avatar
+    st.session_state.messages.append(message)
+    return message
+
+# Función para guardar el estado del chat
+def save_chat_state():
+    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',
+    )
+
+# Función para generar título de chat
+def generate_chat_title(prompt):
+    temp_title = f'SesiónChat-{st.session_state.chat_id}'
+    try:
+        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}'")
         
-        # 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
+        generated_title = title_response.text.strip()
+        if generated_title:
+            return generated_title
+    except Exception as e:
+        print(f"Error al generar título: {e}")
+    return temp_title
+
+# Función para operaciones con manejo de errores
+def safe_operation(operation, default_value, error_message="Error en operación"):
+    try:
+        return operation()
+    except Exception as e:
+        print(f"{error_message}: {e}")
+        return default_value
+
+# Definición de perfiles de expertos en PUV
+PUV_PERSONAS = {
+    "estratega": {
+        "nombre": "Estratega de Marketing",
+        "descripcion": "Experto en posicionamiento estratégico y diferenciación de marca",
+        "prompt_prefix": "Como estratega de marketing especializado en posicionamiento de marca, voy a ayudarte a crear una PUV que destaque tu ventaja competitiva. ",
+        "emoji": "🎯"
+    },
+    "copywriter": {
+        "nombre": "Copywriter Persuasivo",
+        "descripcion": "Especialista en redacción persuasiva y mensajes de alto impacto",
+        "prompt_prefix": "Como copywriter especializado en mensajes persuasivos, voy a ayudarte a crear una PUV que conecte emocionalmente con tu audiencia. ",
+        "emoji": "✍️"
+    },
+    "analista": {
+        "nombre": "Analista de Mercado",
+        "descripcion": "Experto en análisis de mercado y comportamiento del consumidor",
+        "prompt_prefix": "Como analista de mercado especializado en comportamiento del consumidor, voy a ayudarte a crear una PUV basada en insights de tu audiencia. ",
+        "emoji": "📊"
+    },
+    "innovador": {
+        "nombre": "Innovador Disruptivo",
+        "descripcion": "Especialista en propuestas innovadoras y disruptivas",
+        "prompt_prefix": "Como especialista en innovación disruptiva, voy a ayudarte a crear una PUV que rompa con los paradigmas de tu industria. ",
+        "emoji": "💡"
+    }
+}
+
+# Modificar el procesamiento del prompt
+if prompt := st.chat_input('¿En qué puedo ayudarte hoy con tu Propuesta Única de Valor?'):
+    # Guardar el prompt original
+    original_prompt = prompt
+    
+    # Aplicar la personalidad seleccionada al prompt
+    enhanced_prompt = apply_persona_to_prompt(prompt, st.session_state.selected_persona)
+    
+    # Generar título para el chat si es nuevo
+    if st.session_state.chat_id not in past_chats.keys():
+        st.session_state.chat_title = generate_chat_title(original_prompt)
+        past_chats[st.session_state.chat_id] = st.session_state.chat_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)
-    # Add user message to chat history
-    st.session_state.messages.append(
-        dict(
-            role='user',
-            content=prompt,
-        )
-    )
-    ## Send message to AI
+    # Mostrar mensaje del usuario (siempre el original)
+    with st.chat_message('user', avatar=USER_AVATAR_ICON):
+        st.markdown(original_prompt)
+    
+    # Añadir mensaje a la historia
+    add_message('user', original_prompt, USER_AVATAR_ICON)
+    
+    # Enviar el prompt mejorado al modelo
     response = st.session_state.chat.send_message(
-        prompt,
+        enhanced_prompt,
         stream=True,
     )
     # Display assistant response in chat message container
@@ -191,11 +280,105 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):  # Mensaje más ami
     )
     st.session_state.gemini_history = st.session_state.chat.history
     # Save to file
-    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',
-    )
+    # Función para manejar mensajes
+    def add_message(role, content, avatar=None):
+        message = {
+            'role': role,
+            'content': content
+        }
+        if avatar:
+            message['avatar'] = avatar
+        st.session_state.messages.append(message)
+        return message
+    
+    # Función para guardar el estado del chat
+    def save_chat_state():
+        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',
+        )
+        
+    # Función para generar título de chat
+    def generate_chat_title(prompt):
+        temp_title = f'SesiónChat-{st.session_state.chat_id}'
+        try:
+            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}'")
+            
+            generated_title = title_response.text.strip()
+            if generated_title:
+                return generated_title
+        except Exception as e:
+            print(f"Error al generar título: {e}")
+        return temp_title
+
+# Función para operaciones con manejo de errores
+def safe_operation(operation, default_value, error_message="Error en operación"):
+    try:
+        return operation()
+    except Exception as e:
+        print(f"{error_message}: {e}")
+        return default_value
+
+# Definición de perfiles de expertos en PUV
+PUV_PERSONAS = {
+    "estratega": {
+        "nombre": "Estratega de Marketing",
+        "descripcion": "Experto en posicionamiento estratégico y diferenciación de marca",
+        "prompt_prefix": "Como estratega de marketing especializado en posicionamiento de marca, voy a ayudarte a crear una PUV que destaque tu ventaja competitiva. ",
+        "emoji": "🎯"
+    },
+    "copywriter": {
+        "nombre": "Copywriter Persuasivo",
+        "descripcion": "Especialista en redacción persuasiva y mensajes de alto impacto",
+        "prompt_prefix": "Como copywriter especializado en mensajes persuasivos, voy a ayudarte a crear una PUV que conecte emocionalmente con tu audiencia. ",
+        "emoji": "✍️"
+    },
+    "analista": {
+        "nombre": "Analista de Mercado",
+        "descripcion": "Experto en análisis de mercado y comportamiento del consumidor",
+        "prompt_prefix": "Como analista de mercado especializado en comportamiento del consumidor, voy a ayudarte a crear una PUV basada en insights de tu audiencia. ",
+        "emoji": "📊"
+    },
+    "innovador": {
+        "nombre": "Innovador Disruptivo",
+        "descripcion": "Especialista en propuestas innovadoras y disruptivas",
+        "prompt_prefix": "Como especialista en innovación disruptiva, voy a ayudarte a crear una PUV que rompa con los paradigmas de tu industria. ",
+        "emoji": "💡"
+    }
+}
+
+# Función para seleccionar persona
+def select_puv_persona():
+    st.sidebar.write("## 🧠 Selecciona un Experto")
+    
+    # Inicializar la selección de persona si no existe
+    if 'selected_persona' not in st.session_state:
+        st.session_state.selected_persona = "estratega"
+    
+    # Crear botones para cada persona
+    cols = st.sidebar.columns(len(PUV_PERSONAS))
+    for i, (persona_id, persona) in enumerate(PUV_PERSONAS.items()):
+        with cols[i]:
+            if st.button(f"{persona['emoji']}\n{persona['nombre']}", key=f"btn_{persona_id}"):
+                st.session_state.selected_persona = persona_id
+                st.rerun()
+    
+    # Mostrar descripción de la persona seleccionada
+    persona_actual = PUV_PERSONAS[st.session_state.selected_persona]
+    st.sidebar.markdown(f"**{persona_actual['emoji']} {persona_actual['nombre']}**")
+    st.sidebar.markdown(f"_{persona_actual['descripcion']}_")
+    
+    return st.session_state.selected_persona
+
+# Función para modificar el prompt con la personalidad seleccionada
+def apply_persona_to_prompt(prompt, persona_id):
+    if persona_id in PUV_PERSONAS:
+        persona = PUV_PERSONAS[persona_id]
+        return f"{persona['prompt_prefix']}\n\nConsulta original: {prompt}"
+    return prompt