Множественные улучшения.

src/OpenAI/GPT35turbo/OA_processing.py

@@ -59,9 +59,6 @@ def openai_response(message_formated_text):
         #запись ошибки в лог с указанием времени и даты
         with open(os.path.join(mother_path, 'src/OpenAI/GPT35turbo/log.txt'), 'a') as log_file:
             log_file.write('\n' + time.strftime("%d.%m.%Y %H:%M:%S") + ' ' + str(ex))
-    #Проверка на то что ответ не содержит ошибку
-    count_length = 0
-
     return response
 
 def sort_message_from_user(message_formated_text, message_id):
@@ -137,4 +134,65 @@ def openai_message_processing(message_id):
         else:
             return f"Сообщение слишком длинное, максимальная длина сообщения \
             {max_token_count - len(message_formated_text[0]['content'])} символов, укоротите его на \
-            {len(str(cursor.execute('SELECT message_text FROM message_list WHERE message_id'))) - max_token_count} символов"
+            {len(str(cursor.execute('SELECT message_text FROM message_list WHERE message_id'))) - max_token_count} символов"
+
+def openai_collecting_history_context(start_id, end_id, message_formated_text, message_id = 0):
+    # собираем список сообщений для OpenAI длинной до 14500 символов начиная с end_id и заканчивая start_id
+    if message_id == 0:
+        message_id = end_id
+        message_formated_text = openai_collecting_history_context(start_id, end_id, message_formated_text, message_id)
+    elif message_id > start_id:
+        message_formated_text = openai_collecting_history_context(start_id, end_id, message_formated_text, (message_id - 1))
+    try:
+        message_formated_text.append({
+            "role": "user",
+            "content": str(*(cursor.execute("SELECT message_text FROM message_list WHERE message_id = ?",
+                                            (message_id,)).fetchone()))
+        })
+    except Exception:
+        pass
+
+    return message_formated_text
+
+def openai_message_history_processing(start_id, end_id):
+    message_formated_text = []
+    message_formated_text = openai_collecting_history_context(start_id, end_id, message_formated_text)
+
+    max_token_count_history = 15000
+    min_token_for_answer_history = 1500
+    count_length = 0
+    try:
+        for message in message_formated_text:
+            count_length += len(message['content'])
+        while count_length > max_token_count_history - min_token_for_answer_history:
+            message_formated_text.pop(1)
+            count_length = 0
+            for message in message_formated_text:
+                count_length += len(message['content'])
+    except IndexError:
+        message_formated_text = [
+            {
+                "role": "system",
+                "content": "Выведи сообщение об ошибке."
+            }
+        ]
+    message_formated_text.append({
+        "role": "user",
+        "content": "Сделай пересказ всей истории сообщений без потери смысла от 3-его лица."
+    })
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo-16k",
+            messages=message_formated_text,
+            max_tokens=max_token_count_history - count_length
+        )
+    except OpenAIError as ex:
+        if 'on requests per min. Limit: 3 / min. Please try again' in str(ex):
+            response = ('Извини мой процессор перегрелся, дай мне минутку отдохнуть')
+        elif 'Bad gateway.' in str(ex):
+            response = (
+                'Ой, где я? Кажется кто то перерзал мой интернет кабель, подожди немного пока я его починю')
+        # запись ошибки в лог с указанием времени и даты
+        with open(os.path.join(mother_path, 'src/OpenAI/GPT35turbo/log.txt'), 'a') as log_file:
+            log_file.write('\n' + time.strftime("%d.%m.%Y %H:%M:%S") + ' ' + str(ex))
+    return response

src/TelegramBot/main.py

@@ -62,6 +62,7 @@ cursor.execute("""CREATE TABLE IF NOT EXISTS user_list (
     user_name TEXT NOT NULL,
     user_role INTEGER,
     user_stats INTEGER
+    user_rep INTEGER
 )""")
 #запись информации о чате в базу данных
 
@@ -69,7 +70,6 @@ async def empty_role(id):
     cursor.execute("UPDATE user_list SET user_role = ? WHERE user_id = ?", (0, id))
     database.commit()
 
