from dotenv import load_dotenv
import streamlit as st
import os
import google.generativeai as genai
from bullet_formulas import bullet_formulas
from style import styles
from prompts import create_instruction
from angles import bullet_angles

# Cargar las variables de entorno
load_dotenv()

# Configurar la API de Google
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

# Función para generar bullets de beneficios
@st.cache_resource
def get_model(temperature):
    generation_config = {
        "temperature": temperature,
    }
    return genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)

# Update the generate_benefits function to include selected_angle
def generate_benefits(number_of_benefits, target_audience, product, temperature, selected_formula, selected_angle):
    if not target_audience or not product:
        return "Por favor, completa todos los campos requeridos."
    
    model = get_model(temperature)
    benefits_instruction = create_instruction(
        number_of_benefits, 
        target_audience, 
        product, 
        selected_formula,
        selected_angle
    )
    
    response = model.generate_content([benefits_instruction], generation_config={"temperature": temperature})
    return response.parts[0].text if response and response.parts else "Error generating content."

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

# Leer el contenido del archivo manual.md
with open("manual.md", "r", encoding="utf-8") as file:
    manual_content = file.read()

# Mostrar el contenido del manual en el sidebar
st.sidebar.markdown(manual_content)

# Ocultar elementos de la interfaz
st.markdown(styles["main_layout"], unsafe_allow_html=True)

# Centrar el título y el subtítulo
st.markdown("<h1 style='text-align: center;'>Bullet Benefits Generator</h1>", unsafe_allow_html=True)
st.markdown("<h4 style='text-align: center;'>Transforma características en beneficios irresistibles que conectan emocionalmente con tu audiencia.</h4>", unsafe_allow_html=True)

# Añadir CSS personalizado para el botón
st.markdown(styles["button"], unsafe_allow_html=True)

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

# Columnas de entrada
# Inside the column section
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")
    number_of_benefits = st.selectbox("Número de Beneficios", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)

    # Fix indentation - remove extra spaces
    with st.expander("Opciones avanzadas"):
        selected_formula_key = st.selectbox(
            "Tipo de bullet",  # Added label
            options=list(bullet_formulas.keys()),
            label_visibility="visible"
        )
        selected_formula = bullet_formulas[selected_formula_key]

        selected_angle_key = st.selectbox(
            "Ángulo del bullet",  # Added label
            options=list(bullet_angles.keys()),
            label_visibility="visible"
        )
        selected_angle = bullet_angles[selected_angle_key]

        temperature = st.slider(
            "Nivel de creatividad",  # Added label
            min_value=0.0, 
            max_value=2.0, 
            value=1.0, 
            step=0.1,
            label_visibility="visible"
        )

    submit = st.button("GENERAR LOS BULLETS POINTS")

# Mostrar los beneficios generados
# Update the submit section to include selected_angle
if submit:
    if target_audience and product and selected_formula:
        with st.spinner('Generando beneficios...'):
            generated_benefits = generate_benefits(
                number_of_benefits, 
                target_audience, 
                product, 
                temperature, 
                selected_formula,
                selected_angle
            )
            if not isinstance(generated_benefits, str):
                col2.error("Error al generar beneficios")
            else:
                col2.markdown(f"""
                    <div style="{styles['results_container']}">
                        <h3>Beneficios Generados:</h3>
                        <p>{generated_benefits}</p>
                    </div>
                """, unsafe_allow_html=True)

    else:
        col2.warning("Por favor, completa todos los campos antes de generar beneficios.")