Update app.py

app.py

@@ -1,53 +1,87 @@
-import gradio as gr
-import anthropic
-import os  # Importa el módulo os para manejar variables de entorno
+from dotenv import load_dotenv
+import streamlit as st
+import os
+import google.generativeai as genai
+import random
 
-# Obtén la clave de API desde la variable de entorno
-api_key = os.getenv("ANTHROPIC_API_KEY")
+# Cargar las 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.")
+# Configurar la API de Google
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Configura el cliente de la API de Claude (Anthropic)
-client = anthropic.Anthropic(api_key=api_key)
+# Función para obtener una mención del producto de manera probabilística
+def get_random_product_mention():
+    mentions = ["Directa", "Indirecta", "Metafórica"]
+    probabilities = [0.34, 0.33, 0.33]  
+    return random.choices(mentions, probabilities)[0]
 
+# Crear la instrucción de mención basada en la opción seleccionada
+def get_mention_instruction(product_mention, product):
+    if product_mention == "Directa":
+        return f"Introduce directamente el producto '{product}' como la solución clara al problema que enfrenta el lector."
+    elif product_mention == "Indirecta":
+        return f"Referencia sutilmente el producto '{product}' como una posible solución al problema del lector sin nombrarlo explícitamente."
+    elif product_mention == "Metafórica":
+        return f"Introduce el producto '{product}' usando una metáfora, conectándolo simbólicamente a la solución que necesita el lector."
+    return ""
+
+# Función para generar titulares
 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=[
+    product_mention = get_random_product_mention()
+    mention_instruction = get_mention_instruction(product_mention, product)
+
+    # Crear la configuración del modelo
+    generation_config = {
+        "temperature": temperature,  # Usar el valor del slider aquí
+        "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, 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."
+    )
+
+    chat_session = model.start_chat(
+        history=[
             {
                 "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."
-                    }
-                ]
-            }
+                "parts": [
+                    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}. "
+                    f"Usa la siguiente mención: {mention_instruction}. "
+                    "Los ganchos deben ser 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 message.content
 
-# Configura 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)
+    response = chat_session.send_message("Genera los titulares")  # Enviar mensaje para obtener la respuesta
+    return response.text  # Regresar la respuesta directamente
 
-ifaces = gr.Interface(
-    fn=gradio_generate_headlines,
+# 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.Textbox(label="Titulares Generados", lines=10, placeholder="Los titulares aparecerán aquí..."),
+    outputs=gr.Markdown(label="Titulares Generados"), 
     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()