Update app.py

app.py

@@ -12,12 +12,12 @@ if not api_key:
 # Configura el cliente de la API de Claude (Anthropic)
 client = anthropic.Anthropic(api_key=api_key)
 
-def generate_headlines(number_of_headlines, target_audience, product):
+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,
-        temperature=0,
+        max_tokens=2500,
+        temperature=temperature,
         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=[
             {
@@ -34,8 +34,8 @@ def generate_headlines(number_of_headlines, target_audience, product):
     return message.content
 
 # 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)
+def gradio_generate_headlines(number_of_headlines, target_audience, product, temperature):
+    return generate_headlines(number_of_headlines, target_audience, product, temperature)
 
 # Define los colores de la interfaz según el logo de Anthropic (ejemplo)
 logo_colors = {
@@ -44,26 +44,10 @@ logo_colors = {
     "text_color": "#212529"
 }
 
-# Configura la interfaz de usuario
 with gr.Blocks(css=f"""
-    .gradio-container {{
-        background-color: {logo_colors["background"]};
-    }}
-    .gradio-button {{
-        margin-top: 10px;
-        background-color: {logo_colors["primary"]};
-        color: white;
-        border: none;
-        border-radius: 4px;
-        padding: 10px;
-    }}
-    .gradio-button:hover {{
-        background-color: #1b5bb5;
-    }}
-    .result-text {{
-        white-space: pre-line;
-        text-align: left;
-    }}
+    .gradio-container {{ background-color: {logo_colors["background"]}; }}
+    .headline-output {{ text-align: left; white-space: pre-line; }}
+    .gradio-slider .slider-container {{ margin: 20px 0; }}
 """) as demo:
     gr.Markdown(
         f"""
@@ -73,19 +57,30 @@ with gr.Blocks(css=f"""
     )
     
     with gr.Row():
-        with gr.Column(scale=2):
-            number_of_headlines = gr.Number(label="Número de Titulares", value=5)
+        with gr.Column(scale=1):
+            number_of_headlines = gr.Dropdown(
+                label="Número de Titulares",
+                choices=[i for i in range(1, 11)],
+                value=5
+            )
             target_audience = gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios")
             product = gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés")
-        
+            temperature = gr.Slider(
+                minimum=0,
+                maximum=1,
+                value=0,
+                step=0.1,
+                label="Temperature",
+                interactive=True
+            )
         with gr.Column(scale=1):
-            submit_btn = gr.Button("Generar Titulares")
+            submit_btn = gr.Button("Generar Titulares", elem_id="submit-btn")
 
-    output = gr.Textbox(label="Titulares Generados", lines=10, elem_id="result-text")
+    output = gr.Textbox(label="Titulares Generados", lines=10, css_class="headline-output")
 
     submit_btn.click(
         fn=gradio_generate_headlines,
-        inputs=[number_of_headlines, target_audience, product],
+        inputs=[number_of_headlines, target_audience, product, temperature],
         outputs=output
     )