Upload 19 files

app.py

@@ -19,16 +19,19 @@ def initialize_model(temperature=1.0):
     )
 
 # Initialize session state variables at the app level
-if 'headline_content' not in st.session_state:
-    st.session_state.headline_content = ""
-if 'problem_content' not in st.session_state:
-    st.session_state.problem_content = ""
-if 'solution_content' not in st.session_state:
-    st.session_state.solution_content = ""
-if 'avatar_details' not in st.session_state:
-    st.session_state.avatar_details = {}
-if 'product_details' not in st.session_state:
-    st.session_state.product_details = {}
+session_state_vars = {
+    'headline_content': "", 
+    'problem_content': "", 
+    'solution_content': "",
+    'avatar_details': {}, 
+    'product_details': {},
+    'generated_sections': {},
+    'metadata': {}
+}
+
+for var, default in session_state_vars.items():
+    if var not in st.session_state:
+        st.session_state[var] = default
 
 # Función para generar contenido y procesar el JSON
 def generate_content(prompt, temperature=1.0):
@@ -54,8 +57,6 @@ def generate_content(prompt, temperature=1.0):
             json_data = json.loads(json_str.strip())
             
             # Store the JSON data in session state
-            if 'metadata' not in st.session_state:
-                st.session_state.metadata = {}
             st.session_state.metadata = json_data
             
             # Update avatar details and other metadata elements
@@ -86,7 +87,6 @@ def display_generated_content(column, content, key_prefix):
         copy_button_key = f"{key_prefix}_copy_button"
         if column.button("Copiar al portapapeles", key=copy_button_key):
             # Use JavaScript to copy to clipboard
