Categorizador.py

from openai import OpenAI
from dotenv import load_dotenv
import os

load_dotenv()
cliente = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))  # Cria um objeto cliente da OpenAI usando a chave da API carregada do arquivo .env # Chama o método de completions.create no cliente para enviar uma série de mensagens e obter uma resposta
modelo = "gpt-4"

def categoriza_produto(nome_produto, lista_categorias_possiveis):
 
    prompt_sistema = f"""  
            Você é um categorizador de produtos.
            Você deve assumir as categorias presentes na lista abaixo.

            # Lista de Categorias Válidas
            {lista_categorias_possiveis.split(",")}

            # Formato da Saída
            Produto: Nome do Produto
            Categoria: apresente a categoria do produto

            # Exemplo de Saída
            Produto: Escova elétrica com recarga solar
            Categoria: Eletrônicos Verdes

            """
    resposta = cliente.chat.completions.create(
        messages=[
            {
                "role": "system",  # Define o papel como "system" para a primeira mensagem
                "content": prompt_sistema, # Prompt: Conteúdo da mensagem do sistema
            },
            {
                "role": "user",  # Define o papel como "usuário" para a segunda mensagem
                "content": nome_produto, # Conteúdo da mensagem do usuário
            }
        ],
        model = modelo,
        temperature = 0,
        max_tokens = 200,
        # Especifica o modelo da OpenAI a ser utilizado, neste caso, gpt-4
    )

    return resposta.choices[0].message.content

categorias_validas = input("Informe as categorias válidas, separadas por vírgula: ")
while True:
    nome_produto = input("Digite o nome do produto: ")
    texto_resposta = categoriza_produto(nome_produto, categorias_validas)
    print(texto_resposta)
    if nome_produto == "stop":
        break