import streamlit as st
from about import about_ask2democracy
from hallazgos import hallazgos_comision_verdad_2022
from reformas import reforma_salud_febrero_2023
from pinecode_quieries import PinecodeProposalQueries
from config import Config

# Define the sidebar
api_key = st.sidebar.text_input("OpenAI API Key", type="password", 
                                value=st.session_state.get("OPENAI_API_KEY", ""))
if api_key:
    st.session_state["OPENAI_API_KEY"] = api_key


# Define the navigation between pages
page_options = {
    "About": about_ask2democracy,
    "Propuesta: Reforma a la salud febrero de 2023": reforma_salud_febrero_2023,
    "Hallazgos de la comisión de la verdad 2022": hallazgos_comision_verdad_2022
}

# Define the page selector
selected_page = st.sidebar.radio("Selecciona la página que deseas explorar:", list(page_options.keys()))

queries = PinecodeProposalQueries (index_name= Config.index_name,
                                        api_key = Config.es_password,
                                        environment = Config.pinecone_environment,
                                        embedding_dim = Config.embedding_dim,
                                        reader_name_or_path = Config.reader_model_name_or_path,
                                        use_gpu = Config.use_gpu,
                                        OPENAI_key=api_key)
    
# Render the selected page
#print("key: " + api_key)
page_options[selected_page](api_key, queries)