Upload 10 files

app.py

@@ -4,8 +4,8 @@ import joblib
 import streamlit as st
 import google.generativeai as genai
 from dotenv import load_dotenv
-# Importación directa del diccionario puv_formulas
 from puv_formulas import puv_formulas
+from system_prompts import get_puv_system_prompt
 
 # Función para cargar CSS personalizado
 def load_css(file_path):
@@ -108,7 +108,9 @@ 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()
     with st.chat_message(
         name=MODEL_ROLE,
         avatar=AI_AVATAR_ICON,
@@ -130,29 +132,35 @@ if not st.session_state.messages:
         ¿Empezamos con tu producto o servicio?
         """)
     
-    # Add system message to chat history
-    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,
-        )
+    # Inicializar el chat con el sistema prompt
+    st.session_state.chat = st.session_state.model.start_chat(
+        context=system_prompt,
+        history=st.session_state.gemini_history,
+    )
+
+# Add system message to chat history
+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,
     )
+)
 
 # React to user input
 if prompt := st.chat_input('Describe tu producto/servicio y audiencia objetivo...'):