Update main.py

main.py

@@ -23,9 +23,9 @@ from pydub import AudioSegment
 # --- کد مدیریت برای ارتقای کاربران بدون نیاز به لاگین ---
 ADMIN_CODE = "3011"
 
-# --- ثبت زمان روشن شدن ربات برای جلوگیری از پرتاب پیام‌های قدیمی ---
-# ده ثانیه زودتر در نظر می‌گیریم تا خطای هماهنگی ساعت سرورها مشکلی ایجاد نکند
-BOT_STARTUP_TIME = time.time() - 10.0 
+# --- ثبت زمان دقیق روشن شدن ربات ---
+# (حذف ارفاق 10 ثانیه‌ای قبلی برای جلوگیری کامل از پیام‌های قدیمی)
+BOT_STARTUP_TIME = time.time()
 
 # متغیر سراسری برای ذخیره آیدی خود ربات (جلوگیری از جواب دادن ربات به خودش)
 BOT_GUID = None
@@ -38,7 +38,7 @@ db_lock = threading.Lock()
 
 # --- الگوریتم تبدیل تاریخ میلادی به شمسی (بدون نیاز به نصب کتابخانه) ---
 def gregorian_to_jalali(gy, gm, gd):
-    g_d_m = [0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334]
+    g_d_m =[0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334]
     gy2 = (gy + 1) if (gm > 2) else gy
     days = 355666 + (365 * gy) + ((gy2 + 3) // 4) - ((gy2 + 99) // 100) + ((gy2 + 399) // 400) + gd + g_d_m[gm - 1]
     jy = -1595 + (33 * (days // 12053))
@@ -185,43 +185,43 @@ def run_flask():
 
 # --- ساختار کیبورد آپدیت شده ---
 MAIN_KEYPAD_DICT = {
-    "rows": [
+    "rows":[
         {
-            "buttons": [
+            "buttons":[
                 {"id": "chat_btn", "type": "Simple", "button_text": "چت با هوش مصنوعی 🤖"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "img_btn", "type": "Simple", "button_text": "ساخت تصاویر🎨"},
                 {"id": "edit_img_btn", "type": "Simple", "button_text": "ویرایش تصاویر 🪄"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "podcast_btn", "type": "Simple", "button_text": "ساخت پادکست 🎙️"},
                 {"id": "tts_btn", "type": "Simple", "button_text": "تبدیل متن به صدا🗣️"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "stt_btn", "type": "Simple", "button_text": "فایل صوتی به متن 📝"},
                 {"id": "file_btn", "type": "Simple", "button_text": "تحلیل فایل 📁"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "account_btn", "type": "Simple", "button_text": "حساب کاربری 👤"},
                 {"id": "buy_btn", "type": "Simple", "button_text": "خرید اشتراک 💎"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "transfer_btn", "type": "Simple", "button_text": "انتقال اکانت از برنامه به ربات"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "cancel_btn", "type": "Simple", "button_text": "برگشت♻️"}
             ]
         }
@@ -251,10 +251,10 @@ async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
 
 # --- 🚨 تابع هوشمند دانلود فایل (با استفاده از UUID جهت جلوگیری از تداخل حریم خصوصی) 🚨 ---
 async def helper_download_file(client, msg_obj):
-    errors = []
+    errors =[]
     file_id = None
     file_obj = None
-    for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
+    for attr in['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
         val = getattr(msg_obj, attr, None)
         if val:
             file_obj = val
@@ -307,7 +307,7 @@ async def helper_download_file(client, msg_obj):
 GEMINI_KEYS_STR1 = os.environ.get("GEMINI_API_KEYS1", "")
 GEMINI_KEYS_STR2 = os.environ.get("GEMINI_API_KEYS2", "")
 
-_raw_keys = []
+_raw_keys =[]
 if GEMINI_KEYS_STR1:
     _raw_keys.extend(GEMINI_KEYS_STR1.split(","))
 if GEMINI_KEYS_STR2:
@@ -332,10 +332,10 @@ def get_next_gemini_keys(count=100):
     with gemini_key_lock:
         total_keys = len(GEMINI_KEYS)
         if total_keys == 0:
-            return []
+            return[]
             
         actual_count = min(count, total_keys)
-        selected_keys = []
+        selected_keys =[]
         
         for _ in range(actual_count):
             selected_keys.append(GEMINI_KEYS[current_gemini_key_index])
@@ -344,7 +344,7 @@ def get_next_gemini_keys(count=100):
         return selected_keys
 
 HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
-HF_TOKENS = [k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
+HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
@@ -352,7 +352,7 @@ bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 # --- 🚨 تابع اختصاصی آپلود فایل به روبیکا 🚨 ---
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
-    error_logs = []
+    error_logs =[]
     
     api_file_type = "Image" if file_type in ["photo", "Image", "image"] else "Voice" if file_type in ["voice", "Voice", "audio"] else "File"
     
@@ -413,7 +413,7 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
     return "\n".join(error_logs)
 
 
-WORKER_URLS = [
+WORKER_URLS =[
     "https://hamed744-ttspro.hf.space/generate",
     "https://hamed744-ttspro2.hf.space/generate",
     "https://hamed744-ttspro3.hf.space/generate",
@@ -454,7 +454,7 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
     
     proc_msg = await send_with_keyboard(client, chat_id, "🧠 در حال پردازش...", False)
     history = user_states[chat_id].get("history", [])
-    new_parts = []
+    new_parts =[]
     
     if prompt: new_parts.append({"text": prompt})
     elif file_bytes: new_parts.append({"text": "لطفاً این فایل را به دقت بررسی کن."})
@@ -520,7 +520,7 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
 
     try:
         max_len = 1000 
-        chunks = []
+        chunks =[]
         temp_text = final_answer
         while len(temp_text) > max_len:
             split_idx = temp_text.rfind('\n', 0, max_len)
@@ -559,7 +559,7 @@ async def process_image(client, chat_id, prompt):
         async with aiohttp.ClientSession() as session:
             for key in keys_to_try_gemini:
                 url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
-                payload = {"contents": [{"parts": [{"text": gemini_sys_prompt}]}], "generationConfig": {"temperature": 0.7}}
+                payload = {"contents": [{"parts":[{"text": gemini_sys_prompt}]}], "generationConfig": {"temperature": 0.7}}
                 try:
                     async with session.post(url, json=payload, timeout=20) as response:
                         if response.status == 200:
@@ -572,7 +572,7 @@ async def process_image(client, chat_id, prompt):
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
             if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            if msg_id: await client.delete_messages(chat_id, [msg_id])
+            if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
     
     short_preview = enhanced_prompt[:150] + "..." if len(enhanced_prompt) > 150 else enhanced_prompt
@@ -621,7 +621,7 @@ async def translate_text_aloha(prompt_text):
     session_hash = ''.join(random.choices(string.ascii_lowercase + string.digits, k=11))
     join_url = "https://hamed744-translate-tts-aloha.hf.space/gradio_api/queue/join"
     payload = {
-        "data": [prompt_text, "انگلیسی (آمریکا) - جنی (زن)", 0, 0, 0],
+        "data":[prompt_text, "انگلیسی (آمریکا) - جنی (زن)", 0, 0, 0],
         "fn_index": 1,
         "session_hash": session_hash
     }
@@ -748,7 +748,7 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
             if proc_msg:
                 msg_id = getattr(proc_msg, 'message_id', None)
                 if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-                if msg_id: await client.delete_messages(chat_id, [msg_id])
+                if msg_id: await client.delete_messages(chat_id,[msg_id])
         except Exception: pass
 
         if audio_bytes:
@@ -775,7 +775,7 @@ async def process_podcast(client, chat_id, prompt):
         return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت پادکست شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     proc_msg = await send_with_keyboard(client, chat_id, "📻 در حال بررسی موضوع و نگارش سناریوی پادکست توسط هوش مصنوعی...\n(لطفاً صبور باشید)", False)
-    available_speakers = []
+    available_speakers =[]
     for num_key, (spk_name, spk_id) in SPEAKERS.items():
         gender = "male" if "مرد" in spk_name else "female"
         available_speakers.append({"id": spk_id, "name": spk_name.split(' (')[0], "gender": gender})
@@ -799,7 +799,7 @@ async def process_podcast(client, chat_id, prompt):
                     if resp.status == 200:
                         status_data = await resp.json()
                         if status_data.get("status") == "completed":
-                            script_data = status_data.get("data", {}).get("script", [])
+                            script_data = status_data.get("data", {}).get("script",[])
                             break
                         elif status_data.get("status") == "failed":
                             return await send_with_keyboard(client, chat_id, "❌ متأسفانه هوش مصنوعی نتوانست برای این موضوع سناریو بنویسد.", True)
@@ -880,7 +880,7 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     async with aiohttp.ClientSession() as session:
         for key in keys_to_try:
             url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
-            payload = {"contents": [{"parts": [{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}], "generationConfig": {"temperature": 0.2}}
+            payload = {"contents": [{"parts":[{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}], "generationConfig": {"temperature": 0.2}}
             try:
                 async with session.post(url, json=payload, timeout=60) as response:
                     if response.status == 200:
@@ -927,7 +927,7 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     async with aiohttp.ClientSession() as session:
         for key in keys_to_try:
             url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
-            payload = {"contents": [{"parts": [{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}], "generationConfig": {"temperature": 0.6}}
+            payload = {"contents":[{"parts":[{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}], "generationConfig": {"temperature": 0.6}}
             try:
                 async with session.post(url, json=payload, timeout=45) as response:
                     if response.status == 200:
@@ -963,6 +963,13 @@ else:
     async def main_handler(client, update):
         global BOT_GUID
         
+        current_time = time.time()
+        
+        # 🛡 سد دفاعی اول: فریز ۳۰ ثانیه‌ای کامل 🛡
+        # در ۳۰ ثانیه اول راه‌اندازی، هیچ پیامی تحت هیچ شرایطی پردازش نمی‌شود
+        if current_time - BOT_STARTUP_TIME < 30:
+            return
+            
         try:
             # 🛡 دریافت آیدی یکتای ربات برای جلوگیری از لوپ 🛡
             if not BOT_GUID:
@@ -975,7 +982,7 @@ else:
 
             msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
             
-            # --- استخراج دقیق شناسه فرستنده (برای جلوگیری از حرف زدن ربات با خودش) ---
+            # --- استخراج دقیق شناسه فرستنده ---
             author_id = getattr(update, 'author_guid', None)
             if not author_id and msg_obj:
                 author_id = msg_obj.get('author_object_guid') if isinstance(msg_obj, dict) else getattr(msg_obj, 'author_object_guid', None)
@@ -984,7 +991,8 @@ else:
             if BOT_GUID and author_id == BOT_GUID:
                 return
 
-            # --- استخراج زمان دقیق پیام برای جلوگیری از اسپم شدن ربات بعد از ری‌استارت (Backlog) ---
+            # 🛡 سد دفاعی دوم: جلوگیری کامل از تلنبار شدن پیام‌های قدیمی (Backlog) 🛡
+            # حتی اگر ۳۰ ثانیه گذشته باشد، اگر کاربر این پیام را در زمان خاموشی سرور داده باشد، درجا حذف می‌شود
             msg_time = getattr(update, 'timestamp', None)
             if not msg_time and msg_obj:
                 msg_time = msg_obj.get('timestamp') if isinstance(msg_obj, dict) else getattr(msg_obj, 'timestamp', None)
@@ -992,11 +1000,11 @@ else:
             if msg_time:
                 try:
                     msg_time_int = int(msg_time)
-                    if msg_time_int > 1e12: # اگر زمان به میلی‌ثانیه است، تبدیل به ثانیه شود
+                    if msg_time_int > 1e12: # تبدیل میلی‌ثانیه به ثانیه اگر نیاز بود
                         msg_time_int = msg_time_int / 1000.0
                         
                     if msg_time_int < BOT_STARTUP_TIME:
-                        # ⛔️ پیام مربوط به گذشته است و دور ریخته می‌شود
+                        # ⛔️ پیام متعلق به گذشته است و برای همیشه دور ریخته می‌شود
                         return
                 except Exception:
                     pass
@@ -1017,8 +1025,6 @@ else:
                 processed_message_ids.add(unique_msg_key)
                 if len(processed_message_ids) > 5000: processed_message_ids.clear()
             
-            current_time = time.time()
-            
             # 🛡 سیستم محدودکننده سرعت کاربر (Rate Limit) 🛡
             last_req_time = user_last_request_time.get(chat_id, 0)
             if current_time - last_req_time < 1.0:
@@ -1030,12 +1036,12 @@ else:
             user_text_str = str(user_text).strip() if user_text else ""
             user_text_lower = user_text_str.lower()
             
-            # ⛔️ لایه امنیتی دوم برای جلوگیری از پاسخ دادن ربات به پیام‌های سیستم خودش
+            # ⛔️ لایه امنیتی سوم برای جلوگیری از پاسخ دادن ربات به پیام‌های سیستم خودش
             if user_text_str and user_text_str.startswith(("⏳", "❌", "✅", "🧠", "🎨", "🪄", "🎙", "📻", "📝", "👁️", "💡", "📥")):
                 return
 
             if chat_id not in user_states:
-                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
+                user_states[chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None}
 
             # 🛠 --- سیستم پنل مدیریت (بدون نیاز به لاگین) --- 🛠
             if user_text_lower.startswith(f"{ADMIN_CODE} pro=") or user_text_lower.startswith(f"{ADMIN_CODE}pro="):
@@ -1146,7 +1152,7 @@ else:
             file_name = f"unknown_file_{uuid.uuid4().hex[:6]}.jpg"
             
             if msg_obj:
-                for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
+                for attr in['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
                     file_attr = getattr(msg_obj, attr, None)
                     if file_attr:
                         is_file = True
@@ -1155,14 +1161,14 @@ else:
                 
                 if not is_file and hasattr(msg_obj, 'to_dict'):
                     msg_dict = msg_obj.to_dict()
-                    for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
+                    for attr in['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
                         if attr in msg_dict:
                             is_file = True
                             file_attr = msg_dict.get(attr, {})
                             file_name = file_attr.get('file_name', file_name) if isinstance(file_attr, dict) else file_name
                             break
 
-            if user_text_str in ["/start", "سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
+            if user_text_str in["/start", "سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
                 user_states[chat_id]["mode"] = None
                 user_states[chat_id]["file_bytes"] = None
                 await send_with_keyboard(client, chat_id, "سلام! به ربات هوش مصنوعی آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:", True)
@@ -1252,7 +1258,7 @@ else:
                 return
 
             # --- دکمه انتقال اکانت ---
-            if user_text_str in ["/transfer", "انتقال اکانت از برنامه به ربات"]:
+            if user_text_str in["/transfer", "انتقال اکانت از برنامه به ربات"]:
                 transfer_text = f"""🔄 **انتقال اکانت از برنامه به ربات**
 
 کاربر گرامی، در صورتی که داخل برنامه «هوش مصنوعی آلفا» پیش‌تر اشتراک تهیه کرده‌اید، نیازی به خرید مجدد اشتراک داخل ربات نیست! 🎉
@@ -1268,7 +1274,7 @@ else:
 
             if user_text_str in ["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
                 user_states[chat_id]["mode"] = "chat"
-                user_states[chat_id]["history"] = []
+                user_states[chat_id]["history"] =[]
                 await send_with_keyboard(client, chat_id, "💬 شما وارد بخش **چت با هوش مصنوعی** شدید.\n\nهر سوالی دارید بفرستید تا جواب بدم:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
@@ -1277,13 +1283,13 @@ else:
                 await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس پیشرفته** شدید.\n\nمتن خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
-            if user_text_str in ["/edit_image", "ویرایش تصاویر 🪄"]:
+            if user_text_str in["/edit_image", "ویرایش تصاویر 🪄"]:
                 user_states[chat_id]["mode"] = "image_edit_waiting_for_image"
                 user_states[chat_id]["file_bytes"] = None
                 await send_with_keyboard(client, chat_id, "🪄 به بخش **ویرایش عکس (Flux.2)** خوش آمدید.\n\nلطفاً ابتدا عکسی که می‌خواهید ویرایش کنید را بفرستید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
-            if user_text_str in ["/tts", "🎙️ صدا", "تبدیل متن به صدا🗣️"]:
+            if user_text_str in["/tts", "🎙️ صدا", "تبدیل متن به صدا🗣️"]:
                 user_states[chat_id]["mode"] = "tts_waiting_for_text"
                 await send_with_keyboard(client, chat_id, "🎙️ شما وارد بخش **تبدیل متن به صدا** شدید.\n\nلطفاً متنی که می‌خواهید به صدا تبدیل شود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
@@ -1299,7 +1305,7 @@ else:
                 await send_with_keyboard(client, chat_id, "📁 شما وارد بخش **تحلیل فایل اختصاصی** شدید.\n\nلطفاً فایل خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
-            if user_text_str in ["/stt", "فایل صوتی به متن 📝"]:
+            if user_text_str in["/stt", "فایل صوتی به متن 📝"]:
                 user_states[chat_id]["mode"] = "stt_waiting_for_audio"
                 await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل خود (ویس، آهنگ، ویدیو و...) را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
@@ -1418,5 +1424,5 @@ else:
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
     if bot_token:
-        print("ربات آلفا پرو با سیستم اشتراک نامحدود روشن شد...")
-        bot.run()
+        print("ربات آلفا پرو با سیستم اشتراک نامحدود و دیواره امنیتی ضد Backlog روشن شد...")
+        bot.run()