Update app.py

app.py

@@ -16,7 +16,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Fórmulas con ejemplos y explicaciones
 # headline_formulas dictionary has been moved to formulas/headline_formulas.py
 
-def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle):
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle, base_copy=None):
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,
@@ -72,6 +72,18 @@ IMPORTANT:
     # Iniciar el prompt con las instrucciones del sistema
     headlines_instruction = f"{system_prompt}\n\n"
     
+    # Add base copy instructions if provided
+    if base_copy and base_copy.strip():
+        headlines_instruction += f"""
+BASE COPY TO USE:
+The following text should be used as the primary source of information and inspiration for generating headlines:
+
+{base_copy}
+
+IMPORTANT: Extract key concepts, benefits, and language from this base copy to create your headlines.
+Use the tone, style, and specific terminology from this text whenever possible.
+"""
+
     # Añadir instrucciones de ángulo solo si no es "NINGUNO"
     if selected_angle != "NINGUNO":
         headlines_instruction += f"""
@@ -79,79 +91,79 @@ IMPORTANT:
 INSTRUCCIONES DE ÁNGULO ESPECÍFICAS:
 {angles[selected_angle]["instruction"]}
 
-IMPORTANTE: El ángulo {selected_angle} debe aplicarse como una "capa de estilo" sobre la estructura de la fórmula:
-1. Mantén la estructura base de la fórmula intacta
-2. Aplica el tono y estilo del ángulo {selected_angle}
-3. Asegura que cada elemento de la fórmula refleje el ángulo
-4. El ángulo afecta al "cómo" se dice, no al "qué" se dice
+IMPORTANT: The angle {selected_angle} must be applied as a "style layer" over the formula structure:
+1. Keep the base structure of the formula intact
+2. Apply the tone and style of the angle {selected_angle}
+3. Ensure that each element of the formula reflects the angle
+4. The angle affects "how" it is said, not "what" is said
 
-EJEMPLOS EXITOSOS DEL ÁNGULO {selected_angle}:
+SUCCESSFUL EXAMPLES OF THE ANGLE {selected_angle}:
 """
         for example in angles[selected_angle]["examples"]:
             headlines_instruction += f"- {example}\n"
 
     headlines_instruction += (
-        f"\nTu tarea es crear {number_of_headlines} titulares irresistibles para {target_audience} "
-        f"que capturen la atención instantáneamente y generen curiosidad sobre {product}. "
+        f"\nYour task is to create {number_of_headlines} irresistible headlines for {target_audience} "
+        f"that instantly capture attention and generate curiosity about {product}. "
     )
 
     if selected_angle != "NINGUNO":
-        headlines_instruction += f"IMPORTANTE: Cada titular DEBE seguir el ángulo {selected_angle} de manera clara y consistente.\n\n"
+        headlines_instruction += f"IMPORTANT: Each headline MUST follow the {selected_angle} angle clearly and consistently.\n\n"
 
     headlines_instruction += (
-        f"Evita menciones obvias de {product} y enfócate en despertar interés genuino"
+        f"Avoid obvious mentions of {product} and focus on awakening genuine interest"
     )
 
     if selected_angle != "NINGUNO":
-        headlines_instruction += f" usando el ángulo seleccionado"
+        headlines_instruction += f" using the selected angle"
 
     headlines_instruction += ".\n\n"
 
     headlines_instruction += (
-        f"IMPORTANTE: Estudia cuidadosamente estos ejemplos de la fórmula seleccionada. "
-        f"Cada ejemplo representa el estilo y estructura a seguir"
+        f"IMPORTANT: Carefully study these examples of the selected formula. "
+        f"Each example represents the style and structure to follow"
     )
 
     if selected_angle != "NINGUNO":
-        headlines_instruction += f", adaptados al ángulo {selected_angle}"
+        headlines_instruction += f", adapted to the {selected_angle} angle"
 
     headlines_instruction += ":\n\n"
 
-    # Agregar 5 ejemplos aleatorios de la fórmula
+    # Add 5 random examples of the formula
     random_examples = random.sample(selected_formula['examples'], min(5, len(selected_formula['examples'])))
 
-    headlines_instruction += "EJEMPLOS DE LA FÓRMULA A SEGUIR:\n"
+    headlines_instruction += "FORMULA EXAMPLES TO FOLLOW:\n"
     for i, example in enumerate(random_examples, 1):
         headlines_instruction += f"{i}. {example}\n"
 
