Update app.py

app.py

@@ -2,22 +2,18 @@ import warnings
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
-# Ignore specific warnings
 warnings.filterwarnings("ignore", category=FutureWarning, module="transformers.tokenization_utils_base")
 
-# Load model and tokenizer
-model_name = "EleutherAI/gpt-neo-1.3B"  # Use a lightweight model
+model_name = "EleutherAI/gpt-neo-1.3B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Function to generate text
 def generate_text(prompt, max_length=100):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(inputs.input_ids, max_length=max_length, do_sample=True, top_p=0.95, temperature=0.7)
     text = tokenizer.decode(outputs[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
     return text
 
-# Function to generate situations and copy according to user selections
 def generate_situations_and_copys(stage, product, situation):
     prompt = f"""
     Estás creando una campaña publicitaria para la etapa {stage} del funnel de marketing. El producto es: {product}.
@@ -26,12 +22,6 @@ def generate_situations_and_copys(stage, product, situation):
     """
     return generate_text(prompt)
 
-# Gradio Interface
-def main(stage, product, situation):
-    generated_copys = generate_situations_and_copys(stage, product, situation)
-    return generated_copys
-
-# Dropdown options for Gradio
 stages = ["Upper", "Middle", "Lower"]
 products = {
     "Upper": ["Convierte tu teléfono en un POS", "Más de 700 plantillas web con Wix"],
@@ -53,33 +43,37 @@ situations = {
     ]
 }
 
-def update_products(stage):
-    return gr.Dropdown.update(choices=products[stage])
+def get_products(stage):
+    return products[stage]
+
+def get_situations(stage):
+    return situations[stage]
 
-def update_situations(stage):
-    return gr.Dropdown.update(choices=situations[stage])
+def main(stage):
+    product_choices = get_products(stage)
+    situation_choices = get_situations(stage)
+    
+    # Return default selections for product and situation
+    return gr.Dropdown.update(choices=product_choices, value=product_choices[0]), gr.Dropdown.update(choices=situation_choices, value=situation_choices[0])
 
-# Gradio App Interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Herramienta de Generación de Copys Publicitarios")
+    gr.Markdown("# Herramienta de Generación de Copies Publicitarios")
 
     with gr.Row():
-        stage_input = gr.Dropdown(choices=stages, label="Etapa del Funnel", interactive=True)
-        product_input = gr.Dropdown(choices=[], label="Producto", interactive=True)
-        situation_input = gr.Dropdown(choices=[], label="Situación", interactive=True)
+        stage_input = gr.Dropdown(choices=stages, label="Etapa del Funnel", value="Upper")
+        product_input = gr.Dropdown(choices=[], label="Producto")
+        situation_input = gr.Dropdown(choices=[], label="Situación")
 
-    stage_input.change(fn=update_products, inputs=stage_input, outputs=product_input)
-    stage_input.change(fn=update_situations, inputs=stage_input, outputs=situation_input)
-    
-    generate_button = gr.Button("Generar Copys")
-    output_text = gr.Textbox(label="Copys Generados")
+    stage_input.change(main, inputs=[stage_input], outputs=[product_input, situation_input])
+
+    generate_button = gr.Button("Generar Copies")
+    output_text = gr.Textbox(label="Copies Generados")
 
-    generate_button.click(main, inputs=[stage_input, product_input, situation_input], outputs=output_text)
+    generate_button.click(generate_situations_and_copys, inputs=[stage_input, product_input, situation_input], outputs=output_text)
 
-    # Optional Custom Prompt Section
     custom_prompt_input = gr.Textbox(label="Ingresa un prompt para personalizar el copy", value="")
-    generate_custom_button = gr.Button("Generar Copys Personalizados")
-    custom_output_text = gr.Textbox(label="Copys Personalizados")
+    generate_custom_button = gr.Button("Generar Copies Personalizados")
+    custom_output_text = gr.Textbox(label="Copies Personalizados")
 
     generate_custom_button.click(fn=generate_text, inputs=custom_prompt_input, outputs=custom_output_text)