#!/usr/bin/env python3
"""
Teste rápido da OpenAI API
"""
import json
import urllib.request
import urllib.error

# Configuração
API_KEY = "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"
API_URL = "https://api.openai.com/v1/chat/completions"

# Payload
payload = {
    "model": "gpt-4o-mini",
    "messages": [
        {"role": "user", "content": "Responda com exatamente 5 palavras: Sistema NEOG está funcionando?"}
    ],
    "max_tokens": 20
}

# Request
headers = {
    "Content-Type": "application/json",
    "Authorization": f"Bearer {API_KEY}"
}

try:
    print("🔄 Testando OpenAI API...")
    print(f"   Modelo: gpt-4o-mini")
    print(f"   URL: {API_URL}")
    print()

    request = urllib.request.Request(
        API_URL,
        data=json.dumps(payload).encode('utf-8'),
        headers=headers,
        method='POST'
    )

    with urllib.request.urlopen(request, timeout=10) as response:
        data = json.loads(response.read().decode('utf-8'))

        if 'choices' in data and len(data['choices']) > 0:
            resposta = data['choices'][0]['message']['content']
            print("✅ API OpenAI funcionando!")
            print(f"   Resposta: {resposta}")
            print()
            print(f"📊 Tokens usados: {data.get('usage', {}).get('total_tokens', 'N/A')}")
            print(f"   Modelo: {data.get('model', 'N/A')}")
            exit(0)
        else:
            print("❌ Resposta inesperada da API")
            print(json.dumps(data, indent=2))
            exit(1)

except urllib.error.HTTPError as e:
    print(f"❌ Erro HTTP {e.code}: {e.reason}")
    print(f"   Detalhes: {e.read().decode('utf-8')}")
    exit(1)
except urllib.error.URLError as e:
    print(f"❌ Erro de conexão: {e.reason}")
    exit(1)
except Exception as e:
    print(f"❌ Erro inesperado: {e}")
    exit(1)