Spaces:
Sleeping
Sleeping
File size: 2,332 Bytes
d51e47b 0128eaf d51e47b 0128eaf d51e47b 0128eaf d51e47b 0128eaf d51e47b 0128eaf d51e47b a7e8a16 d51e47b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import gradio as gr
import google.generativeai as genai
import os
from dotenv import load_dotenv
from gradio import Markdown
import textwrap
# Cargar variables de entorno
load_dotenv()
# Configurar la API de Google Gemini
genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
def to_markdown(text):
text = text.replace('•', ' *') # Convertir los puntos en listas con asteriscos
return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
def generate_headlines(number_of_headlines, target_audience, product, temperature):
try:
# Crear la configuración del modelo
generation_config = {
"temperature": temperature,
"top_p": 0.95,
"top_k": 64,
"max_output_tokens": 2048,
"response_mime_type": "text/plain",
}
model = genai.GenerativeModel(
model_name="gemini-1.5-flash",
generation_config=generation_config,
system_instruction="Eres un copywriter de clase mundial..."
)
chat_session = model.start_chat(
history=[
{
"role": "user",
"parts": [
f"Tu tarea es crear {number_of_headlines} ganchos o encabezados titulares llamativos..."
],
},
]
)
response = chat_session.send_message("Genera los titulares")
return to_markdown(response.text)
except Exception as e:
return f"Error al generar los titulares: {str(e)}"
# Configurar la interfaz de usuario con Gradio
iface = gr.Interface(
fn=generate_headlines,
inputs=[
gr.Dropdown(choices=[str(i) for i in range(1, 11)], label="Número de Titulares", value="5"),
gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios"),
gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés"),
gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="Creatividad")
],
outputs=gr.Markdown(label="Titulares Generados"),
title="Generador de Titulares",
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."
)
# Lanza la interfaz
iface.launch()
|