Update main.py

main.py

@@ -610,89 +610,99 @@ HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 # ==============================================================================
-# 🟢 پارت 11: آپلودر قدرتمند به سرورهای بله و ارسال لینک دانلود به روبیکا
+# 🟢 پارت 11: آپلودر دوگانه هوشمند (آپلود در بله + بکاپ خودکار در روبیکا)
 # ==============================================================================
-# فقط توکن ربات بله نیاز است (آیدی چت از روی همین توکن استخراج می‌شود)
 BALE_BOT_TOKEN = os.environ.get("BALE_BOT_TOKEN", "").strip()
 
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
     error_logs =[]
     
-    if not BALE_BOT_TOKEN:
-        return "❌ توکن ربات بله (BALE_BOT_TOKEN) در متغیرهای محیطی (Secrets) تنظیم نشده است."
+    # ==========================================
+    # فاز 1: تلاش برای آپلود در بله (ساخت لینک)
+    # ==========================================
+    if BALE_BOT_TOKEN:
+        bale_chat_id = BALE_BOT_TOKEN.split(":")[0]
+        for attempt in range(3): # ۳ بار تلاش در سرور بله
+            try:
+                bale_url = f"https://tapi.bale.ai/bot{BALE_BOT_TOKEN}/sendDocument"
+                with open(abs_path, "rb") as f:
+                    form = aiohttp.FormData()
+                    form.add_field('chat_id', bale_chat_id)
+                    form.add_field('document', f, filename=os.path.basename(abs_path))
+                    
+                    async with aiohttp.ClientSession() as session:
+                        async with session.post(bale_url, data=form, timeout=120) as resp:
+                            if resp.status == 200:
+                                bale_data = await resp.json()
+                                if bale_data.get("ok"):
+                                    try:
+                                        file_id = bale_data['result']['document']['file_id']
+                                    except KeyError:
+                                        if 'audio' in bale_data['result']: file_id = bale_data['result']['audio']['file_id']
+                                        elif 'video' in bale_data['result']: file_id = bale_data['result']['video']['file_id']
+                                        elif 'voice' in bale_data['result']: file_id = bale_data['result']['voice']['file_id']
+                                        elif 'photo' in bale_data['result']: file_id = bale_data['result']['photo'][-1]['file_id']
+                                        else:
+                                            error_logs.append("فرمت در بله شناخته نشد.")
+                                            continue
+
+                                    download_url = f"https://tapi.bale.ai/file/bot{BALE_BOT_TOKEN}/{file_id}"
+                                    final_text = f"""{caption}\n\n━━━━━━━━━━━━━━━━━━━\n🌐 لینک دانلود فایل شما:\n\n{download_url}\n\n━━━━━━━━━━━━━━━━━━━\n⚠️ **راهنمای دانلود:**\nلطفاً لینک بالا را کپی کرده و در **مرورگر گوشی خود** باز کنید.\n*(مستقیماً کلیک نکنید)*"""
+                                    
+                                    send_res = await send_with_keyboard(client, chat_id, final_text, True)
+                                    if send_res: return True
+                                else:
+                                    error_logs.append(f"Bale API Err: {bale_data.get('description')}")
+                            else:
+                                error_logs.append(f"Bale HTTP Error: {resp.status}")
+                                if resp.status in[500, 502, 503]: await asyncio.sleep(2)
+            except Exception as e:
+                error_logs.append(f"Bale Err: {str(e)[:100]}")
+                await asyncio.sleep(2)
 
-    # استخراج آیدی عددی ربات از توکن برای ارسال فایل به فضای شخصی خود ربات (Self-Chat)
-    bale_chat_id = BALE_BOT_TOKEN.split(":")[0]
+    # ==========================================
+    # فاز 2: سیستم نجات! (آپلود مستقیم در روبیکا)
+    # (اگر بله قطع بود، فایل مستقیما تو روبیکا ارسال میشه)
+    # ==========================================
+    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"
+    
+    fallback_caption = f"{caption}\n\n⚠️ (به دلیل اختلال سرورهای بله، فایل موقتاً به صورت مستقیم در روبیکا ارسال شد)"
 
     for attempt in range(5):
         try:
-            bale_url = f"https://tapi.bale.ai/bot{BALE_BOT_TOKEN}/sendDocument"
+            sent_success = False
             