-
 async def check(id):
     #проверка что у человека есть роль
     user_role = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (id,)).fetchone()[0]
@@ -92,6 +92,7 @@ async def get_role(id):
     user_role = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (id,)).fetchone()[0]
     return await get_role_name(user_role)
 
+
 async def check_admin(id, chat_id):
     #Проверка что человек есть в списке администраторов чата
     chat_admins = await bot.get_chat_administrators(chat_id)
@@ -167,7 +168,7 @@ async def mute(message: types.Message):
                 #получаем id отправителя сообщение на которое отвечает message
                 target_id = message.reply_to_message.from_user.id
                 target_name = cursor.execute("SELECT user_name FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
-                #Проверка что человек пользователь
+                #Проверка, что человек пользователь
                 if cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0] == 0:
                     #ограничения прав пользователя по отправке сообщений на time секунд
                     time_sec = await time_to_seconds(message.text.split(' ')[1])
@@ -247,20 +248,6 @@ async def new_chat_members(message: types.Message):
         #Заносим пользователя в базу данных
         cursor.execute("INSERT INTO user_list VALUES (?, ?, ?, ?)", (message.new_chat_members[0].id, message.new_chat_members[0].username, 0, 0))
 
-# @dp.message_handler(commands=['mail'])
-# async def mail(message: types.Message):
-#     #получаем тег пользователя которому должны написать в лс
-#     target_tag = message.text.split(' ')[1]
-#     target_tag = target_tag[1:]
-#     print(target_tag)
-#     #получаем id пользователя которому должны написать в лс
-#     target_id = int(cursor.execute("SELECT user_id FROM user_list WHERE user_name = ?", (target_tag,)).fetchone()[0])
-#     print(target_id)
-#     #пишем пользователю в лс, привет {его имя}
-#     await bot.send_message(target_id, f"Привет {target_tag}! Я бот чата {message.chat.title}. {message.from_user.first_name} хочет с тобой связаться. Напиши ему в лс.")
-#     #пишем пользователю в чат, что его сообщение отправлено
-#     await message.reply(f"Сообщение отправлено пользователю {target_tag}")
-
 
 @dp.message_handler(commands=['chatinfo'])
 async def chat_info(message: types.Message):
@@ -273,11 +260,12 @@ async def chat_info(message: types.Message):
         chat_moderators = cursor.execute("SELECT COUNT(user_id) FROM user_list WHERE user_role = 1").fetchone()[0]
         chat_messages = cursor.execute("SELECT chat_stats FROM chat_list WHERE chat_id = ?", (chat_id,)).fetchone()[0]
         await message.reply(
-            f"Название чата: {chat_title}\n"
+            f"<b>Название чата:</b> {chat_title}\n"
-            f"Кол-во пользователей: {chat_members}\n"
+            f"<b>Количество пользователей:</b> {chat_members}\n"
-            f"Кол-во администраторов: {len(chat_admins)}\n"
+            f"<b>Количество администраторов:</b> {len(chat_admins)}\n"
-            f"Кол-во модераторов: {chat_moderators}\n"
+            f"<b>Количество модераторов:</b> {chat_moderators}\n"
-            f"Кол-во сообщений: {chat_messages}")
+            f"<b>Количество сообщений:</b> {chat_messages}",
+            parse_mode="HTML")
         await top10(message)
     else:
         await message.reply(
@@ -302,12 +290,18 @@ async def start(message: types.Message):
     # Проверка наличия столбца имени пользователя в базе данных, если его нет, то добавить.
     try:
         cursor.execute("SELECT user_name FROM user_list")
-    # sqlite3.OperationalError: no such column: user_name
     except sqlite3.OperationalError:
         cursor.execute("ALTER TABLE user_list ADD COLUMN user_name TEXT")
         database.commit()
         await message.reply("База данных пользователей реструктурирована.")
-    # Проверка что все участники чата есть в базе данных
+    # Проверка наличия столбца репутации пользователя в базе данных, если его нет, то добавить.
+    try:
+        cursor.execute("SELECT user_rep FROM user_list")
+    except sqlite3.OperationalError:
+        cursor.execute("ALTER TABLE user_list ADD COLUMN user_rep INTEGER")
+        database.commit()
+        await message.reply("База данных пользователей реструктурирована.")
+
 
 
 @dp.message_handler(commands=['top10'])
@@ -323,17 +317,6 @@ async def top10(message: types.Message):
             f"{config['Telegram']['top10_answer']}\n{top10_message}")
 
 
