Update app.py

app.py

@@ -17,7 +17,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # email_formulas dictionary has been moved to formulas/email_formulas.py
 
 # Cambiar el nombre de la función
-def generate_emails(target_audience, product, temperature, selected_formula, selected_angle, file_content="", image_parts=None, is_image=False, emotion="", desired_action=""):
+def generate_emails(target_audience, product, temperature, selected_formula, selected_angle, file_content="", image_parts=None, is_image=False, emotion="", desired_action="", creative_idea=""):
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,
@@ -247,6 +247,12 @@ with col1:
     target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Estudiantes Universitarios")
     product = st.text_input("¿Qué producto/servicio estás promocionando?", placeholder="Ejemplo: Curso de Inglés")
     
+    # Add new Creative Idea input field
+    creative_idea = st.text_area("Idea Creativa", placeholder="Ejemplo: Tu curso es como Netflix: ofrece contenido que engancha y soluciones que la gente realmente quiere ver", height=100)
+    
+    # Submit button
+    submit = st.button("Generar Emails")
+
     # Removed the number_of_emails selectbox
     
     # Submit button
@@ -359,7 +365,7 @@ if submit:
     
     if valid_inputs and selected_formula:
         try:
-            # Update the function call to remove number_of_emails parameter
+            # Update the function call to include creative_idea
             generated_emails = generate_emails(
                 target_audience,
                 product,
@@ -370,9 +376,25 @@ if submit:
                 image_parts if 'image_parts' in locals() else None,
                 is_image if 'is_image' in locals() else False,
                 emotion,
-                desired_action
+                desired_action,
+                creative_idea  # Add the creative idea parameter
             )
             
+            # Add creative idea to the prompt if provided
+            if creative_idea:
+                email_instruction += f"""
+CREATIVE CONCEPT:
+Use the following creative concept as the central theme for all emails in the sequence:
+"{creative_idea}"
+
+CREATIVE CONCEPT INSTRUCTIONS:
+1. This concept should be the unifying theme across all emails
+2. Use it as a metaphor or analogy throughout the sequence
+3. Develop different aspects of this concept in each email
+4. Make sure the concept naturally connects to the product benefits
+5. The concept should make the emails more memorable and engaging
+
+"""
             # Remove the inline CSS since we've moved it to main.css
             
             col2.markdown(f"""