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"))

# Ejemplos de categorías de bullets
directos = [
    "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.",
    "El mejor tiempo verbal que le da a tus clientes la sensación de que ya te han comprado.",
    "La historia de un joven emprendedor que transformó su vida aplicando esta técnica simple pero poderosa."
]

misterios = [
    "Los misterios de cómo algunas personas parecen tener éxito sin esfuerzo, mientras otras luchan. La clave está en esta pequeña diferencia.",
    "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."
]

leyendas = [
    "La leyenda de aquellos que dominaron la productividad con un solo hábito. ¿Te atreves a descubrirlo?",
    "La verdad que nunca te han contado en la escuela, o en casa, sobre cómo ganarte la vida con la música."
]

historias_personales = [
    "La historia de un padre ocupado que, con solo 10 minutos al día, logró transformar su salud y bienestar.",
    "¿Sabías que muchas personas están usando este método y han mejorado su bienestar en solo 7 días?"
]

preguntas_retoricas = [
    "¿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 generar bullets informativos de todas las categorías
def generate_bullets(target_audience, product, call_to_action, number_of_bullets):
    bullets = []
    
    # Lista combinada de todas las categorías
    all_categories = directos + misterios + leyendas + historias_personales + preguntas_retoricas
    
    # Selecciona bullets aleatorios de la lista combinada
    for _ in range(number_of_bullets):
        bullet = random.choice(all_categories)
        bullets.append(bullet)

    return bullets

# Configurar la interfaz de usuario con Streamlit
st.set_page_config(page_title="QuickPrompt", layout="wide")

# Agregar el manual en el sidebar con mejor diseño
st.sidebar.markdown("## **Manual de Usuario para Quick Prompt**")
st.sidebar.write("""
**Bienvenido a Quick Prompt**  
Quick Prompt está diseñado para ayudarte a crear bullets informativos que resalten los beneficios, dolores y curiosidades relevantes para tu audiencia.

### ¿Por qué los bullets son importantes?
Los bullets ayudan a presentar información clave de manera clara y concisa. Aquí te mostramos algunos ejemplos de información que puedes incluir:
- Beneficios de tu producto o servicio
- Problemas comunes que enfrenta tu audiencia
- Datos curiosos que capturan la atención

### ¿Cómo utilizar Quick Prompt?
Sigue estos pasos para sacar el máximo provecho de la herramienta:

1. **Define tu público objetivo**  
   Piensa en quiénes son y qué necesitan. Por ejemplo:
   - Estudiantes universitarios
   - Emprendedores
   - Profesionales en busca de desarrollo personal

2. **Especifica tu producto o servicio**  
   Asegúrate de que el producto esté claro y relacionado con los bullets. Ejemplos:
   - Herramientas de productividad
   - Cursos de marketing digital
   - Recursos de gestión del tiempo

3. **Determina la acción deseada**  
   Define cuál es la acción específica que deseas que realice tu audiencia:
   - Inscribirse en un curso
   - Descargar un recurso gratuito
   - Participar en un evento exclusivo

### Consejos adicionales:
- Asegúrate de que cada bullet sea claro y directo al punto.
- Utiliza datos y ejemplos concretos para hacer tus bullets más impactantes.
""")

# Footer del manual
st.sidebar.write("Con Quick Prompt, comunicarte con claridad y eficacia es más fácil que nunca.")

# Centrar el título y el subtítulo
st.markdown("<h1 style='text-align: center;'>Quick Prompt</h1>", unsafe_allow_html=True)
st.markdown("<h4 style='text-align: center;'>Transforma tu mensaje en bullets informativos que resalten lo esencial para tu audiencia.</h4>", unsafe_allow_html=True)

# Añadir CSS personalizado para el botón
st.markdown("""
    <style>
    div.stButton > button {
        background-color: #FFCC00;
        color: black;
        width: 90%;
        height: 60px;
        font-weight: bold;
        font-size: 22px;
        text-transform: uppercase;
        border: 1px solid #000000;
        border-radius: 8px;
        display: block;
        margin: 0 auto;
    }
    div.stButton > button:hover {
        background-color: #FFD700;
        color: black;
    }
    </style>
""", unsafe_allow_html=True)

# Crear columnas
col1, col2 = st.columns([1, 2])  

# Columnas de entrada
with col1:
    target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Estudiantes Universitarios")
    product = st.text_input("¿Qué producto tienes en mente?", placeholder="Ejemplo: Curso de Inglés")
    call_to_action = st.text_input("¿Qué acción deseas que tomen?", placeholder="Ejemplo: Inscribirse al curso")
    number_of_bullets = st.selectbox("Número de bullets informativos", options=[1, 2, 3, 4, 5], index=2)

    # Botón de enviar
    submit = st.button("Generar Bullets")

# Mostrar los bullets generados
if submit:
    if target_audience and product and call_to_action:
        try:
            # Obtener la respuesta del modelo
            generated_bullets = generate_bullets(target_audience, product, call_to_action, number_of_bullets)
            col2.markdown(f"""
                <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">
                    <h4>Bullets Generados:</h4>
                    <ul>
                        {''.join(f'<li>{bullet}</li>' for bullet in generated_bullets)}
                    </ul>
                </div>
            """, unsafe_allow_html=True)
        except Exception as e:
            col2.error(f"Error inesperado: {str(e)}")
    else:
        col2.warning("Por favor, completa todos los campos.")