-    headlines_instruction += "\nINSTRUCCIONES ESPECÍFICAS:\n"
-    headlines_instruction += "1. Mantén la misma estructura y longitud que los ejemplos anteriores\n"
-    headlines_instruction += "2. Usa el mismo tono y estilo de escritura\n"
-    headlines_instruction += "3. Replica los patrones de construcción de frases\n"
-    headlines_instruction += "4. Conserva el nivel de especificidad y detalle\n"
-    headlines_instruction += f"5. Adapta el contenido para {target_audience} manteniendo la esencia de los ejemplos\n\n"
+    headlines_instruction += "\nSPECIFIC INSTRUCTIONS:\n"
+    headlines_instruction += "1. Maintain the same structure and length as the previous examples\n"
+    headlines_instruction += "2. Use the same tone and writing style\n"
+    headlines_instruction += "3. Replicate the patterns of phrase construction\n"
+    headlines_instruction += "4. Preserve the level of specificity and detail\n"
+    headlines_instruction += f"5. Adapt the content for {target_audience} while maintaining the essence of the examples\n\n"
 
-    headlines_instruction += f"FÓRMULA A SEGUIR:\n{selected_formula['description']}\n\n"
+    headlines_instruction += f"FORMULA TO FOLLOW:\n{selected_formula['description']}\n\n"
 
-    # CORRECTO (con indentación):
+    # CORRECT (with indentation):
     if selected_angle != "NINGUNO":
         headlines_instruction += f"""
-    RECORDATORIO FINAL:
-    1. Sigue la estructura de la fórmula seleccionada
-    2. Aplica el ángulo como una "capa de estilo"
-    3. Mantén la coherencia entre fórmula y ángulo
-    4. Asegura que cada titular refleje ambos elementos
-
-GENERA AHORA:
-Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura de los ejemplos mostrados.
+    FINAL REMINDER:
+    1. Follow the structure of the selected formula
+    2. Apply the angle as a "style layer"
+    3. Maintain coherence between formula and angle
+    4. Ensure each headline reflects both elements
+
+GENERATE NOW:
+Create {number_of_headlines} headlines that faithfully follow the style and structure of the examples shown.
 """
     else:
         headlines_instruction += f"""
-GENERA AHORA:
-Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura de los ejemplos mostrados.
+GENERATE NOW:
+Create {number_of_headlines} headlines that faithfully follow the style and structure of the examples shown.
 """
 
-    # Enviar el mensaje al modelo (sin condiciones de imagen)
+    # Send the message to the model (without image conditions)
     chat_session = model.start_chat(
         history=[
             {
@@ -192,16 +204,26 @@ col1, col2 = st.columns([1, 2])
 with col1:
     target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Estudiantes Universitarios")
     product = st.text_input("¿Qué producto tienes en mente?", placeholder="Ejemplo: Curso de Inglés")
-    number_of_headlines = st.selectbox("Número de Titulares", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
-
+    
     # Crear un único acordeón para fórmula, creatividad y ángulo
     with st.expander("Personaliza tus titulares"):
-        temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1)
-
+        # Moved formula selection above number of headlines
         selected_formula_key = st.selectbox(
             "Selecciona una fórmula para tus titulares",
             options=list(headline_formulas.keys())
         )
+        
+        # Added number of headlines after formula selection
+        number_of_headlines = st.selectbox("Número de Titulares", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
+        
+        # New text area for base copy
+        base_copy = st.text_area(
+            "Texto base para generar titulares (opcional)",
+            placeholder="Agrega aquí un texto que sirva como base para generar los titulares. Si lo dejas vacío, se generarán basados en el producto y público objetivo.",
+            height=150
+        )
+        
+        temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1)
 
         # Automatically use the keys from the angles dictionary
         # Make sure "NINGUNO" appears first, then the rest alphabetically
@@ -221,9 +243,10 @@ if submit:
     # Check if we have valid inputs
     has_product = product.strip() != ""
     has_audience = target_audience.strip() != ""
+    has_base_copy = base_copy and base_copy.strip() != ""
     
-    # Valid combination: Product + Audience
-    valid_inputs = has_product and has_audience
+    # Valid combination: Either (Product + Audience) or Base Copy
+    valid_inputs = (has_product and has_audience) or has_base_copy
     
     if valid_inputs and selected_formula:
         try:
@@ -233,7 +256,8 @@ if submit:
                 product,
                 temperature,
                 selected_formula,
-                selected_angle
+                selected_angle,
+                base_copy if has_base_copy else None
             )
             col2.markdown(f"""
                 <div class="results-container">
@@ -246,5 +270,5 @@ if submit:
     else:
         if not selected_formula:
             col2.error("Por favor, selecciona una fórmula.")
-        else:
-            col2.error("Por favor, proporciona el público objetivo y el producto.")
+        elif not has_base_copy:
+            col2.error("Por favor, proporciona el público objetivo y el producto, o un texto base.")