import json
import logging
import time

import openai

chat_messages = {}


def chatgpt_recognise(text, uuid):
    if len(chat_messages) == 0:
        chatgpt_init(uuid)

    chat_messages[uuid].append({"role": "user", "content": text})

    # Call ChatGPT API
    start_time = time.time()
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=chat_messages[uuid],
    )
    end_time = time.time()
    logging.info("GPT-3 response in " + str(end_time - start_time) + " seconds")

    # Check if the response is a "valid" JSON
    try:
        response = json.loads(str(response.choices[0].message.content))
        if 'action' in response:
            chat_messages[uuid].append({"role": "assistant", "content": get_answer_from_response(response)})
            return response

    except Exception as e:
        logging.error("Error while parsing ChatGPT response, probably not JSON: " + str(response.choices))
        logging.error(str(e))
        return {"action": "answer", "answer": get_answer_from_response(response)}


def clear_chat(uuid):
    logging.info("Cleared chat for uuid " + uuid)
    chat_messages[uuid] = []


def get_answer_from_response(response):
    if 'action' not in response:
        # Fix for when it responds in plaintext to follow-up questions
        # In that case the response is an OpenAIObject not a JSON
        return response.choices[0].message.content
    else:
        if response['action'] == 'clarify':
            return response['question']
        elif response['action'] == 'command':
            return response['comment']
        elif response['action'] == 'query':
            return response['device_description']
        elif response['action'] == 'answer':
            return response['answer']
        else:
            return "I don't know how to respond to that..."


def chatgpt_init(uuid):
    prompt = open("utils/chatgpt_prompt_2_smaller.txt", "r").read()

    prompt.replace("{{timestamp}}", time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
    prompt.replace("{{location}}", "Lausanne in the canton Vaud of Switzerland")

    chat_messages[uuid] = []
    chat_messages[uuid].append({"role": "system", "content": prompt})