app.py

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

# Cargar las variables de entorno
load_dotenv()

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

# Inicializar variables de estado en session_state si no existen
if 'perfil_cliente' not in st.session_state:
    st.session_state.perfil_cliente = None
if 'producto' not in st.session_state:
    st.session_state.producto = ""
if 'habilidades' not in st.session_state:
    st.session_state.habilidades = ""
if 'creatividad' not in st.session_state:
    st.session_state.creatividad = 1.0

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

def generate_buyer_persona(product, skills, target_audience, temperature):
    if not product or not skills:
        return "Por favor, completa los campos de producto y habilidades."
    
    model = get_model(temperature)
    instruction = create_instruction(
        product_service=product,
        skills=skills,
        target_audience=target_audience
    )
    
    # Añadir instrucción explícita para respuesta en español
    instruction += "\n\nIMPORTANTE: La respuesta debe estar completamente en español."
    
    response = model.generate_content([instruction], generation_config={"temperature": temperature})
    return response.parts[0].text if response and response.parts else "Error generando el perfil de cliente ideal."

# Modificar la función update_profile para que no use spinner
def update_profile():
    # Solo actualizar la variable de sesión
    st.session_state.submitted = True

# Configurar la interfaz de usuario con Streamlit
st.set_page_config(page_title="Generador de Cliente Ideal", page_icon="👤", layout="wide")

# Leer el contenido del archivo manual.md si existe
try:
    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)
except FileNotFoundError:
    st.sidebar.warning("Manual not found. Please create a manual.md file.")
except Exception as e:
    st.sidebar.error(f"Error loading manual: {str(e)}")

# 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;'>Generador de Perfil de Cliente Ideal</h1>", unsafe_allow_html=True)
st.markdown("<h4 style='text-align: center;'>Crea un perfil detallado de tu cliente ideal basado en tu producto y habilidades.</h4>", unsafe_allow_html=True)

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

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

# Columna de entrada
with col1:
    producto = st.text_input("¿Qué producto o servicio ofreces?", 
                            value=st.session_state.producto,
                            placeholder="Ejemplo: Curso de Inglés",
                            key="producto_input")
    st.session_state.producto = producto
    
    habilidades = st.text_input("¿Cuáles son tus habilidades principales?", 
                               value=st.session_state.habilidades,
                               placeholder="Ejemplo: Enseñanza, comunicación, diseño de contenidos",
                               key="habilidades_input")
    st.session_state.habilidades = habilidades
    
    # Crear un acordeón para las opciones de personalización
    with st.expander("Personaliza Tu Cliente Ideal Soñado"):
        # Nuevo campo para público objetivo
        if 'publico_objetivo' not in st.session_state:
            st.session_state.publico_objetivo = ""
        
        publico_objetivo = st.text_input("¿Cuál es tu público objetivo? (opcional)", 
                                        value=st.session_state.publico_objetivo,
                                        placeholder="Ejemplo: Profesionales entre 25-40 años interesados en desarrollo personal",
                                        key="publico_objetivo_input")
        st.session_state.publico_objetivo = publico_objetivo
        
        # Nivel de creatividad con slider
        creatividad = st.slider("Nivel de creatividad", 
                               min_value=0.0, 
                               max_value=2.0, 
                               value=st.session_state.creatividad, 
                               step=0.1,
                               key="creatividad_slider")
        st.session_state.creatividad = creatividad
    
    # Botón para generar
    submit = st.button("GENERAR PERFIL DE CLIENTE IDEAL", on_click=update_profile)

# Columna de resultados
with col2:
    # Verificar si se ha enviado el formulario
    if 'submitted' in st.session_state and st.session_state.submitted:
        if st.session_state.producto and st.session_state.habilidades:
            with st.spinner("Creando tu Cliente Ideal Soñado..."):
                # Generar el perfil del cliente
                perfil_cliente = generate_buyer_persona(
                    st.session_state.producto,
                    st.session_state.habilidades,
                    st.session_state.publico_objetivo,
                    st.session_state.creatividad
                )
                # Guardar en session_state
                st.session_state.perfil_cliente = perfil_cliente
                # Resetear el estado de envío
                st.session_state.submitted = False
                
            # Mostrar resultados
            if not isinstance(st.session_state.perfil_cliente, str):
                st.error("Error al generar el perfil de cliente ideal")
            else:
                st.markdown(f"""
                    <div style="{styles['results_container']}">
                        <h3>Tu Cliente Ideal</h3>
                        {st.session_state.perfil_cliente}
                    </div>
                """, unsafe_allow_html=True)
                
                # Opción para descargar
                st.download_button(
                    label="Descargar Perfil",
                    data=st.session_state.perfil_cliente,
                    file_name="cliente_ideal.pdf",
                    mime="text/markdown"
                )
        else:
            st.warning("Por favor, completa los campos de producto y habilidades antes de generar el perfil.")
    # Mostrar resultados anteriores si existen
    elif st.session_state.perfil_cliente:
        st.markdown(f"""
            <div style="{styles['results_container']}">
                <h3>Tu Cliente Ideal</h3>
                {st.session_state.perfil_cliente}
            </div>
        """, unsafe_allow_html=True)
        
        # Opción para descargar
        st.download_button(
            label="Descargar Perfil",
            data=st.session_state.perfil_cliente,
            file_name="cliente_ideal.md",
            mime="text/markdown"
        )