Update app.py

app.py

@@ -1,8 +1,9 @@
 import warnings
-warnings.filterwarnings("ignore", category=FutureWarning, module="transformers.tokenization_utils_base")
-
-import streamlit as st
 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
@@ -25,54 +26,61 @@ def generate_situations_and_copys(stage, product, situation):
     """
     return generate_text(prompt)
 
-# Streamlit App
-def main():
-    st.title("Herramienta de Generación de Copys Publicitarios")
+# 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"],
+    "Middle": ["Elige el modelo de dispositivo POS", "Entrega express en Lima"],
+    "Lower": ["Servicios de cobro inmediato", "Cargas Ilimitadas de productos a Wix"]
+}
+situations = {
+    "Upper": [
+        "No elegiste dónde naciste, pero sí cómo gestionar tus ventas.",
+        "No pudiste elegir el nombre que te pusieron, pero sí puedes elegir que tu teléfono sea tu POS."
+    ],
+    "Middle": [
+        "No elegiste todas las decisiones difíciles, pero sí puedes elegir la herramienta perfecta para cada transacción.",
+        "No puedes elegir todas las circunstancias, pero sí el modelo de POS adecuado para tu negocio."
+    ],
+    "Lower": [
+        "No elegiste que el cliente pagara tarde, pero sí cómo cobrar rápido.",
+        "No puedes elegir el ritmo de las ventas, pero sí cómo acelerar los cobros."
+    ]
+}
+
+def update_products(stage):
+    return gr.Dropdown.update(choices=products[stage])
+
+def update_situations(stage):
+    return gr.Dropdown.update(choices=situations[stage])
 
-    # Step 1: Select the funnel stage
-    st.header("1. Selecciona la etapa del funnel")
-    stage = st.selectbox("Etapa del Funnel", ["Upper", "Middle", "Lower"])
+# Gradio App Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Herramienta de Generación de Copys Publicitarios")
 
-    # Step 2: Show products according to the stage
-    st.header("2. Elige un producto")
-    products = {
-        "Upper": ["Convierte tu teléfono en un POS", "Más de 700 plantillas web con Wix"],
-        "Middle": ["Elige el modelo de dispositivo POS", "Entrega express en Lima"],
-        "Lower": ["Servicios de cobro inmediato", "Cargas Ilimitadas de productos a Wix"]
-    }
-    product = st.selectbox("Producto", products[stage])
+    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)
 
-    # Step 3: Propose everyday situations
-    st.header("3. Elige una situación cotidiana")
-    situations = {
-        "Upper": [
-            "No elegiste dónde naciste, pero sí cómo gestionar tus ventas.",
-            "No pudiste elegir el nombre que te pusieron, pero sí puedes elegir que tu teléfono sea tu POS."
-        ],
-        "Middle": [
-            "No elegiste todas las decisiones difíciles, pero sí puedes elegir la herramienta perfecta para cada transacción.",
-            "No puedes elegir todas las circunstancias, pero sí el modelo de POS adecuado para tu negocio."
-        ],
-        "Lower": [
-            "No elegiste que el cliente pagara tarde, pero sí cómo cobrar rápido.",
-            "No puedes elegir el ritmo de las ventas, pero sí cómo acelerar los cobros."
-        ]
-    }
-    situation = st.selectbox("Situación", situations[stage])
+    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")
 
-    # Step 4: Generate copies
-    if st.button("Generar Copys"):
-        st.header("4. Copys Generados")
-        generated_copys = generate_situations_and_copys(stage, product, situation)
-        st.write(generated_copys)
+    generate_button.click(main, inputs=[stage_input, product_input, situation_input], outputs=output_text)
 
-    # Step 5: Iterate and personalize (optional)
-    st.header("5. Personaliza tu Copy")
-    custom_prompt = st.text_area("Ingresa un prompt para personalizar el copy", value=f"{stage}, {product}, {situation}")
-    if st.button("Generar Copys Personalizados"):
-        custom_copys = generate_text(custom_prompt)
-        st.write(custom_copys)
+    # 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")
 
-if __name__ == "__main__":
-    main()
+    generate_custom_button.click(fn=generate_text, inputs=custom_prompt_input, outputs=custom_output_text)
 
+demo.launch()