from dotenv import load_dotenv import streamlit as st import os import google.generativeai as genai import random # Cargar las variables de entorno load_dotenv() # Configurar la API de Google genai.configure(api_key=os.getenv("GOOGLE_API_KEY")) # Diccionario de ejemplos de bullets bullets_examples = { "1": "El armario del baño es el mejor lugar para guardar medicamentos, ¿verdad? Incorrecto. Es el peor. Los hechos están en la página 10.", "2": "El mejor tiempo verbal que le da a tus clientes la sensación de que ya te han comprado.", "3": "La historia de un joven emprendedor que transformó su vida aplicando esta técnica simple pero poderosa.", "4": "Los misterios de cómo algunas personas parecen tener éxito sin esfuerzo, mientras otras luchan. La clave está en esta pequeña diferencia.", "5": "La leyenda de aquellos que dominaron la productividad con un solo hábito. ¿Te atreves a descubrirlo?", "6": "Un sistema simple para escribir textos sin intentar convencerlos de comprar.", "7": "La verdad que nunca te han contado en la escuela, o en casa, sobre cómo ganarte la vida con la música.", "8": "La historia de un padre ocupado que, con solo 10 minutos al día, logró transformar su salud y bienestar.", "9": "Los misterios de cómo una técnica sencilla te permite reducir el estrés al instante, sin necesidad de dejar tu trabajo o cambiar tu estilo de vida.", "10": "¿Sabías que muchas personas están usando este método y han mejorado su bienestar en solo 7 días?", "11": "¿Cuándo es una buena idea decirle a una chica que te gusta? Si no se lo dices en ese momento, despídete de conocerla íntimamente." } # Función para seleccionar bullets aleatorios def get_random_bullets(num_bullets): # Seleccionar múltiples bullets aleatorios de los ejemplos selected_bullets = random.sample(list(bullets_examples.values()), min(num_bullets, len(bullets_examples))) return selected_bullets # Función para obtener una cantidad de bullets def get_gemini_response_bullets(target_audience, product, num_bullets, temperature): model_choice = "gemini-1.5-flash" # Modelo por defecto # Seleccionar bullets aleatorios usando la nueva función selected_bullets = get_random_bullets(num_bullets) # Configuración del modelo generativo y las instrucciones del sistema model = genai.GenerativeModel( model_name=model_choice, generation_config={ "temperature": temperature, "top_p": 0.85, "top_k": 128, "max_output_tokens": 2048, "response_mime_type": "text/plain", }, system_instruction=( f"You are a world-class copywriter, expert in creating benefits that connect symptoms with problems of {target_audience}. " f"You deeply understand the emotions, desires, and challenges of {target_audience}, allowing you to design personalized copywriting that resonate and motivate action. " "Generate unusual, creative, and fascinating bullets that capture {target_audience}'s attention. Respond in Spanish and use a numbered list format. " f"When responding, always include a heading referencing {target_audience} as follows: 'Aquí hay {num_bullets} bullets para convencer a {target_audience}.'" ) ) # Crear la instrucción para generar bullets chat_session = model.start_chat( history=[ { "role": "user", "parts": [ f"Tu tarea es escribir {num_bullets} bullets que denoten los beneficios que resolverán los problemas de {target_audience}. " "Un buen bullet conecta los síntomas con los problemas enfrentados por {target_audience} de una manera natural, que no se note como manipuladora. " f"Escribe bullets creativos, en un estilo conversacional, que no sean aburridos, sino más bien divertidos. " f"Sé sutil a la hora de crear los bullets para referirte a los beneficios de tu {product}. " f"Usa estos ejemplos como inspiración: {', '.join(selected_bullets)}. " "Por favor, crea los bullets ahora." ], }, ] ) # Crear un mensaje para el modelo que incluye los bullets generados response = model.generate_content(chat_session.history) if response and response.parts: return response.parts[0].text else: raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.") # Inicializar la aplicación Streamlit st.set_page_config(page_title="Generador de Bullets", layout="wide") # Centrar el título y el subtítulo st.markdown("

Impact Bullet Generator

", unsafe_allow_html=True) st.markdown("

Transforma los pensamientos de tu audiencia en balas persuasivas que inspiren a la acción.

", unsafe_allow_html=True) # Añadir CSS personalizado para el botón st.markdown(""" """, unsafe_allow_html=True) # Crear dos columnas para el layout (40% y 60%) col1, col2 = st.columns([2, 3]) with col1: # Campos de entrada target_audience = st.text_input("¿Quién es tu público objetivo?") product = st.text_input("¿Qué producto tienes en mente?") # Campos de personalización sin acordeón num_bullets = st.slider("Número de Bullets", min_value=1, max_value=10, value=5) temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1) # Botón de enviar submit = st.button("Generar Bullets") # Mostrar los bullets generados if submit: if target_audience and product: try: # Obtener la respuesta del modelo generated_bullets = get_gemini_response_bullets(target_audience, product, num_bullets, temperature) col2.markdown(f"""

🧙🏻‍♂️ Mira la magia en acción:

{generated_bullets}
""", unsafe_allow_html=True) except ValueError as e: st.error(str(e)) else: st.error("Por favor, completa todos los campos.")