import json

#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']


def Greetings_or_Goodbyes(query):
    
    response = openai.ChatCompletion.create(
            model='gpt-3.5-turbo-0613',
            temperature=0.0,
            messages=[
            {
                "role": "user",
                "content": f"""
                               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ê deverá responder a essas queries. E se você Identificar \
                               uma Frase Geral da query do usuário, você deverá dizer: Fora do Escopo.                
                            """,
            }
            ],
            functions=[
                {
                    "name": "get_answer_for_user_query",
                    "description": "Responde apenas em português à intenção (query) do usuário, ou seja, tom de SAUDAÇÃO ou tom DESPEDIDA",
                    "parameters": {
                        "type": "object",
                        "properties": {
                            "INTENTION": {
                                "type": "string",
                                "description": "Detecta a INTENÇÃO na query do usuário"
                            },
                            "ANSWER_INTENTION": {
                                "type": "string",
                                "description": "Responder de maneira coerente à query do usuário"
                            }
                            },
                        "required": ["INTENTION", "ANSWER_INTENTION"]
                                  }
                }
            ],
            function_call={"name": "get_answer_for_user_query"}
        )


    output = json.loads(response.choices[0]["message"]["function_call"]["arguments"])

    return output




if __name__ == "__main__":
    while True:
        query = input("User: " )
        resposta = Greetings_or_Goodbyes(query=query)
        #print(resposta['ANSWER_INTENTION'])
        print(resposta)

        
        if not query:
            break