Create app.py

app.py

@@ -0,0 +1,70 @@
+import streamlit as st
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# Cargar modelo y tokenizador (elige un modelo ligero como GPT-NeoX, GPT-J o LLaMA)
+model_name = "EleutherAI/gpt-neo-1.3B"  # Cambiar según el modelo elegido
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Función para generar texto
+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)
+    return text
+
+# Función para generar situaciones y copys según las selecciones del usuario
+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}.
+    La situación cotidiana es: {situation}. Integra el concepto de marca "Tú eliges" en el copy.
+    Genera 5 opciones de copy (titular y bajada):
+    """
+    return generate_text(prompt)
+
+# Streamlit App
+st.title("Herramienta de Generación de Copys Publicitarios")
+
+# Paso 1: Selección de la etapa del funnel
+st.header("1. Selecciona la etapa del funnel")
+stage = st.selectbox("Etapa del Funnel", ["Upper", "Middle", "Lower"])
+
+# Paso 2: Mostrar productos según etapa
+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])
+
+# Paso 3: Propuesta de situaciones cotidianas
+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])
+
+# Paso 4: Generación de copys
+if st.button("Generar Copys"):
+    st.header("4. Copys Generados")
+    generated_copys = generate_situations_and_copys(stage, product, situation)
+    st.write(generated_copys)
+
+# Paso 5: Iteración y personalización (opcional)
+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)