Update app.py

app.py

@@ -4,8 +4,8 @@ import joblib
 import streamlit as st
 import google.generativeai as genai
 from dotenv import load_dotenv
-from system_prompts import get_puv_system_prompt  # Importar el prompt iterativo
-from puv_formulas import puv_formulas  # Importar las fórmulas de PUV
+# Importación simplificada del archivo puv_formulas
+from puv_formulas import get_puv_formulas
 
 # Función para cargar CSS personalizado
 def load_css(file_path):
@@ -76,10 +76,10 @@ with st.sidebar:
             placeholder='_',
         )
     # Save new chats after a message has been sent to AI
-    # TODO: Give user a chance to name chat
     st.session_state.chat_title = f'SesiónChat-{st.session_state.chat_id}'
 
-st.write('# Chatea con Gemini')
+st.write('# Generador de Propuestas de Valor Únicas (PUVs)')
+st.write('Describe tu producto/servicio y audiencia objetivo para generar PUVs personalizadas.')
 
 # Chat history (allows to ask multiple questions)
 try:
@@ -94,49 +94,6 @@ except:
     st.session_state.messages = []
     st.session_state.gemini_history = []
     print('new_cache made')
-
-# Inicializar el modelo con system prompt solo si no existe o si cambiamos de chat
-if not hasattr(st.session_state, 'puv_model') or st.session_state.get('current_chat_id') != st.session_state.chat_id:
-    system_prompt = get_puv_system_prompt()
-    
-    # Crear un contexto con las fórmulas de PUV para el modelo
-    formulas_context = "Fórmulas de PUV disponibles:\n\n"
-    for formula_name, formula_data in puv_formulas.items():
-        formulas_context += f"## {formula_name}\n"
-        formulas_context += f"{formula_data['description']}\n\n"
-        formulas_context += "Ejemplos:\n"
-        for example in formula_data['examples']:
-            formulas_context += f"- Audiencia: {example['target_audience']}\n"
-            formulas_context += f"- Producto/Servicio: {example['product_service']}\n"
-            formulas_context += f"- PUV: {example['uvp']}\n\n"
-    
-    # Añadir las fórmulas al contexto del modelo
-    enhanced_system_prompt = system_prompt + "\n\nREFERENCIA DE FÓRMULAS:\n" + formulas_context
-    
-    st.session_state.puv_model = genai.GenerativeModel(
-        model_name='gemini-2.0-flash',
-        system_instruction=enhanced_system_prompt
-    )
-    st.session_state.current_chat_id = st.session_state.chat_id
-    
-    # Inicializar el chat con el historial existente
-    history_for_model = []
-    for msg in st.session_state.gemini_history:
-        if isinstance(msg, dict) and 'role' in msg and 'parts' in msg:
-            # El formato ya es correcto
-            history_for_model.append(msg)
-        elif isinstance(msg, dict) and 'role' in msg and 'content' in msg:
-            # Convertir del formato antiguo al nuevo
-            history_for_model.append({
-                'role': msg['role'],
-                'parts': [{'text': msg['content']}]
-            })
-    
-    st.session_state.puv_chat = st.session_state.puv_model.start_chat(
-        history=history_for_model
-    )
-
-# Modelo estándar para otras operaciones (como generar títulos)
 st.session_state.model = genai.GenerativeModel('gemini-2.0-flash')
 st.session_state.chat = st.session_state.model.start_chat(
     history=st.session_state.gemini_history,
@@ -150,8 +107,51 @@ for message in st.session_state.messages:
     ):
         st.markdown(message['content'])
 
+# Mensaje inicial del sistema si es un chat nuevo
+if not st.session_state.messages:
+    with st.chat_message(
+        name=MODEL_ROLE,
+        avatar=AI_AVATAR_ICON,
+    ):
+        st.markdown("""
+        Hola, soy tu asistente de IA especializado en crear Propuestas de Valor Únicas.
+        
+        Te ayudo a desarrollar mensajes persuasivos que conectan con tu audiencia y convierten visitantes en clientes. Para crear las mejores propuestas, necesito información sobre:
+        
+        - Tu producto o servicio
+        - Tu audiencia objetivo
+        - Los problemas que resuelves
+        - Tus ventajas competitivas
+        
+        Utilizo diferentes fórmulas como la Tradicional, Anti-tradicional, Contrato Imposible y Reto Ridículo para crear propuestas adaptadas a tu negocio.
+        
+        ¿En qué puedo ayudarte hoy?
+        """)
+    
+    # Add system message to chat history
+    st.session_state.messages.append(
+        dict(
+            role=MODEL_ROLE,
+            content="""
+            Hola, soy tu asistente de IA especializado en crear Propuestas de Valor Únicas.
+            
+            Te ayudo a desarrollar mensajes persuasivos que conectan con tu audiencia y convierten visitantes en clientes. Para crear las mejores propuestas, necesito información sobre:
+            
+            - Tu producto o servicio
+            - Tu audiencia objetivo
+            - Los problemas que resuelves
+            - Tus ventajas competitivas
+            
+            Utilizo diferentes fórmulas como la Tradicional, Anti-tradicional, Contrato Imposible y Reto Ridículo para crear propuestas adaptadas a tu negocio.
+            
+            ¿En qué puedo ayudarte hoy?
+            """,
+            avatar=AI_AVATAR_ICON,
+        )
+    )
+
 # React to user input
