Update main.py

main.py

@@ -698,134 +698,116 @@ HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 # ==============================================================================
-# 🟢 پارت 11: آپلودر فوق‌سریع و مستقیم (نسخه صنعتی چندرشته‌ای - 15 ورکر همزمان)
+# 🟢 پارت 11: آپلودر فوق‌سریع و مستقیم (نسخه مقاوم در برابر ترافیک سنگین)
 # ==============================================================================
 import asyncio
 import os
 import aiohttp
+import uuid
 
-# 🟢 سیستم توزیع بار (Load Balancer): اجازه آپلود همزمان به 15 کاربر (به جای 2 کاربر)
-# این همان اضافه کردن ورکرهایی است که درخواست دادید!
+# 🟢 مدیریت صف آپلود: اجازه پردازش همزمان ۱۵ فایل
 _upload_lock = None
 def get_upload_lock():
     global _upload_lock
     if _upload_lock is None:
-        _upload_lock = asyncio.Semaphore(15) # 🚀 ارتقا به 15 ورکـر همزمان برای آپلود سریع
+        _upload_lock = asyncio.Semaphore(15) 
     return _upload_lock
 
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
+    if not os.path.exists(abs_path):
+        return "File Not Found"
+
     error_logs = []
+    bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
     
+    # تعیین نوع دقیق فایل برای سرور روبیکا
     api_file_type = "Image" if file_type in ["photo", "Image", "image"] else "Voice" if file_type in ["voice", "Voice", "audio"] else file_type
-    
     if api_file_type == "Voice" and not abs_path.lower().endswith('.ogg'): 
         api_file_type = "Music"
     if api_file_type == "File" and abs_path.lower().endswith('.mp3'): 
         api_file_type = "Music"
         
-    bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
-
-    # 🟢 ورود به تونل پرسرعت آپلود (همزمان 15 نفر بدون تداخل عبور می‌کنند)
     async with get_upload_lock():
-        
-        # 🚀 فاز اصلی: استفاده 100٪ از API خام و پرسرعت روبیکا (بدون استفاده از کدهای کند rubpy)
-        for attempt in range(8): # 8 بار تلاش برق‌آسا
+        # 🚀 تلاش برای آپلود مستقیم (Bypass Rubpy Latency)
+        for attempt in range(10): 
             try:
-                url_request = f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile"
-                url_send = f"https://botapi.rubika.ir/v3/{bot_token}/sendFile"
-                
-                # تایم‌اوت‌های کوتاه برای جلوگیری از گیر کردن ربات روی یک فایل خراب
-                timeout_cfg = aiohttp.ClientTimeout(total=45) 
+                # تنظیم تایم‌آوت بالا برای فایل‌های حجیم در ترافیک سنگین
+                timeout_cfg = aiohttp.ClientTimeout(total=300, connect=30) 
                 async with aiohttp.ClientSession(timeout=timeout_cfg) as session:
-                    # 1. گرفتن لینک پرسرعت آپلود از روبیکا
-                    async with session.post(url_request, json={"type": api_file_type}) as resp:
-                        if resp.status != 200:
-                            await asyncio.sleep(1)
-                            continue
-                            
+                    
+                    # مرحله ۱: درخواست لینک آپلود
+                    async with session.post(
+                        f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile", 
+                        json={"type": api_file_type},
+                        timeout=30
+                    ) as resp:
                         req_data = await resp.json()
                         
                         if req_data.get("status") == "TOO_REQUESTS":
-                            error_logs.append("TOO_REQUESTS_REQ")
-                            await asyncio.sleep(3) # 👈 خواب‌های 15 ثانیه‌ای را به 3 ثانیه کاهش دادیم
+                            await asyncio.sleep(5)
                             continue
 
                         if req_data.get("status") == "OK":
                             upload_url = req_data.get("data", {}).get("upload_url")
                             if not upload_url: continue
 
-                            # 2. شلیک مستقیم فایل به سرور ابری روبیکا
+                            # مرحله ۲: آپلود فیزیکی فایل با بافرینگ مناسب
                             with open(abs_path, "rb") as f:
                                 form = aiohttp.FormData()
                                 form.add_field('file', f, filename=os.path.basename(abs_path))
+                                
                                 async with session.post(upload_url, data=form) as up_resp:
