Update app.py

app.py

@@ -2,18 +2,22 @@ import warnings
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
+# Ignore specific warnings
 warnings.filterwarnings("ignore", category=FutureWarning, module="transformers.tokenization_utils_base")
 
-model_name = "EleutherAI/gpt-neo-1.3B"
+# Load model and tokenizer
+model_name = "EleutherAI/gpt-neo-1.3B"  # Use a lightweight model
 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}.
@@ -22,6 +26,7 @@ def generate_situations_and_copys(stage, product, situation):
     """
     return generate_text(prompt)
 
+# 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"],
@@ -43,37 +48,29 @@ situations = {
     ]
 }
 
-def get_products(stage):
-    return products[stage]
-
-def get_situations(stage):
-    return 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])
+def update_dropdowns(stage):
+    product_choices = products[stage]
+    situation_choices = situations[stage]
+    return gr.Dropdown.update(choices=product_choices), gr.Dropdown.update(choices=situation_choices)
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Herramienta de Generación de Copies Publicitarios")
+    gr.Markdown("# Herramienta de Generación de Copys Publicitarios")
 
     with gr.Row():
         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")
+        product_input = gr.Dropdown(choices=products["Upper"], label="Producto")
+        situation_input = gr.Dropdown(choices=situations["Upper"], label="Situación")
 
-    stage_input.change(main, inputs=[stage_input], outputs=[product_input, situation_input])
+    stage_input.change(fn=update_dropdowns, inputs=stage_input, outputs=[product_input, situation_input])
 
-    generate_button = gr.Button("Generar Copies")
-    output_text = gr.Textbox(label="Copies Generados")
+    generate_button = gr.Button("Generar Copys")
+    output_text = gr.Textbox(label="Copys Generados")
 
-    generate_button.click(generate_situations_and_copys, inputs=[stage_input, product_input, situation_input], outputs=output_text)
+    generate_button.click(fn=generate_situations_and_copys, inputs=[stage_input, product_input, situation_input], outputs=output_text)
 
     custom_prompt_input = gr.Textbox(label="Ingresa un prompt para personalizar el copy", value="")
-    generate_custom_button = gr.Button("Generar Copies Personalizados")
-    custom_output_text = gr.Textbox(label="Copies Personalizados")
+    generate_custom_button = gr.Button("Generar Copys Personalizados")
+    custom_output_text = gr.Textbox(label="Copys Personalizados")
 
     generate_custom_button.click(fn=generate_text, inputs=custom_prompt_input, outputs=custom_output_text)