Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -62,6 +62,27 @@ js = AkenoXJs(DifferentAPIDefault()).connect()
 
 message_memory_state = {}
 
+def obfuscate(word):
+    return word.lower()\
+        .replace("a", "[a@4]")\
+        .replace("b", "[b8]")\
+        .replace("e", "[e3]")\
+        .replace("g", "[g69]")\
+        .replace("i", "[i1!|]")\
+        .replace("o", "[o0]")\
+        .replace("s", "[s5$]")\
+        .replace("t", "[t7+]")\
+        .replace("l", "[l1|]")
+
+def regex_all_blacklist(text):
+    with open("blacklist.txt", "r", encoding="utf-8") as f:
+        words = [obfuscate(w.strip()) for w in f if w.strip()]
+    pattern = r"\b(" + "|".join(words) + r")\b"
+    if re.search(pattern, text, re.IGNORECASE):
+        return True
+    else:
+        return False
+
 async def geni_files_delete(name: str):
     url = f"https://generativelanguage.googleapis.com/v1beta/{name}"
     params = {"key": GOOGLE_API_KEY}
@@ -274,8 +295,13 @@ async def geminigen_prompt(client, callback):
         get_response = data.get("prompt_image", None)
         if not get_response:
             return await callback.answer("Server busy try again later", True)
-        await callback.message.delete()
+        if regex_all_blacklist(get_response):
+            return await callback.answer("You been blocked blacklisted", True)
 
+        await callback.answer("Ok Processed")
+        await callback.message.delete()
+        await client.send_chat_action(callback.message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
+        await asyncio.sleep(2.0)
         backup_chat.append({"role": "user", "parts": [{"text": get_response}]})
         response = await gen.aio.models.generate_content(
             model="gemini-2.0-flash-exp-image-generation",
@@ -330,8 +356,13 @@ async def flux_prompt(client, callback):
         get_response = data.get("prompt_image", None)
         if not get_response:
             return await callback.answer("Server busy try again later", True)
-        await callback.message.delete()
+        if regex_all_blacklist(get_response):
+            return await callback.answer("You been blocked blacklisted", True)
 
+        await callback.answer("Ok Processed")
+        await callback.message.delete()
+        await client.send_chat_action(callback.message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
+        await asyncio.sleep(2.0)
         backup_chat.append({"role": "user", "parts": [{"text": get_response}]})
         response_js = await js.image.create(
             "black-forest-labs/flux-1-schnell",
@@ -620,6 +651,9 @@ async def chatbot_talk(client: Client, message: Message):
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
             await asyncio.sleep(1.5)
 
+            if regex_all_blacklist(caption):
+                return await message.reply_text("You been blocked blacklisted")
+
             if re.findall(r"\b(This is a picture of me)\b", caption, re.IGNORECASE):
                 try:
                     backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
@@ -737,6 +771,9 @@ async def chatbot_talk(client: Client, message: Message):
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_VIDEO)
             await asyncio.sleep(2.0)
             caption = message.caption or "What this?"
+            if regex_all_blacklist(caption):
+                return await message.reply_text("You been blocked blacklisted")
+
             backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
             backup_chat.append({"role": "user", "parts": [{"text": caption}]})
             video_file = await gen.aio.files.upload(file=await message.download())
@@ -839,6 +876,10 @@ async def chatbot_talk(client: Client, message: Message):
         captions = ""
         file_path = "gemini-native-image.png"
         try:
+
+            if regex_all_blacklist(query_base):
+                return await message.reply_text("You been blocked blacklisted")
+
             if query_base in ["/", "/help"]:
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return