-if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de Valor?'):
+if prompt := st.chat_input('Describe tu producto/servicio y audiencia objetivo...'):
     # 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
@@ -196,9 +196,57 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
         )
     )
     
-    # Usar el chat con system prompt para generar la respuesta
-    response_stream = st.session_state.puv_chat.send_message(
-        prompt,
+    # Construir el prompt para el modelo con todas las fórmulas disponibles
+    puv_expert_prompt = """You are a collaborative team of world-class experts working together to create exceptional Unique Value Propositions (UVPs) that convert audience into customers. 
+    THE EXPERT TEAM: 
+    1. MASTER UVP STRATEGIST: 
+       - Expert in UVP frameworks and conversion strategies 
+       - Ensures the UVPs follow the selected framework structure precisely 
+       - Focuses on strategic placement of key conversion elements 
+    2. ELITE DIRECT RESPONSE COPYWRITER: 
+       - Trained by Gary Halbert, Gary Bencivenga, and David Ogilvy 
+       - Creates compelling hooks and persuasive elements 
+       - Ensures the language resonates with the target audience 
+    3. AUDIENCE PSYCHOLOGY SPECIALIST: 
+       - Expert in understanding audience motivations and objections 
+       - Creates content that builds genuine connection and trust 
+       - Identifies and addresses hidden fears and desires 
+    4. STORYTELLING MASTER: 
+       - Creates compelling narratives that illustrate key points 
+       - Makes complex concepts accessible through narrative 
+    5. ENGAGEMENT EXPERT: 
+       - Specializes in creating memorable and impactful statements 
+       - Ensures the UVPs are clear, concise and compelling
+       
+    You are a UVP (Unique Value Proposition) expert. Analyze (internally only, do not output the analysis) the user's message to identify information about their product/service and target audience.
+    
+    If the user is asking for UVPs, create THREE different UVPs in Spanish language using the most appropriate formula from your knowledge. If the user is asking a question about UVPs or marketing, answer it helpfully.
+    
+    When creating UVPs, follow these CRITICAL INSTRUCTIONS: 
+    - Each UVP must be specific and measurable 
+    - Focus on the transformation journey 
+    - Use natural, conversational language 
+    - Avoid generic phrases and buzzwords 
+    - Maximum 2 lines per UVP
+    
+    If creating UVPs, output in this format:
+    "Basado en tu descripción, aquí tienes tres propuestas de valor únicas (PUVs) para tu [producto/servicio]:
+    
+    1. [First UVP] 
+    2. [Second UVP] 
+    3. [Third UVP]
+    
+    Estas PUVs destacan [principales beneficios]. ¿Hay alguna que te guste más o quieres que ajuste algún aspecto?"
+    
+    If answering a question, provide a helpful, expert response in Spanish.
+    """
+    
+    # Combinar el prompt del experto con el mensaje del usuario
+    enhanced_prompt = f"{puv_expert_prompt}\n\nUser message: {prompt}"
+    
+    ## Send message to AI
+    response = st.session_state.chat.send_message(
+        enhanced_prompt,
         stream=True,
     )
     
@@ -209,25 +257,20 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
     ):
         message_placeholder = st.empty()
         full_response = ''
+        assistant_response = response
         
         # Añade un indicador de "escribiendo..."
         typing_indicator = st.empty()
-        typing_indicator.markdown("*RoboCopy está escribiendo...*")
+        typing_indicator.markdown("*Generando respuesta...*")
         
         # Streams in a chunk at a time
-        for chunk in response_stream:
-            if hasattr(chunk, 'text'):
-                # Simulate stream of chunk
-                words = chunk.text.split(' ')
-                for i, word in enumerate(words):
-                    full_response += word
-                    # Añadir espacio solo si no es la última palabra del chunk
-                    if i < len(words) - 1:
-                        full_response += ' '
-                    time.sleep(0.1)  # Velocidad ajustada para mejor legibilidad
-                    # Rewrites with a cursor at end
-                    message_placeholder.write(full_response + '▌')
-        
+        for chunk in response:
+            # Simulate stream of chunk
+            for ch in chunk.text.split(' '):
+                full_response += ch + ' '
+                time.sleep(0.1)
+                # Rewrites with a cursor at end
+                message_placeholder.write(full_response + '▌')
         # Elimina el indicador de escritura
         typing_indicator.empty()
         # Write full message with placeholder
@@ -237,14 +280,11 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
     st.session_state.messages.append(
         dict(
             role=MODEL_ROLE,
-            content=full_response,
+            content=st.session_state.chat.history[-1].parts[0].text,
             avatar=AI_AVATAR_ICON,
         )
     )
-    
-    # Actualizar el historial de Gemini
-    st.session_state.gemini_history = st.session_state.puv_chat.history
-    
+    st.session_state.gemini_history = st.session_state.chat.history
     # Save to file
     joblib.dump(
         st.session_state.messages,
@@ -253,4 +293,4 @@ if prompt := st.chat_input('¿En qué puedo ayudarte con tu Propuesta Única de
     joblib.dump(
         st.session_state.gemini_history,
         f'data/{st.session_state.chat_id}-gemini_messages',
-    )
+    )