"""
Data Scientist.: Dr.Eddy Giusepe Chirinos Isidro

Neste script como primeira tentativa estudamos e implementamos
a Intenção de Saudação ou Despedida do usuário.
"""
from typing import List
from pydantic import BaseModel
import json

class StepByStepAIResponse(BaseModel):
    title: str
    steps: List[str]

schema = StepByStepAIResponse.model_json_schema()

#print(schema)


#Substitua sua chave de API OpenAI:
import openai
import os
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv()) # read local .env file
openai.api_key  = os.environ['OPENAI_API_KEY']


response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
       {"role": "user", "content": "Bom dia!"}
    ],
    functions=[
        {
          "name": "get_answer_for_user_query",
          "description": "Você deve detectar a INTENÇÃO de Saudação ou INTENÇÃO de Despedida ou Frase Geral na query do usuário:  ```{query}```. \
                          Se a query conter uma intenção de Saudação ou de Despedida você deve responder a essa query. E se você Identificar \
                          uma Frase Geral na query do usuário, você deverá dizer: Fora do Escopo.",
          "parameters": StepByStepAIResponse.model_json_schema()
        }
    ],
    function_call={"name": "get_answer_for_user_query"}
)


output = json.loads(response.choices[0]["message"]["function_call"]["arguments"])
#sbs = StepByStepAIResponse(**output)
#print(sbs.steps)
print(output)