Update app.py

app.py

@@ -1,42 +1,38 @@
 import gradio as gr
-import anthropic
-import os  # Importa el módulo os para manejar variables de entorno
+import google.generativeai as genai
+import os
+from dotenv import load_dotenv
 
-# Obtén la clave de API desde la variable de entorno
-api_key = os.getenv("ANTHROPIC_API_KEY")
+# Cargar variables de entorno
+load_dotenv()
 
-# Verifica si la clave de API está configurada
-if not api_key:
-    raise ValueError("Falta la clave de API de Anthropoid. Asegúrate de configurarla en los secretos del repositorio.")
-
-# Configura el cliente de la API de Claude (Anthropic)
-client = anthropic.Anthropic(api_key=api_key)
+# Configurar la API de Google Gemini
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 def generate_headlines(number_of_headlines, target_audience, product, temperature):
-    # Llama a la API de Claude para generar titulares
-    message = client.messages.create(
-        model="claude-3-5-sonnet-20240620",
-        max_tokens=1000,  # Configurado a 1000 tokens
-        temperature=temperature,  # Usa el valor del slider aquí
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "text",
-                        "text": f"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, tu tarea es crea {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."
-                    }
-                ]
-            }
-        ]
+    # Crear el prompt para la API de Gemini
+    prompt = (
+        f"Eres un copywriter experto. Tu tarea es crear {number_of_headlines} titulares atractivos "
+        f"para {target_audience} que generen interés en {product}. "
+        "Cada titular debe invitar a la acción, ya sea comprar, registrarse o descargar. "
+        "Utiliza diferentes formatos como preguntas, afirmaciones audaces y hechos intrigantes."
+    )
+
+    # Llamar a la API de Gemini para generar los titulares
+    response = genai.generate_text(
+        model="gemini-1.5-pro",  # Modelo por defecto
+        prompt=prompt,
+        temperature=temperature,
+        max_output_tokens=1000,
     )
-    return message.content
 
-# Configura la interfaz de usuario con Gradio
+    return response.text
+
+# Configurar la interfaz de usuario con Gradio
 def gradio_generate_headlines(number_of_headlines, target_audience, product, temperature):
     return generate_headlines(number_of_headlines, target_audience, product, temperature)
 
-ifaces = gr.Interface(
+iface = gr.Interface(
     fn=gradio_generate_headlines,
     inputs=[
         gr.Dropdown(choices=[str(i) for i in range(1, 11)], label="Número de Titulares", value="5"),
@@ -46,8 +42,8 @@ ifaces = gr.Interface(
     ],
     outputs=gr.Textbox(label="Titulares Generados", lines=10, placeholder="Los titulares aparecerán aquí..."),
     title="Generador de Titulares",
-    description="Usa el poder de Claude AI para crear titulares atractivos. Ajusta los parámetros para generar titulares que capturen la atención de tu audiencia."
+    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
-ifaces.launch()
+iface.launch()