Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -29,15 +29,14 @@ import subprocess
 import sys
 import time
 import traceback
-from gpytranslate import SyncTranslator
-from io import BytesIO
-from types import ModuleType
-from typing import Dict, List, Optional, Tuple, Union
-
 import requests
-import time
 import json
 import io
+from datetime import datetime as dt, timedelta
+
+from gpytranslate import SyncTranslator
+from io import BytesIO
+from typing import Dict, List, Optional, Tuple, Union
 
 from PIL import Image
 from pyrogram import *
@@ -282,6 +281,10 @@ async def check_steal_channel_id(client, event: ChatMemberUpdated):
     user_id = event.new_chat_member.promoted_by.id if event.new_chat_member else None
     if not user_id or event.chat.username is None:
         return
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        return
+
     keyboard = []
     keyboard.append([
         InlineKeyboardButton("🔎 View Channel", url=f"https://t.me/{event.chat.username}"),
@@ -319,6 +322,11 @@ async def cancel_(client, callback_query):
 @Client.on_callback_query(filters.regex("^deletemydatagm$"))
 async def deletemydata(client, callback):
     user_id = callback.from_user.id
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     delm = await db._clear_chatbot_database(user_id)
     if delm:
         await callback.edit_message_text(delm)
@@ -328,6 +336,11 @@ async def deletemydata(client, callback):
 @Client.on_callback_query(filters.regex("^photo_(\d+)$"))
 async def modephoto(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     data = await db.backup_chatbot.find_one({"user_id": user_id})
     if not data:
         return await callback.answer("Can't found user", True)
@@ -356,6 +369,11 @@ async def modephoto(client, callback):
 @Client.on_callback_query(filters.regex("^menutools_(\d+)$"))
 async def menu_tools(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     data = await db.backup_chatbot.find_one({"user_id": user_id})
     if not data:
         return await callback.answer("Can't found user", True)
@@ -382,6 +400,11 @@ async def menu_tools(client, callback):
 @Client.on_callback_query(filters.regex("^autopost_(\d+)$"))
 async def auto_post_channel(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     try:
         backup_chat = await db._get_chatbot_chat_from_db(user_id)
         data = await db.backup_chatbot.find_one({"user_id": user_id})
@@ -428,6 +451,11 @@ async def auto_post_channel(client, callback):
 @Client.on_callback_query(filters.regex("^genprompt_(\d+)$"))
 async def geminigen_prompt(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     captions = ""
     file_path = "gemini-native-image.png"
     try:
@@ -511,6 +539,11 @@ async def geminigen_prompt(client, callback):
 @Client.on_callback_query(filters.regex("^fluxprompt_(\d+)$"))
 async def flux_prompt(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     file_path = None
     file_photo = None
     try:
@@ -603,6 +636,11 @@ async def flux_prompt(client, callback):
 @Client.on_callback_query(filters.regex("^removerbg_(\d+)$"))
 async def remover_bg(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     try:
         data = await db.backup_chatbot.find_one({"user_id": int(user_id)})
         if not data:
@@ -643,6 +681,10 @@ async def closeed(client, callback):
 @Client.on_callback_query(filters.regex("^memory_(\d+)$"))
 async def memory_updated(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
 
     if message_memory_state.get(callback.message.id, False):
         return await callback.answer("Memory already updated!", show_alert=False)
@@ -677,6 +719,11 @@ async def memory_updated(client, callback):
 @Client.on_callback_query(filters.regex("^tr_(\d+)_(\w+)$"))
 async def terjemahkan(client, callback):
     user_id, targetlang = callback.matches[0].groups()
+    user_bl = await db.user_blacklists.find_one({"user_id": int(user_id)})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     try:
         if user_id == 0:
             return await callback.answer("Server busy try again later", True)
@@ -708,6 +755,11 @@ async def terjemahkan(client, callback):
 @Client.on_callback_query(filters.regex("^photomode_(\d+)_(\w+)$"))
 async def photo_mode(client, callback):
     user_id, mode = callback.matches[0].groups()
+    user_bl = await db.user_blacklists.find_one({"user_id": int(user_id)})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     data = await db.backup_chatbot.find_one({"user_id": int(user_id)})
     if not data:
         return await callback.answer("Can't found user", True)
@@ -731,6 +783,11 @@ async def photo_mode(client, callback):
 @Client.on_callback_query(filters.regex("^trmulti_(\d+)$"))
 async def multiple_langagues(client, callback):
     user_id = int(callback.matches[0].group(1))
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     keyboard = []
     keyboard.append([
         InlineKeyboardButton("🇦🇪", callback_data=f"tr_{user_id}_ar"),
@@ -781,6 +838,11 @@ def anonymous_user(user_id: int = None):
 @Client.on_callback_query(filters.regex("^response_(\d+)_(\w+)$"))
 async def response_call(client, callback):
     user_id, chat = callback.matches[0].groups()
+    user_bl = await db.user_blacklists.find_one({"user_id": int(user_id)})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback.answer("Sorry your account is disabled", True)
+        return
+
     keyboard = InlineKeyboardMarkup(
         [
             [
@@ -848,6 +910,10 @@ async def vote(client, callback_query):
         unlike_counts[message_id] = set()
 
     if action == "like":
+        user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+        if user_bl and user_bl.get("is_frozen", False):
+            await callback.answer("Sorry your account is disabled", True)
+            return
         if user_id in like_counts[message_id]:
             like_counts[message_id].remove(user_id)
         else:
@@ -857,6 +923,10 @@ async def vote(client, callback_query):
                 unlike_counts[message_id].remove(user_id)
 
     elif action == "unlike":
+        user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+        if user_bl and user_bl.get("is_frozen", False):
+            await callback.answer("Sorry your account is disabled", True)
+            return
         if user_id in unlike_counts[message_id]:
             unlike_counts[message_id].remove(user_id)
         else:
@@ -876,6 +946,9 @@ async def vote(client, callback_query):
     & ~filters.forwarded
 )
 async def startbot(client: Client, message: Message):
+    user_bl = await db.user_blacklists.find_one({"user_id": message.from_user.id})
+    if user_bl and user_bl.get("is_frozen", False):
+        return
     buttons = [
         [
             InlineKeyboardButton(
@@ -933,6 +1006,10 @@ async def chatbot_talk(client: Client, message: Message):
         captions = ""
         file_path = "gemini-native-image.png"
         try:
+            user_bl = await db.user_blacklists.find_one({"user_id": message.from_user.id})
+            if user_bl and user_bl.get("is_frozen", False):
+                return
+
             caption = message.caption or "What this?"
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
             await asyncio.sleep(1.5)
@@ -1105,6 +1182,10 @@ async def chatbot_talk(client: Client, message: Message):
     if message.video:
         video_file = None
         try:
+            user_bl = await db.user_blacklists.find_one({"user_id": message.from_user.id})
+            if user_bl and user_bl.get("is_frozen", False):
+                return
+
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_VIDEO)
             await asyncio.sleep(2.0)
             caption = message.caption or "What this?"
@@ -1156,6 +1237,10 @@ async def chatbot_talk(client: Client, message: Message):
         sticker_file = None
         media = None
         try:
+            user_bl = await db.user_blacklists.find_one({"user_id": message.from_user.id})
+            if user_bl and user_bl.get("is_frozen", False):
+                return
+
             f = message.sticker.file_id
             backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
             await client.send_chat_action(message.chat.id, enums.ChatAction.CHOOSE_STICKER)
@@ -1213,9 +1298,32 @@ async def chatbot_talk(client: Client, message: Message):
         captions = ""
         file_path = "gemini-native-image.png"
         try:
+            user_bl = await db.user_blacklists.find_one({"user_id": message.from_user.id})
+            if user_bl and user_bl.get("is_frozen", False):
+                return
 
             if regex_all_blacklist(query_base):
-                return await message.reply_text("You been blocked blacklisted")
+                unfreeze_at = dt.now() + timedelta(hours=24)
+                await db.user_blacklists.update_one(
+                    {"user_id": message.from_user.id},
+                    {"$set": {
+                        "is_frozen": True,
+                        "reason": "abusive_language",
+                        "frozen_at": dt.now(),
+                        "unfreeze_at": unfreeze_at
+                    }},
+                    upsert=True
+                )
+                await message.reply_text(
+                    "⚠️ You've been restricted for 24 hours\n"
+                    f"Expires: {unfreeze_at.strftime('%Y-%m-%d %H:%M')}",
+                    reply_markup=InlineKeyboardMarkup([[
+                        InlineKeyboardButton(
+                            "⏳ Understood", 
+                            callback_data="closedd"
+                        )
+                    ]])
+                )
 
             if query_base in DISABLED_COMMANDS:
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)