-@dp.message_handler(commands=['stats'])
-async def stat(message: types.Message):
-    #статистика пользователя в формате: Имя пользователя - количество сообщений
-    user_stats = cursor.execute("SELECT user_stats FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
-    your_id = message.from_id
-    your_name = message.from_user.username
-    await message.reply(
-            f"[{your_name}](tg://user?id={str(your_id)}) - {user_stats[0]}",
-            parse_mode="Markdown")
-
-
 @dp.message_handler(commands=['aboutme'])
 async def aboutme(message: types.Message):
     your_id = message.from_id
@@ -341,6 +324,10 @@ async def aboutme(message: types.Message):
     your_name = message.from_user.username
     user_stats = cursor.execute("SELECT user_stats FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
     rolenum = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
+    user_rep = cursor.execute("SELECT user_rep FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
+    if "None" in str(user_rep):
+        user_rep = 0
+        cursor.execute("UPDATE user_list SET user_rep = ? WHERE user_id = ?", (user_rep, your_id))
     role = await get_role_name(rolenum[0])
     #Имя пользователя на следующей строке статистика сообщений на следующей строке права пользователя
     #если пользователь есть в списке администраторов чата, то выдаём роль администратор
@@ -352,39 +339,43 @@ async def aboutme(message: types.Message):
             if admin.user.id == your_id:
                 if default_for_admin == 0:
                     await message.reply(
-                        f"Вы обнаружены в списке администраторов чата! \n"
+                        f"<b>Вы обнаружены в списке администраторов чата!</b> \n"
-                        f"Вам будет выдана роль: {config['Roles']['user']}\n"
+                        f"<b>Вам будет выдана роль:</b> {config['Roles']['user']}\n"
-                        f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
+                        f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
-                        f"Кол-во сообщений: {user_stats[0]}\n"
+                        f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
-                        f"Роль: {role}",
+                        f"<b>Репутация:</b> {user_rep[0]}\n"
-                        parse_mode="Markdown")
+                        f"<b>Роль:</b> {role}",
+                        parse_mode="HTML")
                     no_answer = True
                 elif default_for_admin == 1:
                     await message.reply(
-                        f"Вы обнаружены в списке администраторов чата.\n"
+                        f"<b>Вы обнаружены в списке администраторов чата!</b> \n"
-                        f"Вам будет выдана роль: {config['Roles']['moderator']}\n"
+                        f"<b>Вам будет выдана роль:</b> {config['Roles']['moderator']}\n"
-                        f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
+                        f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
-                        f"Кол-во сообщений: {user_stats[0]}\n"
+                        f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
-                        f"Роль: {role}",
+                        f"<b>Репутация:</b> {user_rep[0]}\n"
-                        parse_mode="Markdown")
+                        f"<b>Роль:</b> {role}",
+                        parse_mode="HTML")
                     no_answer = True
                 elif default_for_admin == 2:
                     await message.reply(
-                        f"Вы обнаружены в списке администраторов чата.\n"
+                        f"<b>Вы обнаружены в списке администраторов чата!</b> \n"
-                        f"Вам будет выдана роль: {config['Roles']['admin']}\n"
+                        f"<b>Вам будет выдана роль:</b> {config['Roles']['admin']}\n"
-                        f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
+                        f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
-                        f"Кол-во сообщений: {user_stats[0]}\n"
+                        f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
-                        f"Роль: {role}",
+                        f"<b>Репутация:</b> {user_rep[0]}\n"
-                        parse_mode="Markdown")
+                        f"<b>Роль:</b> {role}",
+                        parse_mode="HTML")
                     no_answer = True
                 cursor.execute("UPDATE user_list SET user_role = ? WHERE user_id = ?", (default_for_admin, your_id))
                 database.commit()
     if no_answer == False:
         await message.reply(
-            f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
+            f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
-            f"Кол-во сообщений: {user_stats[0]}\n"
+            f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
-            f"Роль: {role}",
+            f"<b>Репутация:</b> {user_rep[0]}\n"
-            parse_mode="Markdown")
+            f"<b>Роль:</b> {role}",
+            parse_mode="HTML")
 
 @dp.message_handler(commands=['setrole'])
 async def setrole(message: types.Message):
