Update main.py

main.py

@@ -28,25 +28,22 @@ ADMIN_CODE = "3011"
 BOT_GUID = None
 
 # =======================================================
-# 🔥 سیستم کنترل سرعت و ضد رگبار (Burst Controller) 🔥
+# 🔥 سیستم ضد اسپم کاربر-محور (اصلاح شده) 🔥
 # =======================================================
-global_burst_count = 0
-global_burst_time = time.time()
-burst_lock = threading.Lock()
-
-def is_backlog_burst():
-    global global_burst_count, global_burst_time
-    with burst_lock:
-        now = time.time()
-        if now - global_burst_time > 1.0:
-            global_burst_time = now
-            global_burst_count = 1
-            return False
-        else:
-            global_burst_count += 1
-            if global_burst_count > 5:
-                return True
-            return False
+user_message_times = {}
+
+def is_user_spamming(chat_id):
+    now = time.time()
+    times = user_message_times.get(chat_id,[])
+    # فقط زمان پیام‌های 3 ثانیه اخیر را نگه می‌داریم
+    times = [t for t in times if now - t < 3.0]
+    times.append(now)
+    user_message_times[chat_id] = times
+    
+    # اگر یک کاربر به تنهایی در 3 ثانیه بیش از 4 پیام داد، پیامش نادیده گرفته شود
+    if len(times) > 4:
+        return True
+    return False
 
 # --- سیستم دیتابیس حساب کاربری متصل به دیتاست هاگینگ فیس ---
 DB_FILE = "users_db.json"
@@ -97,28 +94,40 @@ def load_db():
             pass
     return {}
 
-def _upload_db_background():
-    if HF_TOKEN_DB:
-        api = HfApi(token=HF_TOKEN_DB)
-        try:
-            api.upload_file(
-                path_or_fileobj=DB_FILE,
-                path_in_repo=DB_FILE,
-                repo_id=DATASET_REPO,
-                repo_type="dataset"
-            )
-        except Exception as e:
-            print("❌ خطا در آپلود دیتابیس به هاگینگ فیس:", str(e)[:100])
+db_needs_upload = False
 
 def save_db(db_data):
+    global db_needs_upload
     with db_lock:
         try:
             with open(DB_FILE, "w", encoding="utf-8") as f:
                 json.dump(db_data, f, ensure_ascii=False, indent=4)
-            threading.Thread(target=_upload_db_background, daemon=True).start()
+            db_needs_upload = True  # فقط تیک می‌زنیم که نیاز به آپلود دارد
         except Exception as e:
             print("خطا در ذخیره دیتابیس:", e)
 
+# یک تسک پس‌زمینه برای آپلود منظم و بدون فشار به سرور
+async def background_db_uploader():
+    global db_needs_upload
+    while True:
+        await asyncio.sleep(300) # هر ۵ دقیقه چک می‌کند
+        if db_needs_upload and HF_TOKEN_DB:
+            db_needs_upload = False
+            api = HfApi(token=HF_TOKEN_DB)
+            try:
+                # استفاده از to_thread برای جلوگیری از قفل شدن ربات
+                await asyncio.to_thread(
+                    api.upload_file,
+                    path_or_fileobj=DB_FILE,
+                    path_in_repo=DB_FILE,
+                    repo_id=DATASET_REPO,
+                    repo_type="dataset"
+                )
+                print("✅ بکاپ دیتابیس در هاگینگ فیس ذخیره شد.")
+            except Exception as e:
+                print("❌ خطا در آپلود دیتابیس به هاگینگ فیس:", str(e)[:100])
+                db_needs_upload = True # اگر خطا داد دفعه بعد دوباره تلاش کند
+
 user_credits_db = load_db()
 
 def get_or_create_referral_code(chat_id):
@@ -644,7 +653,6 @@ SPEAKERS = {
 
 user_states = {}
 processed_message_ids = set()
-user_last_request_time = {}
 
 
 async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=None):
@@ -1572,9 +1580,6 @@ else:
     async def main_handler(client, update):
         global BOT_GUID
         
-        if is_backlog_burst():
-            return
-            
         try:
             if not BOT_GUID:
                 try:
@@ -1605,13 +1610,6 @@ else:
                 if unique_msg_key in processed_message_ids: return
                 processed_message_ids.add(unique_msg_key)
                 if len(processed_message_ids) > 5000: processed_message_ids.clear()
-            
-            current_time = time.time()
-            
-            last_req_time = user_last_request_time.get(chat_id, 0)
-            if current_time - last_req_time < 1.0:
-                return
-            user_last_request_time[chat_id] = current_time
 
             user_text = getattr(update, "text", "") or getattr(msg_obj, "text", "")
             user_text_str = str(user_text).strip() if user_text else ""
@@ -1619,6 +1617,10 @@ else:
 
             str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
 
+            # بررسی اسپم فقط برای همین کاربر
+            if is_user_spamming(str_chat_id):
+                return
+
             if str_chat_id not in user_states:
                 user_states[str_chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None}
 
@@ -2220,7 +2222,10 @@ if __name__ == "__main__":
     
     if bot_token:
         loop = asyncio.get_event_loop()
-        loop.set_default_executor(concurrent.futures.ThreadPoolExecutor(max_workers=16))
+        loop.set_default_executor(concurrent.futures.ThreadPoolExecutor(max_workers=32))
+        
+        # روشن کردن سیستم آپلود دیتابیس هوشمند
+        loop.create_task(background_db_uploader())
         
-        print("ربات آلفا پرو با سیستم اشتراک نامحدود + سپر امنیتی + دانلود پیشرفته فایل + 16 Worker پس‌زمینه روشن شد...")
+        print("ربات آلفا پرو با سیستم اشتراک نامحدود + سپر امنیتی + دانلود پیشرفته فایل + 32 Worker پس‌زمینه روشن شد...")
         bot.run()