-                                    if up_resp.status != 200:
-                                        await asyncio.sleep(1)
-                                        continue
-                                        
                                     up_data = await up_resp.json()
                                     
                                     if up_data.get("status") == "TOO_REQUESTS":
-                                        await asyncio.sleep(3)
+                                        await asyncio.sleep(5)
                                         continue
 
                                     final_file_id = None
-                                    
-                                    if up_data.get("status") == "INVALID_INPUT" and "format" in str(up_data).lower():
-                                        api_file_type = "File"
-                                        await asyncio.sleep(1)
-                                        continue
-
                                     if up_data.get("status") in ["OK", "OK_DET"]:
-                                        if "data" in up_data and isinstance(up_data["data"], dict):
-                                            final_file_id = up_data["data"].get("file_id") or up_data["data"].get("id")
-                                        elif "file_id" in up_data: 
-                                            final_file_id = up_data.get("file_id")
+                                        data_part = up_data.get("data", {})
+                                        final_file_id = data_part.get("file_id") or data_part.get("id") or up_data.get("file_id")
                                             
-                                    # 3. ارسال فایل به پی‌وی کاربر در کسری از ثانیه
+                                    # مرحله ۳: نهایی‌سازی و ارسال پیام به کاربر
                                     if final_file_id:
                                         send_payload = {
                                             "chat_id": str(chat_id),
                                             "file_id": str(final_file_id),
                                             "text": caption,
                                             "chat_keypad_type": "New",
-                                            "chat_keypad": MAIN_KEYPAD_DICT
+                                            "chat_keypad": MAIN_KEYPAD_DICT # از متغیر سراسری استفاده می‌کند
                                         }
-                                        async with session.post(url_send, json=send_payload) as send_resp:
+                                        async with session.post(
+                                            f"https://botapi.rubika.ir/v3/{bot_token}/sendFile", 
+                                            json=send_payload,
+                                            timeout=30
+                                        ) as send_resp:
                                             s_data = await send_resp.json()
-                                            if s_data.get("status") == "TOO_REQUESTS":
-                                                await asyncio.sleep(3)
-                                                continue
                                             if s_data.get("status") == "OK": 
                                                 return True
+                                            elif s_data.get("status") == "TOO_REQUESTS":
+                                                await asyncio.sleep(5)
+                                                continue
             except Exception as e:
-                error_logs.append(f"Fast HTTP Error: {str(e)[:30]}")
-                await asyncio.sleep(1.5)
-
-        # --- فاز دوم (سنگر آخر): اگر API خام مسدود بود، فایل را از طریق روب‌پای می‌فرستیم ---
-        for attempt in range(3):
-            try:
-                if api_file_type == "Image" and hasattr(client, "send_photo"):
-                    try:
-                        await client.send_photo(chat_id, photo=abs_path, caption=caption)
-                        return True
-                    except TypeError:
-                        await client.send_photo(chat_id, abs_path, caption=caption)
-                        return True
-                else:
-                    if hasattr(client, "send_document"):
-                        try:
-                            await client.send_document(chat_id, document=abs_path, caption=caption)
-                            return True
-                        except TypeError:
-                            await client.send_document(chat_id, abs_path, caption=caption)
-                            return True
-            except Exception as e:
-                error_logs.append(f"Fallback Error: {str(e)[:30]}")
+                error_logs.append(f"Attempt {attempt+1}: {str(e)[:40]}")
                 await asyncio.sleep(2)
 
-        return "\n".join(error_logs[-5:])
+        # 🛡️ سیستم پشتیبان (Fallback): اگر آپلود مستقیم شکست خورد، از متد خود کتابخانه استفاده کن
+        try:
+            if api_file_type == "Image":
+                await client.send_photo(chat_id, photo=abs_path, caption=caption)
+                return True
+            elif api_file_type in ["Music", "Voice"]:
+                await client.send_music(chat_id, abs_path, caption=caption)
+                return True
+            else:
+                await client.send_file(chat_id, abs_path)
+                return True
+        except Exception as e:
+            error_logs.append(f"Fallback Fail: {str(e)[:40]}")
+
+    return "❌ خطا در آپلود پس از چندین تلاش:\n" + "\n".join(error_logs[-3:])
 
 # ==============================================================================
 # 🟢 پارت 12: توابع تغییر صدا و لیست گویندگان