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, base_copy=None):
+def generate_headlines(number_of_headlines, target_audience, product, temperature, selected_formula, selected_angle):
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,
@@ -72,18 +72,6 @@ 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"""
@@ -91,79 +79,79 @@ Use the tone, style, and specific terminology from this text whenever possible.
 INSTRUCCIONES DE ÁNGULO ESPECÍFICAS:
 {angles[selected_angle]["instruction"]}
 
-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
+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
 
-SUCCESSFUL EXAMPLES OF THE ANGLE {selected_angle}:
+EJEMPLOS EXITOSOS DEL ÁNGULO {selected_angle}:
 """
         for example in angles[selected_angle]["examples"]:
             headlines_instruction += f"- {example}\n"
 
     headlines_instruction += (
-        f"\nYour task is to create {number_of_headlines} irresistible headlines for {target_audience} "
-        f"that instantly capture attention and generate curiosity about {product}. "
+        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}. "
     )
 
     if selected_angle != "NINGUNO":
-        headlines_instruction += f"IMPORTANT: Each headline MUST follow the {selected_angle} angle clearly and consistently.\n\n"
+        headlines_instruction += f"IMPORTANTE: Cada titular DEBE seguir el ángulo {selected_angle} de manera clara y consistente.\n\n"
 
     headlines_instruction += (
-        f"Avoid obvious mentions of {product} and focus on awakening genuine interest"
+        f"Evita menciones obvias de {product} y enfócate en despertar interés genuino"
     )
 
     if selected_angle != "NINGUNO":
-        headlines_instruction += f" using the selected angle"
+        headlines_instruction += f" usando el ángulo seleccionado"
 
     headlines_instruction += ".\n\n"
 
     headlines_instruction += (
-        f"IMPORTANT: Carefully study these examples of the selected formula. "
-        f"Each example represents the style and structure to follow"
+        f"IMPORTANTE: Estudia cuidadosamente estos ejemplos de la fórmula seleccionada. "
+        f"Cada ejemplo representa el estilo y estructura a seguir"
     )
 
     if selected_angle != "NINGUNO":
-        headlines_instruction += f", adapted to the {selected_angle} angle"
+        headlines_instruction += f", adaptados al ángulo {selected_angle}"
 
     headlines_instruction += ":\n\n"
 
-    # Add 5 random examples of the formula
+    # Agregar 5 ejemplos aleatorios de la fórmula
     random_examples = random.sample(selected_formula['examples'], min(5, len(selected_formula['examples'])))
 
-    headlines_instruction += "FORMULA EXAMPLES TO FOLLOW:\n"
+    headlines_instruction += "EJEMPLOS DE LA FÓRMULA A SEGUIR:\n"
     for i, example in enumerate(random_examples, 1):
         headlines_instruction += f"{i}. {example}\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 += "\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 += f"FORMULA TO FOLLOW:\n{selected_formula['description']}\n\n"
+    headlines_instruction += f"FÓRMULA A SEGUIR:\n{selected_formula['description']}\n\n"
 
-    # CORRECT (with indentation):
+    # CORRECTO (con indentación):
     if selected_angle != "NINGUNO":
         headlines_instruction += f"""
-    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.
+    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.
 """
     else:
         headlines_instruction += f"""
-GENERATE NOW:
-Create {number_of_headlines} headlines that faithfully follow the style and structure of the examples shown.
+GENERA AHORA:
+Crea {number_of_headlines} titulares que sigan fielmente el estilo y estructura de los ejemplos mostrados.
 """
 
-    # Send the message to the model (without image conditions)
+    # Enviar el mensaje al modelo (sin condiciones de imagen)
     chat_session = model.start_chat(
         history=[
             {
@@ -204,49 +192,38 @@ 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")
-    
-    # Botón de enviar con estilo personalizado (moved right after product input)
-    submit = st.button("Generar Titulares", key="submit_button", use_container_width=True)
-    
+    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"):
-        # Moved text area for base copy above formula selection
-        base_copy = st.text_area(
-            "Correo o copy para inspiración de titulares",
-            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
-        )
-        
-        # Formula selection after the text area
+        temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1)
+
         selected_formula_key = st.selectbox(
             "Selecciona una fórmula para tus titulares",
             options=list(headline_formulas.keys())
         )
-        
-        # Angle selection moved before number of headlines
+
+        # Automatically use the keys from the angles dictionary
+        # Make sure "NINGUNO" appears first, then the rest alphabetically
         angle_keys = ["NINGUNO"] + sorted([key for key in angles.keys() if key != "NINGUNO"])
         selected_angle = st.selectbox(
             "Selecciona el ángulo para tus titulares",
             options=angle_keys
         )
-        
-        # Number of headlines moved after angle selection
-        number_of_headlines = st.selectbox("Número de Titulares", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
-        
-        # Temperature slider remains at the bottom
-        temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1)
 
     selected_formula = headline_formulas[selected_formula_key]
 
+    # Botón de enviar
+    submit = st.button("Generar Titulares")
+
 # Mostrar los titulares generados
 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: Either (Product + Audience) or Base Copy
-    valid_inputs = (has_product and has_audience) or has_base_copy
+    # Valid combination: Product + Audience
+    valid_inputs = has_product and has_audience
     
     if valid_inputs and selected_formula:
         try:
@@ -256,8 +233,7 @@ if submit:
                 product,
                 temperature,
                 selected_formula,
-                selected_angle,
-                base_copy if has_base_copy else None
+                selected_angle
             )
             col2.markdown(f"""
                 <div class="results-container">
@@ -270,5 +246,5 @@ if submit:
     else:
         if not selected_formula:
             col2.error("Por favor, selecciona una fórmula.")
-        elif not has_base_copy:
-            col2.error("Por favor, proporciona el público objetivo y el producto, o un texto base.")
+        else:
+            col2.error("Por favor, proporciona el público objetivo y el producto.")