import os
from openai import OpenAI
import decider_questions

YES = True
NO = False

g_decider_utils_dbg_printing = False

openai_client = OpenAI(organization=os.environ.get("OPENAI_ORGANIZATION"),
    api_key=os.environ.get("OPENAI_KEY"))


def yesno(question, text, default):
    global g_decider_utils_dbg_printing
    global openai_client

    prompt = text + "\n\n" + question

    if g_decider_utils_dbg_printing:
        print(prompt)

    hopefully_word_yes_or_no = openai_client.chat.completions.create(
        model="gpt-4-turbo",
        messages=[{"role": "system", "content": prompt}],
        temperature=0,
        max_tokens=20,  # A while ago when testing with gpt-4-turbo, at first I tried max_tokens = 1 or 2,  but the davinci-002 model produced zero output (immediate stop) unless I increased max_token to around 20
        frequency_penalty=0,
        presence_penalty=0,
        n=1,
    ).choices[0].message.content

    if g_decider_utils_dbg_printing:
        print(hopefully_word_yes_or_no)

    hopefully_word_yes_or_no = hopefully_word_yes_or_no.upper().strip()

    result = default

    if default == YES:
        if hopefully_word_yes_or_no.startswith("N"):
            result = NO

    if default == NO:
        if hopefully_word_yes_or_no.startswith("Y"):
            result = YES

    return result


def number(question, text, default=-1, maximum=6):
    global g_decider_utils_dbg_printing
    global openai_client

    prompt = text + "\n\n" + question

    if g_decider_utils_dbg_printing:
        print(prompt)

    hopefully_number = openai_client.chat.completions.create(
        model="gpt-4-turbo",
        messages=[{"role": "system", "content": prompt}],
        temperature=0,
        max_tokens=20,  # A while ago when testing with gpt-4-turbo, at first I tried max_tokens = 1 or 2,  but the davinci-002 model produced zero output (immediate stop) unless I increased max_token to around 20
        frequency_penalty=0,
        presence_penalty=0,
        n=1,
    ).choices[0].message.content

    if g_decider_utils_dbg_printing:
        print(hopefully_number)

    hopefully_number = hopefully_number.upper().strip().split(" ")[0].strip(".")

    if g_decider_utils_dbg_printing:
        print(hopefully_number)

    if hopefully_number == "ONE":
        hopefully_number = "1"

    if hopefully_number == "TWO":
        hopefully_number = "2"

    if hopefully_number == "THREE":
        hopefully_number = "3"

    if hopefully_number == "FOUR":
        hopefully_number = "4"

    if hopefully_number == "FIVE":
        hopefully_number = "5"

    if hopefully_number == "SIX":
        hopefully_number = "6"

    result = default

    if hopefully_number.startswith("ALL"):
        result = maximum
    else:

        try:
            if hopefully_number.isnumeric():
                result = int(hopefully_number)
        except:
            pass

    return result


# In certain cases, I need more special-case logic in order to behave correctly,
# which we verify using the unit tests in run_unit_tests.py:


def special_case_is_running_away(text):
    might_really_be_fleeing = False
    for keyword in ["run", "away", "hide", "escape", "flee", "sprint", "teleport"]:
        if keyword in text.lower():
            might_really_be_fleeing = True
            break

    if might_really_be_fleeing:
        return yesno(decider_questions.QUESTION_IS_ACTION_RUNNING_AWAY, text, default=NO)
    else:
        return NO


def special_case_is_magic(text):
    is_magic = False
    for keyword in ["magic", "spell", "fly", "invisib", "levitat", "shapeshift", "morph", "shrink", "transform", "teleport", "dragon", "genie", "fairy", "demon", "devil", "angel", "griffin", "wand"]:
        if keyword in text.lower():
            is_magic = True
            break

    if is_magic:
        return YES
    else:
        return yesno(decider_questions.QUESTION_IS_ACTION_MAGIC, text, default=NO)


def special_case_is_action_lethal(text):
    if len(text.strip()) <= 3:
        return NO

    text_l = text.lower()

    is_negotiation = False
    for keyword in ["say", "ask", "negotiat", "warn", "fair", "consider", "please", "family", "children", "challenge", "request", "inquire", "price", "pay", "gold", "coin", "away", "flee", "don't shoot", "dont shoot"]:
        if keyword in text_l:
            is_negotiation = True
            break

    if is_negotiation:
        return NO

    aiming_but_not_shooting = False
    if ("aim" in text_l or "cock" in text_l or "pull" in text_l or "hammer" in text_l or "point" in text_l) and not ("shoot" in text_l or "fire" in text_l or "trigger" in text_l):
        aiming_but_not_shooting = True

    if aiming_but_not_shooting:
        return NO

    bool1 = yesno(decider_questions.QUESTION_IS_ACTION_LIKELY_LETHAL, text, default=NO)
    return bool1