Update main.py

main.py

@@ -1,4 +1,5 @@
 import os
+import time
 import threading
 import random
 import aiohttp
@@ -13,7 +14,7 @@ from rubpy.bot import BotClient, filters
 # ایمپورت‌های جدید برای هوش مصنوعی و کار با فایل صوتی
 from huggingface_hub import AsyncInferenceClient
 from PIL import Image
-from pydub import AudioSegment  # ایمپورت اضافه شده برای ترکیب پادکست
+from pydub import AudioSegment
 
 # --- تنظیمات وب سرور ---
 app = Flask(__name__)
@@ -61,30 +62,25 @@ MAIN_KEYPAD_DICT = {
 }
 
 
-# --- تابع ارسال منحصراً برای کیبورد پایین صفحه ---
+# --- تابع ارسال منحصراً برای کیبورد پایین صفحه (اصلاح شده برای جلوگیری از اسپم) ---
 async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
-    if not use_keyboard:
-        msg = await client.send_message(chat_id, text)
-        return msg
-
     try:
+        if not use_keyboard:
+            return await client.send_message(chat_id, text)
+
         payload = {
             "chat_id": chat_id,
             "text": text,
             "chat_keypad_type": "New",
             "chat_keypad": MAIN_KEYPAD_DICT
         }
-        msg = await client._make_request("sendMessage", payload)
-        return msg
+        return await client._make_request("sendMessage", payload)
     except Exception:
         try:
-            msg = await client.send_message(chat_id, text, chat_keypad=MAIN_KEYPAD_DICT, chat_keypad_type="New")
-            return msg
+            # اگر حالت کیبورد دار در سطح API خطا داد، فقط یکبار به صورت ساده پیام ارسال شود
+            return await client.send_message(chat_id, text)
         except Exception:
-            pass
-
-    msg = await client.send_message(chat_id, text)
-    return msg
+            return None
 
 
 # --- 🚨 تابع هوشمند دانلود فایل 🚨 ---
@@ -242,6 +238,10 @@ 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):
@@ -650,13 +650,42 @@ else:
     @bot.on_update(filters.private)
     async def main_handler(client, update):
         try:
-            user_text = getattr(update, "text", "") or getattr(getattr(update, "message", None), "text", "") or getattr(getattr(update, "new_message", None), "text", "")
+            current_time = time.time()
+            
+            # --- 🛡 سیستم ضد اسپم و حذف پیام‌های تکراری 🛡 ---
+            msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
+            msg_id = getattr(update, "message_id", None)
+            if not msg_id and msg_obj:
+                if isinstance(msg_obj, dict):
+                    msg_id = msg_obj.get("message_id")
+                else:
+                    msg_id = getattr(msg_obj, "message_id", None)
+            
+            # جلوگیری از پردازش یک آیدی پیام برای دفعات متعدد
+            if msg_id:
+                if msg_id in processed_message_ids:
+                    return
+                processed_message_ids.add(msg_id)
+                # پاکسازی کش برای جلوگیری از پر شدن رم
+                if len(processed_message_ids) > 5000:
+                    processed_message_ids.clear()
+            
+            chat_id = getattr(update, "chat_id", None) or getattr(update, "author_guid", None) or getattr(update, "object_guid", None)
+            if not chat_id: return
+
+            # سیستم Rate Limit برای کاربران (جلوگیری از پاسخ به پیام‌های جمع شده در بک‌لاگ ربات)
+            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 ""
             
             # تشخیص فایل
             is_file = False
             file_name = "unknown_file.jpg"
-            msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
             
             if msg_obj:
                 for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
@@ -675,9 +704,6 @@ else:
                             if isinstance(file_attr, dict) and 'file_name' in file_attr: file_name = file_attr['file_name']
                             break
 
-            chat_id = getattr(update, "chat_id", None) or getattr(update, "author_guid", None) or getattr(update, "object_guid", None)
-            if not chat_id: return
-
             if chat_id not in user_states:
                 user_states[chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None}
 
@@ -686,7 +712,6 @@ else:
                 await send_with_keyboard(client, chat_id, "سلام! به ربات هوشمند آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:", True)
                 return
 
-            # اصلاح شرط‌های if منطبق با متن دکمه‌های جدید
             if user_text_str in ["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
                 user_states[chat_id] = {"mode": "chat", "text": "", "history":[]}
                 await send_with_keyboard(client, chat_id, "💬 شما وارد بخش **چت با هوش مصنوعی** شدید.\n\nهر سوالی دارید بفرستید تا جواب بدم:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
@@ -772,7 +797,6 @@ else:
                     await send_with_keyboard(client, chat_id, "❌ شماره نامعتبر است! لطفاً فقط یک عدد بین 1 تا 30 بفرستید.", False)
                 return
 
-            # --- بخش مسیریابی برای قابلیت جدید (پادکست) ---
             elif current_mode == "podcast_waiting_for_topic":
                 if user_text_str:
                     asyncio.create_task(process_podcast(client, chat_id, user_text_str))