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):
    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