Upload app.py

app.py

@@ -105,142 +105,53 @@ for message in st.session_state.messages:
     ):
         st.markdown(message['content'])
 
-# 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}'")
-        
-        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
+# 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():
-        st.session_state.chat_title = generate_chat_title(original_prompt)
-        past_chats[st.session_state.chat_id] = st.session_state.chat_title
+        # 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
+        
+        # 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
     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')
     
-    # 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
+    # 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
     response = st.session_state.chat.send_message(
-        enhanced_prompt,
+        prompt,
         stream=True,
     )
     # Display assistant response in chat message container
@@ -259,12 +170,12 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy con tu Propuesta Única
         # Streams in a chunk at a time
         for chunk in response:
             # Simulate stream of chunk
-            for word in chunk.text.split(' '):
-                full_response += word + ' '
-                time.sleep(0.1)  # Velocidad ajustada para mejor legibilidad
+            # TODO: Chunk missing `text` if API stops mid-stream ("safety"?)
+            for ch in chunk.text.split(' '):
+                full_response += ch + ' '
+                time.sleep(0.1)  # Aumentado de 0.05 a 0.1 segundos para una velocidad más lenta
                 # Rewrites with a cursor at end
                 message_placeholder.write(full_response + '▌')
-        
         # Elimina el indicador de escritura
         typing_indicator.empty()
         # Write full message with placeholder
@@ -280,105 +191,11 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy con tu Propuesta Única
     )
     st.session_state.gemini_history = st.session_state.chat.history
     # Save to file
-    # 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
+    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',
+    )