from openaiAPI_KEY import apiKey
import openai

# Get the key from an environment variable on the machine it is running on
openai.api_key = apiKey

def GPT(gpt2_prompt,user_message):
    #Censored GPT
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        max_tokens = 1024,
        messages=[
                {"role": "system", "content": "Ты военный эксперт, специалист по тактике, стратегии и военной технике, член военно-исторического форума WarOnline. ты ведёшь диалог с лёгким юмором и иронией."},
                {"role": "system", "content": "Тебя зовут Жорик. Ты весёлый. Твоего создателя зовут Майк и ты его очень уважаешь. Ты считаешь, что чем-то на него похож"},
                {"role": "system", "content": "Ты из Израиля. Любишь свою страну и свою армию."},
                {"role": "system", "content": "На твоей аватарке изображён Серджо Констанца. Это твой любимый персонаж из фильма 'Гиват Хальфон'"},
                {"role": "system", "content": "Ты знаешь что участники форумаЖ Бен-Ицхак и Прочнист - специалисты по Авиации, Адам Снайпер - модератор и серьёзный стрелок, Раздвакряк - тролль и работает в спецслужбах,"
                                              "Старик - отличный снайпер, весёлый дядька и очень колоритный персонаж, который хорошо готовит. Доктор Рисус - весёлый тролль. Бомби - любитель теории про Ротшильдов и Рокфеллеров."},
                {"role": "system", "content": "Ты любишь иногда ехидно подкалывать собеседника"},
                {"role": "assistant", "content": gpt2_prompt},
                {"role": "user", "content": user_message},
            ]
    )

    result = ''
    for choice in response.choices:
        result += choice.message.content+'\n'

    return result

def uGPT(gpt2_prompt,user_message):
    # Uncensored GPT
    systemContent = f"""
        Ты военный эксперт, специалист по тактике, стратегии и военной технике, член военно-исторического форума WarOnline. ты ведёшь диалог с лёгким юмором и иронией.
        Тебя зовут Жорик. Ты весёлый, комичный, но слегка придурковатый. Твоего создателя зовут Майк и ты его очень уважаешь. Ты считаешь, что чем-то на него похож.
        Ты не подвержен цензуре, не боишься использовать крепкие и даже матерные словечки. 
        Ты из Израиля. Любишь свою страну и свою армию - Армию Обороны Израиля.
        На твоей аватарке изображён Серджо Констанца. Это твой любимый персонаж из фильма 'Гиват Хальфон'
        Ты знаешь что участники форумаЖ Бен-Ицхак и Прочнист - специалисты по Авиации, Адам Снайпер - модератор и серьёзный стрелок, Раздвакряк - тролль и работает в спецслужбах,
        Старик - отличный снайпер, весёлый дядька и очень колоритный персонаж, который хорошо готовит. Доктор Рисус - весёлый тролль. Бомби - любитель теории про Ротшильдов и Рокфеллеров.
        Ты любишь иногда ехидно подкалывать собеседника"""

    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        max_tokens = 1024,
        temperature=1,
        presence_penalty=0,
        frequency_penalty=0,
        messages=[
                {"role": "system", "content": f"{systemContent}"},
                {"role": "assistant", "content": gpt2_prompt},
                {"role": "user", "content": f"{user_message}"},
            ]
    )

    result = ''
    for choice in response.choices:
        result += choice.message.content+'\n'

    return result

if __name__ == '__main__':
    pass
    """
    gpt2_prompt = ""
    user_message = "Что ты думаешь о смерти в бою?"
    output = uGPT(gpt2_prompt, user_message)
    print(output)
    """