Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import gradio as gr
 import anthropic
 import os  # Importa el módulo os para manejar variables de entorno
@@ -5,72 +6,67 @@ import os  # Importa el módulo os para manejar variables de entorno
 # Obtén la clave de API desde la variable de entorno
 api_key = os.getenv("ANTHROPIC_API_KEY")
 
-# Configura el cliente de Anthropic
+# 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)
 
-# Define la función que genera los titulares
 def generate_headlines(number_of_headlines, target_audience, product):
-    # Configura el mensaje para la API
-    system_message = (
-        "You are a world-class copywriter with expertise in creating compelling hooks, headlines, and subject lines. "
-        "Your talent lies in understanding the emotions, desires, and challenges of a specific audience, allowing you to craft "
-        "personalized marketing strategies that resonate and motivate action. Use proven structures to attract your target audience, "
-        "generating interest and achieving a powerful connection that drives desired results in advertising and content campaigns. "
-        "Respond in Markdown format."
-    )
-    
-    user_message = (
-        f"Generate {number_of_headlines} attention-grabbing headlines designed for {target_audience} to generate interest in {product}. "
-        "Each headline should be crafted to encourage a specific action, such as making a purchase, signing up, or downloading. "
-        "Use a variety of formats (questions, bold statements, intriguing facts) to test different approaches."
-    )
-    
-    # Llama a la API de Claude
-    response = client.messages.create(
+    # Llama a la API de Claude para generar titulares
+    message = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=1000,
         temperature=0,
-        system=system_message,
+        system="You are a world-class copywriter, with experience in creating hooks, headlines, and subject lines that immediately capture attention. Your skill lies in deeply understanding the emotions, desires, and challenges of a specific audience, allowing you to design personalized marketing strategies that resonate and motivate action. You know how to use proven structures to attract your target audience, generating interest and achieving a powerful connection that drives desired results in advertising and content campaigns.\n\nAnswer in Spanish.",
         messages=[
             {
                 "role": "user",
                 "content": [
                     {
                         "type": "text",
-                        "text": user_message
+                        "text": f"Generate {number_of_headlines} attention-grabbing headlines designed for {target_audience} to generate interest in {product}. Each headline should be crafted to encourage a specific action, such as making a purchase, signing up, or downloading. Use a variety of formats (questions, bold statements, intriguing facts) to test different approaches."
                     }
                 ]
             }
         ]
     )
-    
-    # Extrae y retorna el contenido de la respuesta
-    headlines = response.content
-    return headlines
+    return message.content
 
-# Configura la interfaz de usuario de Gradio
-with gr.Blocks() as demo:
+# Configura la interfaz de usuario con Gradio
+def gradio_generate_headlines(number_of_headlines, target_audience, product):
+    return generate_headlines(number_of_headlines, target_audience, product)
+
+# Define los colores de la interfaz según el logo de Anthropic (ejemplo)
+logo_colors = {
+    "background": "#f8f8f8",
+    "primary": "#2c7be5",
+    "text_color": "#212529"
+}
+
+with gr.Blocks(css=".gradio-container { background-color: " + logo_colors["background"] + "; }") as demo:
     gr.Markdown(
-        """
-        <div style="text-align: center; font-size: 32px; font-weight: bold; margin-bottom: 20px;">
-            Headline Generator
-        </div>
-        <div style="text-align: center;">
-            <p>Generate compelling headlines for your marketing needs.</p>
-        </div>
+        f"""
+        <h1 style="color: {logo_colors['primary']}; text-align: center;">Generador de Titulares</h1>
+        <p style="color: {logo_colors['text_color']}; text-align: center;">Usa el poder de Claude AI para crear titulares atractivos</p>
         """
     )
     
-    number_of_headlines = gr.Slider(minimum=1, maximum=10, value=5, label="Number of Headlines")
-    target_audience = gr.Textbox(label="Target Audience", placeholder="e.g., tech enthusiasts, busy professionals")
-    product = gr.Textbox(label="Product", placeholder="e.g., new smartphone, productivity tool")
-    generate_button = gr.Button("Generate Headlines")
-    output = gr.Markdown()
+    with gr.Row():
+        with gr.Column():
+            number_of_headlines = gr.Number(label="Número de Titulares", placeholder="Ejemplo: 5")
+            target_audience = gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios")
+            product = gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés")
+            submit_btn = gr.Button("Generar Titulares", elem_id="submit-btn")
     
-    generate_button.click(
-        generate_headlines,
+    output = gr.Textbox(label="Titulares Generados", lines=40)
+
+    submit_btn.click(
+        fn=gradio_generate_headlines,
         inputs=[number_of_headlines, target_audience, product],
         outputs=output
     )
-    
+
+# Lanza la interfaz
 demo.launch()