main.py

# titulo
# input do chat
# a cada mensagem enviada:
    # mostrar a mensagem que o usuario enviou no chat
    # enviar essa mensagem para a IA responder
    # aparece na tela a resposta da IA

# streamlit - frontend e backend
# python3 -m pip install openai streamlit
# python -m pip install openai streamlit

# Roda o projeto: streamlit run main.py

import streamlit as st
from openai import OpenAI
from dotenv import load_dotenv
import os

# 🔑 Carrega variáveis do .env
load_dotenv()
api_key = os.getenv("OPENAI_API_KEY")

# Validação de segurança
if not api_key or not api_key.startswith("sk-"):
    st.error("🔑 API Key inválida ou não configurada!")
    st.info("Crie o arquivo `.env` com: `OPENAI_API_KEY=sk-proj-...`")
    st.stop()

modelo = OpenAI(api_key=api_key)

st.write("### ChatBot com IA")

if "lista_mensagens" not in st.session_state:
    # st.session_state["lista_mensagens"] = [{"role": "system", "content": "Adicionar as instruções iniciais no do seu sistema para a ia se basear"}]
    st.session_state["lista_mensagens"] = []

# session_state = memoria do streamlit
for msg in st.session_state["lista_mensagens"]:
    st.chat_message(msg["role"]).write(msg["content"])

if mensagem_usuario := st.chat_input("Escreva sua mensagem aqui"):
    # user -> ser humano
    # assistant -> inteligencia artificial
    st.chat_message("user").write(mensagem_usuario)
    st.session_state["lista_mensagens"].append({"role": "user", "content": mensagem_usuario})

    try:
        with st.spinner("🤔 Pensando..."):
            # Para criar um agente com base os dados na minha empresa precisa 
            # resposta da IA
            resposta = modelo.chat.completions.create(
                messages=st.session_state["lista_mensagens"],
                model="gpt-4o"
            )
        resposta_ia = resposta.choices[0].message.content

        # exibir a resposta da IA na tela
        st.chat_message("assistant").write(resposta_ia)
        st.session_state["lista_mensagens"].append({"role": "assistant", "content": resposta_ia})
        
    except Exception as e:
        st.error(f"❌ Erro: {str(e)}")
        if "401" in str(e):
            st.warning("💡 Chave inválida! Revogue e gere uma nova em: https://platform.openai.com/api-keys")