from dotenv import load_dotenv
import streamlit as st
from cases.optimize_content import display_content_optimization
from cases.content_generation import display_content_generation

load_dotenv()

# Movemos la asignación de la clave API a un lugar más abajo en el código.

def display_home(api_key, model):
    st.title("Bienvenido a la página de inicio")
    st.markdown('''
        En esta aplicación, exploraremos los siguientes casos de uso de SEO potenciado por IA:
        1. **Generación de Contenido**: Las herramientas de AI pueden generar tu contenido, como artículos, newsletters o posts para RRSS, de modo que sea más amigable para SEO.
        2. **Optimización de Contenido**: Las herramientas de AI pueden sugerir mejoras a tu contenido, como cambios en la redacción o la ubicación de palabras clave, para que sea más amigable para SEO. También pueden analizar el rendimiento SEO de tu contenido existente y sugerir mejoras.
        Para continuar, selecciona el caso de uso en la barra lateral.
        ''')


MENU_FUNCTIONS = {
    'Home': display_home,
    'Generación de Contenido': display_content_generation,
    'Optimización de Contenido': display_content_optimization,
}

def main():
    st.sidebar.title("Menú")
    selection = st.sidebar.radio("SEO Powered By AI", list(MENU_FUNCTIONS.keys()))

    st.sidebar.markdown("Por favor selecciona el modelo de AI que prefieres utilizar:")
    model = st.sidebar.selectbox('Modelo de AI', ('gpt-3.5-turbo', 'gpt-3.5-turbo-0613', 'gpt-4'))

    st.sidebar.markdown("Por favor ingresa tu API Key:")
    api_key = st.sidebar.text_input("API Key", type='password')

    menu_function = MENU_FUNCTIONS[selection]
    menu_function(api_key, model)

if __name__ == '__main__':
    main()