Commited backup

D4n13l3k00/FTG-Modules/DeepAI.py

@@ -0,0 +1,78 @@
+# .------.------.------.------.------.------.------.------.------.------.
+# |D.--. |4.--. |N.--. |1.--. |3.--. |L.--. |3.--. |K.--. |0.--. |0.--. |
+# | :/\: | :/\: | :(): | :/\: | :(): | :/\: | :(): | :/\: | :/\: | :/\: |
+# | (__) | :\/: | ()() | (__) | ()() | (__) | ()() | :\/: | :\/: | :\/: |
+# | '--'D| '--'4| '--'N| '--'1| '--'3| '--'L| '--'3| '--'K| '--'0| '--'0|
+# `------`------`------`------`------`------`------`------`------`------'
+#
+#                     Copyright 2023 t.me/D4n13l3k00
+#           Licensed under the Creative Commons CC BY-NC-ND 4.0
+#
+#                    Full license text can be found at:
+#       https://creativecommons.org/licenses/by-nc-nd/4.0/legalcode
+#
+#                           Human-friendly one:
+#            https://creativecommons.org/licenses/by-nc-nd/4.0
+
+# meta developer: @D4n13l3k00
+
+
+import io
+
+import requests
+
+from .. import loader  # type: ignore
+
+
+def register(cb):
+    cb(DeepAIMod())
+
+
+class DeepAIMod(loader.Module):
+    strings = {"name": "DeepAI"}
+
+    def __init__(self):
+        self.name = self.strings["name"]
+
+    async def client_ready(self, client, db):
+        self._db = db
+
+    async def dai_set_tokencmd(self, m):
+        self._db.set("deepai", "token", str(m.raw_text.split(" ", 1)[1]))
+        await m.edit("[DeepAI] Токен установлен")
+
+    async def cnsfwcmd(self, m):
+        """.cnsfw <reply to photo/sticker> - Check on nsfw content by DeepAI"""
+        reply = await m.get_reply_message()
+        if not reply:
+            await m.edit("<b>Reply to media</b>")
+            return
+        try:
+            media = reply.media
+        except Exception:
+            await m.edit("<b>Only media</b>")
+            return
+        if token := self._db.get("deepai", "token", None):
+            await m.edit("[DeepAI] Детектим nsfw...")
+            photo = io.BytesIO()
+            await m.client.download_media(media, photo)
+            photo.seek(0)
+            r = requests.post(
+                "https://api.deepai.org/api/nsfw-detector",
+                files={
+                    "image": photo.read(),
+                },
+                headers={"api-key": token},
+            )
+            try:
+                await m.edit(
+                    "[DeepAI] Тут Nsfw на "
+                    + str(round(r.json()["output"]["nsfw_score"] * 100, 1))
+                    + "%"
+                )
+            except Exception:
+                await m.edit(f"[DeepAI] {str(r.json())}")
+        else:
+            await m.edit(
+                "[DeepAI] Укажите токен для работы с API\n<code>.dai_set_token TOKEN</code>"
+            )