-            with open(abs_path, "rb") as f:
-                form = aiohttp.FormData()
-                form.add_field('chat_id', bale_chat_id)
-                form.add_field('document', f, filename=os.path.basename(abs_path))
+            if api_file_type == "Image" and hasattr(client, "send_photo"):
+                try: await client.send_photo(chat_id, photo=abs_path, caption=fallback_caption); sent_success = True
+                except: await client.send_photo(chat_id, abs_path, caption=fallback_caption); sent_success = True
+            elif api_file_type == "Voice" and hasattr(client, "send_voice"):
+                try: await client.send_voice(chat_id, abs_path, caption=fallback_caption); sent_success = True
+                except: await client.send_voice(chat_id, file=abs_path, caption=fallback_caption); sent_success = True
+            elif api_file_type == "Music" and hasattr(client, "send_music"):
+                try: await client.send_music(chat_id, abs_path, caption=fallback_caption); sent_success = True
+                except: await client.send_music(chat_id, music=abs_path, caption=fallback_caption); sent_success = True
+            
+            if not sent_success:
+                if hasattr(client, "send_document"):
+                    try: await client.send_document(chat_id, document=abs_path, caption=fallback_caption); sent_success = True
+                    except: await client.send_document(chat_id, abs_path, caption=fallback_caption); sent_success = True
+                elif hasattr(client, "send_file"):
+                    try: await client.send_file(chat_id, file=abs_path)
+                    except: await client.send_file(chat_id, abs_path)
+                    if fallback_caption: await send_with_keyboard(client, chat_id, fallback_caption, True)
+                    sent_success = True
+                    
+            if sent_success:
+                return True
                 
-                async with aiohttp.ClientSession() as session:
-                    async with session.post(bale_url, data=form, timeout=120) as resp:
-                        if resp.status == 200:
-                            bale_data = await resp.json()
-                            if bale_data.get("ok"):
-                                # استخراج file_id از پاسخ بله
-                                try:
-                                    file_id = bale_data['result']['document']['file_id']
-                                except KeyError:
-                                    if 'audio' in bale_data['result']:
-                                        file_id = bale_data['result']['audio']['file_id']
-                                    elif 'video' in bale_data['result']:
-                                        file_id = bale_data['result']['video']['file_id']
-                                    elif 'voice' in bale_data['result']:
-                                        file_id = bale_data['result']['voice']['file_id']
-                                    elif 'photo' in bale_data['result']:
-                                        file_id = bale_data['result']['photo'][-1]['file_id']
-                                    else:
-                                        error_logs.append("Format not recognized in Bale response.")
-                                        await asyncio.sleep(2)
-                                        continue
-
-                                # ساخت لینک دانلود بله
-                                download_url = f"https://tapi.bale.ai/file/bot{BALE_BOT_TOKEN}/{file_id}"
-                                
-                                # ترکیب کپشن اصلی شما با متن درخواستی برای لینک
-                                final_text = f"""{caption}
-
-━━━━━━━━━━━━━━━━━━━
-🌐 لینک دانلود فایل شما:
-
-{download_url}
-
-━━━━━━━━━━━━━━━━━━━
-⚠️ **راهنمای دانلود:**
-لطفاً لینک بالا را کپی کرده و در **مرورگر گوشی خود (مثل کروم یا فایرفاکس)** باز کنید تا فایلتان دانلود شود. 
-*(لطفاً مستقیماً از اینجا روی لینک کلیک نکنید، چون روبیکا اجازه باز کردن و دانلود این نوع فایل‌ها را نمی‌دهد)*.
-
-🇮🇷 این لینک با دامنه سایت «بله» ایجاد شده و می‌توانید فایل خود را به راحتی با **اینترنت داخلی (ملی)** دانلود کنید.
-
-💡 **نکته:** در صورتی که محدودیت‌های سرور روبیکا در دریافت فا��ل برطرف شود، سیستم را تغییر می‌دهیم تا در آینده بتوانید فایل‌ها را مستقیماً داخل همین ربات دریافت کنید."""
-
-                                # ارسال متن حاوی لینک به کاربر در روبیکا
-                                send_res = await send_with_keyboard(client, chat_id, final_text, True)
-                                if send_res:
-                                    return True
-                                else:
-                                    error_logs.append("Failed to send text to Rubika.")
-                            else:
-                                error_logs.append(f"Bale API Error: {bale_data.get('description')}")
-                        else:
-                            error_logs.append(f"Bale HTTP Error: {resp.status}")
-                            if resp.status in [500, 502, 503]:
-                                await asyncio.sleep(3)
         except Exception as e:
-            error_logs.append(f"Upload logic Error: {str(e)[:150]}")
-            await asyncio.sleep(2.5)
+            error_logs.append(f"Rubpy Send Err: {str(e)[:100]}")
+            await asyncio.sleep(3)
 
-    return "\n".join(error_logs[-3:])
+    return "\n".join(error_logs[-5:])
 
 # ==============================================================================
 # 🟢 پارت 12: توابع تغییر صدا و لیست گویندگان