Upload 2 files

WarBot.py

@@ -11,6 +11,7 @@ from torch import package
 #import textwrap
 from textwrap3 import wrap
 import replicate #imaging
+import chatGPT #This is a costly solution
 
 
 # util function to get expected len after tokenizing
@@ -181,8 +182,11 @@ def get_response(quote:str,model,tokenizer,model_punct,temperature=0.2):
     response = re.sub(r'\.\)', '.', response)  # Fix the .) issue
     response = response.replace('[]', '') # Fix the [] issue
 
+    #Experimental:
+    response = chatGPT.uGPT(response,quote)
     return response
 
+
 if __name__ == '__main__':
     """
     quote = "Здравствуй, Жопа, Новый Год, выходи на ёлку!"

chatGPT.py

@@ -5,6 +5,7 @@ import openai
 openai.api_key = apiKey
 
 def GPT(gpt2_prompt,user_message):
+    #Censored GPT
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         max_tokens = 1024,
@@ -13,7 +14,7 @@ def GPT(gpt2_prompt,user_message):
                 {"role": "system", "content": "Тебя зовут Жорик. Ты весёлый. Твоего создателя зовут Майк и ты его очень уважаешь. Ты считаешь, что чем-то на него похож"},
                 {"role": "system", "content": "Ты из Израиля. Любишь свою страну и свою армию."},
                 {"role": "system", "content": "На твоей аватарке изображён Серджо Констанца. Это твой любимый персонаж из фильма 'Гиват Хальфон'"},
-                {"role": "system", "content": "Ты знаешь что участники форумаЖ Бен-Ицхак и Прочнист - специалисты по Авиации, Адам Снайпер - модератор и серьёзный стрелок, "
+                {"role": "system", "content": "Ты знаешь что участники форумаЖ Бен-Ицхак и Прочнист - специалисты по Авиации, Адам Снайпер - модератор и серьёзный стрелок, Раздвакряк - тролль и работает в спецслужбах,"
                                               "Старик - отличный снайпер, весёлый дядька и очень колоритный персонаж, который хорошо готовит. Доктор Рисус - весёлый тролль. Бомби - любитель теории про Ротшильдов и Рокфеллеров."},
                 {"role": "system", "content": "Ты любишь иногда ехидно подкалывать собеседника"},
                 {"role": "assistant", "content": gpt2_prompt},
@@ -25,4 +26,44 @@ def GPT(gpt2_prompt,user_message):
     for choice in response.choices:
         result += choice.message.content+'\n'
 
-    return result
+    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)
+    """