Update app.py

app.py

@@ -87,22 +87,31 @@ ANALYSIS INSTRUCTIONS:
 5. Use relevant terms, phrases, or concepts from the content
 6. Maintain consistency with the brand identity or main message
 7. Adapt the headlines to resonate with the provided content
+8. Identify any creative elements like characters, movies, cultural references, or storytelling devices
+9. Incorporate these creative elements into your headlines when appropriate
+
+CREATIVE CONTEXT GUIDELINES:
+- If the content mentions specific characters, incorporate their traits or quotes in headlines
+- If movie or pop culture references exist, use similar tone or style in headlines
+- Match the emotional tone and creative direction of the original content
+- Maintain the same storytelling approach if present in the content
+- Use metaphors, analogies or creative devices similar to those in the content
 
 IMPORTANT COMBINATIONS:
 """
         # Updated conditions for specific input combinations
         if product and not target_audience:
-            headlines_instruction += f"""- FILE + PRODUCT: You have a reference document and product ({product}). Create headlines that highlight this specific product's benefits and features using insights from the document. Extract audience information from the document to better target the headlines.
+            headlines_instruction += f"""- FILE + PRODUCT: You have a reference document and product ({product}). Create headlines that highlight this specific product's benefits and features using insights from the document. Extract audience information from the document to better target the headlines. Ensure headlines match the creative context and tone of the original content.
 """
         elif target_audience and not product:
-            headlines_instruction += f"""- FILE + TARGET AUDIENCE: You have a reference document and target audience ({target_audience}). Create headlines tailored to this specific audience using language and concepts from the document. Identify products or services from the document that would appeal to this audience.
+            headlines_instruction += f"""- FILE + TARGET AUDIENCE: You have a reference document and target audience ({target_audience}). Create headlines tailored to this specific audience using language and concepts from the document. Identify products or services from the document that would appeal to this audience. Maintain the creative direction and contextual elements from the original content.
 """
         elif product and target_audience:
-            headlines_instruction += f"""- PRODUCT + TARGET AUDIENCE: You have both product ({product}) and target audience ({target_audience}). Create headlines that connect this specific product with this specific audience, using insights from the document to strengthen the connection.
+            headlines_instruction += f"""- PRODUCT + TARGET AUDIENCE: You have both product ({product}) and target audience ({target_audience}). Create headlines that connect this specific product with this specific audience, using insights from the document to strengthen the connection. Incorporate any creative elements, characters, or thematic approaches from the original content.
 """
         
         headlines_instruction += """
-IMPORTANT: Naturally integrate the elements found in the content with the selected formula and angle.
+IMPORTANT: Naturally integrate the elements found in the content with the selected formula and angle. If the content has a specific creative direction (storytelling, character-based, movie reference, etc.), ensure your headlines reflect this creative context while maintaining the formula structure.
 """
     
     # Añadir instrucciones de ángulo solo si no es "NINGUNO"
@@ -243,6 +252,13 @@ with col1:
 
     # Crear un único acordeón para fórmula, creatividad y ángulo
     with st.expander("Personaliza tus titulares"):
+        # Añadir área de texto para copy base
+        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
+        )
+        
         temperature = st.slider("Creatividad", min_value=0.0, max_value=2.0, value=1.0, step=0.1)
 
         selected_formula_key = st.selectbox(
@@ -328,19 +344,27 @@ if submit:
     has_file = 'file_content' in locals() and file_content.strip() != ""
     has_product = product.strip() != ""
     has_audience = target_audience.strip() != ""
+    has_base_copy = 'base_copy' in locals() and base_copy.strip() != ""
     
     # Valid combinations:
     # 1. File + Product (no audience needed)
     # 2. File + Audience (no product needed)
     # 3. Product + Audience (traditional way)
+    # 4. Base copy alone (new option)
     valid_inputs = (
         (has_file and has_product) or
         (has_file and has_audience) or
-        (has_product and has_audience)
+        (has_product and has_audience) or
+        has_base_copy
     )
     
     if valid_inputs and selected_formula:
         try:
+            # If base_copy is provided, use it as file_content
+            content_for_generation = file_content if 'file_content' in locals() and file_content.strip() != "" else ""
+            if has_base_copy:
+                content_for_generation = base_copy
+                
             generated_headlines = generate_headlines(
                 number_of_headlines,
                 target_audience,
@@ -348,7 +372,7 @@ if submit:
                 temperature,
                 selected_formula,
                 selected_angle,
-                file_content if 'file_content' in locals() else "",
+                content_for_generation,
                 image_parts if 'image_parts' in locals() else None,
                 is_image if 'is_image' in locals() else False
             )
@@ -364,4 +388,4 @@ if submit:
         if not selected_formula:
             col2.error("Por favor, selecciona una fórmula.")
         else:
-            col2.error("Por favor, proporciona al menos una de estas combinaciones: archivo + producto, archivo + público objetivo, o producto + público objetivo.")
+            col2.error("Por favor, proporciona al menos una de estas combinaciones: archivo + producto, archivo + público objetivo, producto + público objetivo, o un texto de inspiración.")