import os
from dotenv import load_dotenv
from openai import OpenAI

# Carrega as variáveis do ficheiro .env
load_dotenv()

OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

if not OPENAI_API_KEY:
    print("ERRO: A variável OPENAI_API_KEY não foi encontrada no ficheiro .env!")
else:
    print("Chave de API encontrada. A tentar contactar a OpenAI...")
    try:
        client = OpenAI(api_key=OPENAI_API_KEY)

        resp = client.chat.completions.create(
            model="gpt-4o-mini", # Usamos um modelo simples para o teste
            messages=[{"role": "user", "content": "Olá! Responda apenas com 'OK' se estiveres a funcionar."}],
            temperature=0
        )

        message_content = resp.choices[0].message.content
        print(f"SUCESSO! A OpenAI respondeu: '{message_content}'")

    except Exception as e:
        print(f"FALHA: Ocorreu um erro ao comunicar com a API da OpenAI.")
        print(f"Detalhes do Erro: {e}")