@@ -439,6 +430,10 @@ async def about(message: types.Message):
             target_name = target_name[1:]
             target_id = cursor.execute("SELECT user_id FROM user_list WHERE user_name = ?", (target_name,)).fetchone()
             target_id = target_id[0]
+            user_rep = cursor.execute("SELECT user_rep FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
+            if "None" in str(user_rep):
+                user_rep = 0
+                cursor.execute("UPDATE user_list SET user_rep = ? WHERE user_id = ?", (user_rep, target_id))
             await check(target_id)
         except:
             await message.reply(
@@ -448,13 +443,43 @@ async def about(message: types.Message):
         user_role = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
         user_role = await get_role_name(user_role)
         await message.reply(
-            f"Имя: {target_name}\n"
+            f"<b>Имя:</b> {target_name}\n"
-            f"Кол-во сообщений: {user_stats}\n"
+            f"<b>Кол-во сообщений:</b> {user_stats}\n"
-            f"Роль: {user_role}")
+            f"<b>Репутация:</b> {user_rep}\n"
+            f"<b>Роль:</b> {user_role}", parse_mode="HTML")
     else:
         await message.reply(
             f"Ошибка! Проверьте правильность написания команды.")
 
+from src.OpenAI.GPT35turbo.OA_processing import openai_message_history_processing
+
+@dp.message_handler(commands=['history'])
+async def history(message: types.Message):
+    if message.reply_to_message is not None:
+        # Получаем id сообщения на которое отвечает message
+        start_message_id = message.reply_to_message.message_id
+        # Получаем id сообщения message
+        end_message_id = message.message_id
+    elif message.text.split()[1] != '':
+        end_message_id = message.message_id
+        start_message_id = end_message_id - int(message.text.split()[1])
+    else:
+        await message.reply("Ошибка! Проверьте правильность написания команды.")
+        return
+    temp_message = await message.reply("Подождите немного, я обрабатываю историю сообщений... Функция ЭКСПЕРЕМЕНТАЛЬНАЯ "
+                                       "и может работать некорректно.")
+    # Получаем историю сообщений
+    response = openai_message_history_processing(start_message_id, end_message_id)
+    if response is None:
+        await message.reply("Я не понял тебя, попробуй перефразировать")
+    else:
+        bot_message_id = await message.reply(response['choices'][0]['message']['content'], parse_mode="markdown")
+        # заносим сообщение в базу данных в качестве message_id мы пишем id сообщения в bot_message_id
+        cursor.execute("INSERT INTO message_list VALUES (?, ?, ?, ?)",
+                          (bot_message_id.message_id, response['choices'][0]['message']['content'], 0, message.message_id))
+        # очищаем сообщение
+        asyncio.create_task(delete_message(temp_message, 0))
+
 from src.OpenAI.GPT35turbo.OA_processing import openai_message_processing
 
 
@@ -493,6 +518,18 @@ async def in_message(message: types.Message):
                 send_answer = True
                 typing_mode = False
 
+        if send_answer == False:
+            # Если сообщение отвечает на другое сообщение, то проверяем наличие слова спасибо
+            if message.reply_to_message is not None:
+                if message.text.startswith("Спасибо"):
+                    target_id = message.reply_to_message.from_user.id
+                    target_name = cursor.execute("SELECT user_name FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
+                    cursor.execute("UPDATE user_list SET user_rep = user_rep + 1 WHERE user_id = ?", (target_id,))
+                    database.commit()
+                    await message.reply(
+                        f"Спасибо, [{target_name}](tg://user?id={str(target_id)}) за ваш ответ!",
+                        parse_mode="Markdown")
+
         if send_answer:
             if typing_mode is False:
                 your_id = message.from_id