open_ia.py

import streamlit as st
import openai
import dotenv as env
import os

env.load_dotenv()
openai_key = os.getenv('OPENAI_API_KEY')


def retorna_resposta_modelo(mensagens,
                            openai_key,
                            modelo='gpt-3.5-turbo',
                            temperatura=0,
                            stream=False):
    openai.api_key = openai_key 
    response = openai.ChatCompletion.create(
        model=modelo,
        messages=mensagens,
        temperature=temperatura,
        stream=stream
    )
    return response

def pagina_principal():

    if not 'mensagens' in st.session_state:
        st.session_state.mensagens = []
    
    mensagens = st.session_state['mensagens']

    st.header('Chatbot do Marcus', divider=True)