Update app.py

app.py

@@ -21,54 +21,29 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
         "temperature": temperature,  # Usar el valor del slider aquí
         "top_p": 0.95,
         "top_k": 64,
-        "max_output_tokens": 4096,  # Mantener el valor original de 4096
+        "max_output_tokens": 8192,
         "response_mime_type": "text/plain",
     }
 
-    try:
-        # Crear una sesión de chat con el modelo
-        chat_session = genai.Chat(model="gemini-1.5-flash", config=generation_config)
-
-        # Enviar mensaje del sistema (con rol 'system') e incluir las instrucciones específicas
-        response = chat_session.send_message(
-            messages=[
-                {
-                    "role": "system",
-                    "content": (
-                        "Eres un copywriter de clase mundial, con experiencia en la creación de ganchos, "
-                        "titulares y líneas de asunto que capturan la atención de inmediato. Tu habilidad radica en "
-                        "comprender profundamente las emociones, deseos y desafíos de una audiencia específica, lo que "
-                        "te permite diseñar estrategias de marketing personalizadas que resuenan y motivan la acción. "
-                        "Sabes cómo utilizar estructuras probadas para atraer a tu audiencia objetivo, generando interés "
-                        "y logrando una conexión poderosa que impulsa los resultados deseados en campañas publicitarias y de contenido. "
-                        "Responde en español, en tipo lista numerada. Haz ganchos inusuales que atrapen la atención. No menciones el "
-                        "producto en el gancho. No expliques el gancho o encabezado. Al responder escribe un encabezado que diga: "
-                        "'Estos son tus encabezados para enganchar a {target_audience}'."
-                    )
-                },
-                {
-                    "role": "user",
-                    "content": (
-                        f"Tu tarea es crear {number_of_headlines} ganchos o encabezados titulares llamativos "
-                        f"diseñados para {target_audience} con el fin de generar interés en {product}. "
-                        "La idea es que los ganchos sean de este tipo: "
-                        "1. Secretos: 'El secreto detrás de...'; "
-                        "2. Consejos: 'Consejos para que...'; "
-                        "3. Historias: 'La historia del...', 'Los misterios de...', 'La leyenda de...'; "
-                        "4. Deseos: 'Cómo...'; "
-                        "5. Listas: '10 razones por las que...'; "
-                        "6. Haciendo una pregunta: '¿Sabías que...'; "
-                        "7. Curiosidad: '¿Por qué...'."
-                    ),
-                }
-            ]
-        )
-
-        return to_markdown(response["candidates"][0]["content"])  # Usar to_markdown para formatear la respuesta
-
-    except Exception as e:
-        # Captura cualquier error que ocurra durante la llamada a la API
-        return f"Error al generar los titulares: {str(e)}"
+    model = genai.GenerativeModel(
+        model_name="gemini-1.5-pro",
+        generation_config=generation_config,
+        system_instruction="Eres un copywriter de clase mundial, con experiencia en la creación de ganchos, titulares y líneas de asunto que capturan la atención de inmediato. Tu habilidad radica en comprender profundamente las emociones, deseos y desafíos de una audiencia específica, lo que te permite diseñar estrategias de marketing personalizadas que resuenan y motivan la acción. Sabes cómo utilizar estructuras probadas para atraer a tu audiencia objetivo, generando interés y logrando una conexión poderosa que impulsa los resultados deseados en campañas publicitarias y de contenido. Responde en español y escribe en forma de listado para que sea fácil copiar los encabezados.",
+    )
+
+    chat_session = model.start_chat(
+        history=[
+            {
+                "role": "user",
+                "parts": [
+                    f"Tu tarea es crear {number_of_headlines} titulares llamativos diseñados para {target_audience} con el fin de generar interés en {product}. Cada titular debe ser elaborado para fomentar una acción específica, como hacer una compra, registrarse o descargar. Utiliza una variedad de formatos (preguntas, afirmaciones audaces, hechos intrigantes) para probar diferentes enfoques.",
+                ],
+            },
+        ]
+    )
+
+    response = chat_session.send_message("Genera los titulares")  # Enviar mensaje para obtener la respuesta
+    return to_markdown(response.text)  # Usar to_markdown para formatear la respuesta
 
 # Configurar la interfaz de usuario con Gradio
 iface = gr.Interface(