from django.contrib.auth.decorators import login_required
from django.views.decorators.csrf import csrf_exempt
from django.http import JsonResponse
from django.shortcuts import render
import openai
import os

openai.api_key = os.getenv("OPENAI_API_KEY")  # ou use config('OPENAI_API_KEY') se estiver usando python-decouple

@login_required
def home(request):
    return render(request, 'analise/home.html')

@csrf_exempt
@login_required
def chat_assistente_view(request):
    if request.method == 'POST':
        mensagem = request.POST.get('mensagem', '')

        if not mensagem:
            return JsonResponse({'resposta': 'Nenhuma mensagem recebida.'})

        try:
            completion = openai.ChatCompletion.create(
                model="gpt-4",  # ou "gpt-3.5-turbo"
                messages=[
                    {"role": "system", "content": "Você é um engenheiro de tubulações muito técnico, especialista em normas como ASME B31.3, B36.10, e responde com precisão e clareza."},
                    {"role": "user", "content": mensagem}
                ]
            )
            resposta = completion.choices[0].message['content']
        except Exception as e:
            resposta = f"Erro ao consultar IA: {str(e)}"

        return JsonResponse({'resposta': resposta})

    return JsonResponse({'resposta': 'Método não permitido'}, status=405)