Commited backup

vsecoder/hikka_modules/chatgptfree.py

@@ -0,0 +1,110 @@
+"""
+                              _
+__   _____  ___  ___ ___   __| | ___ _ __
+\ \ / / __|/ _ \/ __/ _ \ / _` |/ _ \ '__|
+ \ V /\__ \  __/ (_| (_) | (_| |  __/ |
+  \_/ |___/\___|\___\___/ \__,_|\___|_|
+
+  Copyleft 2022 t.me/vsecoder
+  This program is free software; you can redistribute it and/or modify
+
+"""
+# meta developer: @vsecoder_m
+
+version = (1, 0, 0)
+
+from telethon import functions
+from telethon.tl.types import Message
+import asyncio
+import logging
+
+from .. import loader, utils  # type: ignore
+
+logger = logging.getLogger(__name__)
+
+
+@loader.tds
+class ChatGPTfreeMod(loader.Module):
+    """
+    Бесплатный модуль для ChatGPT
+    https://t.me/Jarvis_IT_Assistant_bot
+    Сначала запустите бота и отключите уведомления
+    """
+
+    strings = {
+        "name": "ChatGPTfree",
+        "loading": "🔄 Ваш запрос обрабатывается...",
+        "no_args": "🚫 Не указан текст для обработки!",
+        "start_text": "<b>🤖 ChatGPT:</b>\n",
+        "context_text": "❕ Создался новый диалог. Предыдущие запросы удалены.",
+    }
+
+    def __init__(self):
+        self.name = self.strings["name"]
+
+    async def client_ready(self, client, db):
+        self.client = client
+        self.db = db
+        self.gpt_free = "@Jarvis_IT_Assistant_bot"
+
+    async def message_q(
+        self,
+        text: str,
+        user_id: int,
+        mark_read: bool = False,
+        delete: bool = False,
+        ignore_answer: bool = False,
+    ):
+        """Отправляет сообщение и возращает ответ"""
+        async with self.client.conversation(user_id) as conv:
+            msg = await conv.send_message(text)
+            while True:
+                await asyncio.sleep(1)
+                response = await conv.get_response()
+                if mark_read:
+                    await conv.mark_read()
+
+                if delete:
+                    await msg.delete()
+                    await response.delete()
+
+                if ignore_answer:
+                    return response
+
+                if "✅ Запрос отправлен" in response.text:
+                    continue
+
+                if "Ожидание ответа" in response.text:
+                    continue
+
+                return response
+
+    async def chatgptfreecmd(self, message: Message):
+        """
+        {text} - обработать текст через ChatGPT
+        """
+        args = utils.get_args_raw(message)
+
+        if not args:
+            return await utils.answer(message, self.strings["no_args"])
+
+        await utils.answer(message, self.strings["loading"])
+
+        response = await self.message_q(
+            args, self.gpt_free, mark_read=True, delete=True, ignore_answer=False
+        )
+
+        text = self.strings["start_text"] + response.text.replace(
+            "/context", "<code>.contextgpt</code>"
+        )
+
+        return await utils.answer(message, text)
+
+    async def contextgptcmd(self, message: Message):
+        """
+        - сбросить диалог и начать новый
+        """
+        await self.message_q(
+            "/context", self.gpt_free, mark_read=True, delete=True, ignore_answer=True
+        )
+        return await utils.answer(message, self.strings["context_text"])