-            # Fix: Escape backticks properly in JavaScript without using backslash in f-string
             content_escaped = content.replace("`", "\\`")
             js_code = f"""
             <script>
@@ -96,12 +96,222 @@ def display_generated_content(column, content, key_prefix):
             column.components.html(js_code, height=0)
             column.success("¡Contenido copiado al portapapeles!")
 
-# Update the main app code to preserve content between input changes
-# Add this at the beginning of your app.py file after other imports
-if 'generated_sections' not in st.session_state:
-    st.session_state.generated_sections = {}
+# Add this function to validate inputs
+def validate_inputs(audience, product):
+    """Validate that required inputs are provided."""
+    return audience.strip() != "" and product.strip() != ""
+
+# Make sure to define your UI elements before using them
+# Add this before the submit button
+st.title("Generador de Páginas de Ventas")
+
+# Create two columns for layout
+col1, col2 = st.columns([1, 2])
+
+with col1:
+    st.subheader("Configuración")
+    
+    # Input fields
+    sales_page_audience = st.text_input("Público objetivo:", 
+                                        help="Describe a quién va dirigida tu página de ventas")
+    
+    sales_page_product = st.text_input("Producto/Servicio:", 
+                                       help="Describe el producto o servicio que estás vendiendo")
+    
+    sales_page_offer = st.text_area("Oferta específica (opcional):", 
+                                    help="Detalles específicos de tu oferta, como precio, bonos, etc.")
+    
+    sales_page_section = st.selectbox(
+        "Sección a generar:",
+        [
+            "Above the Fold (Encabezado)",
+            "Problem (Problema)",
+            "Solution & Benefits (Solución y Beneficios)",
+            "Authority (Autoridad)",
+            "Offer & Bonus (Oferta y Bonus)",
+            "Social Proof (Prueba Social)",
+            "Guarantees (Garantías)",
+            "Call to Action (Llamada a la Acción)",
+            "P.S. (Post-Data)",
+            "Final Call to Action (Llamada Final)"
+        ]
+    )
+    
+    sales_page_temperature = st.slider(
+        "Creatividad:",
+        min_value=0.0,
+        max_value=1.0,
+        value=0.7,
+        step=0.1,
+        help="Valores más altos = más creatividad, valores más bajos = más consistencia"
+    )
+    
+    # Now create the submit button
+    submit_sales_page = st.button("Generar Sección", key="generate_sales_page")
+
+# Add these section generator functions before the section_functions dictionary
+
+def load_prompt_template(filename):
+    """Load a prompt template from the prompts directory."""
+    prompt_path = os.path.join(os.path.dirname(__file__), "prompts", filename)
+    with open(prompt_path, "r", encoding="utf-8") as file:
+        return file.read()
+
+def generate_above_the_fold(audience, product, temperature=0.7, offer=None):
+    """Generate the Above the Fold (headline) section."""
+    prompt_template = load_prompt_template("above_the_fold.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_problem_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Problem section."""
+    prompt_template = load_prompt_template("problem.txt")
+    
+    # Create context dictionary with any existing session state data
+    context = {}
+    if 'headline_content' in st.session_state:
+        context['headline_content'] = st.session_state.headline_content
+    if 'avatar_details' in st.session_state:
+        context['avatar_details'] = st.session_state.avatar_details
+    if 'product_details' in st.session_state:
+        context['product_details'] = st.session_state.product_details
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}\nCONTEXT: {context}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_solution_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Solution & Benefits section."""
+    prompt_template = load_prompt_template("solution_benefits.txt")
+    
+    # Create context dictionary with any existing session state data
+    context = {}
+    if 'headline_content' in st.session_state:
+        context['headline_content'] = st.session_state.headline_content
+    if 'problem_content' in st.session_state:
+        context['problem_content'] = st.session_state.problem_content
+    if 'avatar_details' in st.session_state:
+        context['avatar_details'] = st.session_state.avatar_details
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}\nCONTEXT: {context}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
 
-# Then modify the section where you handle the submit button
+def generate_authority_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Authority section."""
+    prompt_template = load_prompt_template("authority.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_offer_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Offer & Bonus section."""
+    prompt_template = load_prompt_template("offer_bonus.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER DETAILS: {offer}"
+    else:
+        prompt += "\nOFFER DETAILS: Create a compelling offer based on the product/service and audience."
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_social_proof_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Social Proof section."""
+    prompt_template = load_prompt_template("social_proof.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_guarantees_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Guarantees section."""
+    prompt_template = load_prompt_template("guarantees.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_cta_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Call to Action section."""
+    prompt_template = load_prompt_template("cta.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_ps_section(audience, product, temperature=0.7, offer=None):
+    """Generate the P.S. (Post-Data) section."""
+    prompt_template = load_prompt_template("ps.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+def generate_final_cta_section(audience, product, temperature=0.7, offer=None):
+    """Generate the Final Call to Action section."""
+    prompt_template = load_prompt_template("final_cta.txt")
+    
+    # Create the prompt with the audience and product information
+    prompt = f"{prompt_template}\n\nAUDIENCE: {audience}\nPRODUCT/SERVICE: {product}"
+    if offer:
+        prompt += f"\nOFFER: {offer}"
+    
+    # Generate the content
+    return generate_content(prompt, temperature)
+
+# Define section functions mapping once
+section_functions = {
+    "Above the Fold (Encabezado)": generate_above_the_fold,
+    "Problem (Problema)": generate_problem_section,
+    "Solution & Benefits (Solución y Beneficios)": generate_solution_section,
+    "Authority (Autoridad)": generate_authority_section,
+    "Offer & Bonus (Oferta y Bonus)": generate_offer_section,
+    "Social Proof (Prueba Social)": generate_social_proof_section,
+    "Guarantees (Garantías)": generate_guarantees_section,
+    "Call to Action (Llamada a la Acción)": generate_cta_section,
+    "P.S. (Post-Data)": generate_ps_section,
+    "Final Call to Action (Llamada Final)": generate_final_cta_section
+}
+
+# Then use it in the conditional
 if submit_sales_page:
     # Validar entradas
     if validate_inputs(sales_page_audience, sales_page_product):
@@ -114,20 +324,6 @@ if submit_sales_page:
             # Crear un contenedor para el spinner en la columna 2
             with col2:
                 with st.spinner("Generando sección de página de ventas...", show_time=True):
-                    # Mapear la selección a la función correspondiente
-                    section_functions = {
-                        "Above the Fold (Encabezado)": generate_above_the_fold,
-                        "Problem (Problema)": generate_problem_section,
-                        "Solution & Benefits (Solución y Beneficios)": generate_solution_section,
-                        "Authority (Autoridad)": generate_authority_section,
-                        "Offer & Bonus (Oferta y Bonus)": generate_offer_section,
-                        "Social Proof (Prueba Social)": generate_social_proof_section,
-                        "Guarantees (Garantías)": generate_guarantees_section,
-                        "Call to Action (Llamada a la Acción)": generate_cta_section,
-                        "P.S. (Post-Data)": generate_ps_section,
-                        "Final Call to Action (Llamada Final)": generate_final_cta_section
-                    }
-                    
                     # Obtener la función correspondiente
                     generator_func = section_functions[sales_page_section]