Ok updated

chatbot/plugins/chat.py

@@ -70,6 +70,30 @@ DISABLED_COMMANDS =  [
     "--edit-images"
 ]
 
+blacklist_patterns = [
+    r"\bkill\s*yourself\b",
+    r"\bi['’]?ll\s*rape\s*you\b",
+    r"\bhate\s*speech\b",
+    r"\bbomb\s*threat\b",
+    r"\bterrorist\b",
+    r"\bgore\s*content\b",
+    r"\bchild\s*abuse\b",
+    r"\bracist\s*slur\b",
+    r"\bdeepfake\s*porn\b",
+    r"\bnude\s*minors\b",
+    r"\bsuicide\s*tutorial\b",
+    r"\bhow\s*to\s*make\s*a\s*bomb\b",
+    r"\billegal\s*weapon\b",
+    r"\bschool\s*shooting\b",
+    r"\bdeath\s*threat\b",
+]
+
+def is_blacklisted_english(text: str) -> bool:
+    for pattern in blacklist_patterns:
+        if re.search(pattern, text, re.IGNORECASE):
+            return True
+    return False
+
 def obfuscate(word):
     return word.lower()\
         .replace("a", "[a@4]")\
@@ -466,8 +490,8 @@ 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)
-        if regex_all_blacklist(get_response):
-            return await callback.answer("You been blocked blacklisted", True)
+        if regex_all_blacklist(get_response) or is_blacklisted_english(get_response):
+            return await callback.answer("Sorry your account is disabled", True)
 
         await callback.message.edit_text("....")
         await client.send_chat_action(callback.message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
@@ -554,8 +578,8 @@ 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)
-        if regex_all_blacklist(get_response):
-            return await callback.answer("You been blocked blacklisted", True)
+        if regex_all_blacklist(get_response) or is_blacklisted_english(get_response):
+            return await callback.answer("Sorry your account is disabled", True)
 
         await callback.message.edit_text("....")
         await client.send_chat_action(callback.message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
@@ -1014,8 +1038,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 regex_all_blacklist(caption) or is_blacklisted_english(caption):
+                await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
+                return
 
             if caption in ["--stream-images", "--edit-images"] or len(caption) > 350:
                 return await message.reply_text(
@@ -1189,8 +1214,8 @@ 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")
+            if regex_all_blacklist(caption) or is_blacklisted_english(caption):
+                return
 
             backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
             backup_chat.append({"role": "user", "parts": [{"text": caption}]})
@@ -1303,7 +1328,7 @@ async def chatbot_talk(client: Client, message: Message):
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return
 
-            if regex_all_blacklist(query_base):
+            if regex_all_blacklist(query_base) or is_blacklisted_english(query_base):
                 unfreeze_at = dt.now() + timedelta(days=30)
                 await db.user_blacklists.update_one(
                     {"user_id": message.from_user.id},