JeCabrera commited on
Commit
50f0928
·
verified ·
1 Parent(s): a7e8a16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -30
app.py CHANGED
@@ -16,38 +16,42 @@ def to_markdown(text):
16
  return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
17
 
18
  def generate_headlines(number_of_headlines, target_audience, product, temperature):
19
- try:
20
- # Crear la configuración del modelo
21
- generation_config = {
22
- "temperature": temperature,
23
- "top_p": 0.95,
24
- "top_k": 64,
25
- "max_output_tokens": 2048,
26
- "response_mime_type": "text/plain",
27
- }
28
 
29
- model = genai.GenerativeModel(
30
- model_name="gemini-1.5-flash",
31
- generation_config=generation_config,
32
- system_instruction="Eres un copywriter de clase mundial..."
33
- )
34
 
35
- chat_session = model.start_chat(
36
- history=[
37
- {
38
- "role": "user",
39
- "parts": [
40
- f"Tu tarea es crear {number_of_headlines} ganchos o encabezados titulares llamativos..."
41
- ],
42
- },
43
- ]
44
- )
 
 
 
 
 
 
 
 
45
 
46
- response = chat_session.send_message("Genera los titulares")
47
- return to_markdown(response.text)
48
-
49
- except Exception as e:
50
- return f"Error al generar los titulares: {str(e)}"
51
 
52
  # Configurar la interfaz de usuario con Gradio
53
  iface = gr.Interface(
@@ -58,7 +62,7 @@ iface = gr.Interface(
58
  gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés"),
59
  gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="Creatividad")
60
  ],
61
- outputs=gr.Markdown(label="Titulares Generados"),
62
  title="Generador de Titulares",
63
  description="Usa el poder de Gemini AI para crear titulares atractivos. Ajusta los parámetros para generar titulares que capturen la atención de tu audiencia."
64
  )
 
16
  return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
17
 
18
  def generate_headlines(number_of_headlines, target_audience, product, temperature):
19
+ # Crear la configuración del modelo
20
+ generation_config = {
21
+ "temperature": temperature, # Usar el valor del slider aquí
22
+ "top_p": 0.95,
23
+ "top_k": 64,
24
+ "max_output_tokens": 2048,
25
+ "response_mime_type": "text/plain",
26
+ }
 
27
 
28
+ model = genai.GenerativeModel(
29
+ model_name="gemini-1.5-flash",
30
+ generation_config=generation_config,
31
+ 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, en tipo lista numerada. Haz ganchos inusuales que atrapen la atención. No menciones el producto directamente 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}'."
32
+ )
33
 
34
+ chat_session = model.start_chat(
35
+ history=[
36
+ {
37
+ "role": "user",
38
+ "parts": [
39
+ f"Tu tarea es crear {number_of_headlines} ganchos o encabezados titulares llamativos diseñados para {target_audience} con el fin de generar interés en {product}. "
40
+ "La idea es que los ganchos sean de este tipo: "
41
+ "1. Secretos: 'El secreto detrás de...'; "
42
+ "2. Consejos: 'Consejos para que...'; "
43
+ "3. Historias: 'La historia del...', 'Los misterios de...', 'La leyenda de...'; "
44
+ "4. Deseos: 'Cómo...'; "
45
+ "5. Listas: '10 razones por las que...'; "
46
+ "6. Haciendo una pregunta: '¿Sabías que...'; "
47
+ "7. Curiosidad: '¿Por qué...'."
48
+ ],
49
+ },
50
+ ]
51
+ )
52
 
53
+ response = chat_session.send_message("Genera los titulares") # Enviar mensaje para obtener la respuesta
54
+ return to_markdown(response.text) # Usar to_markdown para formatear la respuesta
 
 
 
55
 
56
  # Configurar la interfaz de usuario con Gradio
57
  iface = gr.Interface(
 
62
  gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés"),
63
  gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="Creatividad")
64
  ],
65
+ outputs=gr.Markdown(label="Titulares Generados"), # Eliminado el placeholder
66
  title="Generador de Titulares",
67
  description="Usa el poder de Gemini AI para crear titulares atractivos. Ajusta los parámetros para generar titulares que capturen la atención de tu audiencia."
68
  )