import openai
import streamlit as st
from streamlit_chat import message
import os
from dotenv import load_dotenv

from chat_settings import (
    get_initial_message,
    get_chatgpt_response,
    update_chat,
)


# Carga las claves
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")
LLM = "gpt-3.5-turbo-0613" 


# Streamlit Application
def main():
    st.title("ProPilot - OpenAI Demo Function Calling")
    st.markdown(
        """
        Demo of OpenAI function calling using  gpt-3.5-turbo-0613. ProPilot - QuePlan
        """
    )

    if 'messages' not in st.session_state:
        st.session_state['messages'] = get_initial_message()

    query = st.text_input("Ingresa tu texto")

    if st.button("Enviar") and query:
        st.session_state['messages'] = update_chat(st.session_state['messages'], "user", query)
        chatgpt_response = get_chatgpt_response(st.session_state['messages'], LLM)
        st.session_state['messages'] = update_chat(st.session_state['messages'], "assistant", chatgpt_response)

    if st.session_state['messages']:
        for i, msg in enumerate(st.session_state['messages']):
            if msg['role'] == 'user':
                message(msg['content'], is_user=True, key=str(i))
            else:
                message(msg['content'], key=str(i))

if __name__ == "__main__":
    main()