Update main.py

main.py

@@ -704,9 +704,67 @@ HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 # ==============================================================================
 BALE_BOT_TOKEN = os.environ.get("BALE_BOT_TOKEN", "").strip()
 
+async def background_rubika_upload(client, chat_id, original_path, file_type, caption, is_background=True):
+    """کارگر جداگانه برای آپلود فایل در روبیکا بدون ایجاد تاخیر در ربات اصلی"""
+    target_path = original_path
+    
+    if is_background:
+        # کپی کردن فایل برای کارگر بک‌گراند (چون فایل اصلی توسط تابع والد بلافاصله حذف می‌شود)
+        target_path = f"{original_path}_bg_{uuid.uuid4().hex[:6]}.tmp"
+        try:
+            with open(original_path, 'rb') as f_src, open(target_path, 'wb') as f_dst:
+                f_dst.write(f_src.read())
+        except Exception:
+            return False
+
+    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 target_path.lower().endswith('.ogg'): api_file_type = "Music"
+    if api_file_type == "File" and target_path.lower().endswith('.mp3'): api_file_type = "Music"
+    
+    # اگر آپلود در بک‌گراند (همزمان با لینک بله) است، کپشن ساده باشد وگرنه متن نجات
+    final_caption = "📥 فایل ضمیمه:" if is_background else f"{caption}\n\n⚠️ (به دلیل اختلال سرورهای بله، فایل موقتاً به صورت مستقیم در روبیکا ارسال شد)"
+
+    sent_success = False
+    for attempt in range(5):
+        try:
+            if api_file_type == "Image" and hasattr(client, "send_photo"):
+                try: await client.send_photo(chat_id, photo=target_path, caption=final_caption); sent_success = True
+                except: await client.send_photo(chat_id, target_path, caption=final_caption); sent_success = True
+            elif api_file_type == "Voice" and hasattr(client, "send_voice"):
+                try: await client.send_voice(chat_id, target_path, caption=final_caption); sent_success = True
+                except: await client.send_voice(chat_id, file=target_path, caption=final_caption); sent_success = True
+            elif api_file_type == "Music" and hasattr(client, "send_music"):
+                try: await client.send_music(chat_id, target_path, caption=final_caption); sent_success = True
+                except: await client.send_music(chat_id, music=target_path, caption=final_caption); sent_success = True
+            
+            if not sent_success:
+                if hasattr(client, "send_document"):
+                    try: await client.send_document(chat_id, document=target_path, caption=final_caption); sent_success = True
+                    except: await client.send_document(chat_id, target_path, caption=final_caption); sent_success = True
+                elif hasattr(client, "send_file"):
+                    try: await client.send_file(chat_id, file=target_path)
+                    except: await client.send_file(chat_id, target_path)
+                    if final_caption: await send_with_keyboard(client, chat_id, final_caption, True)
+                    sent_success = True
+                    
+            if sent_success:
+                break
+        except Exception:
+            await asyncio.sleep(3)
+
+    # پاک کردن فایل موقتِ کارگر
+    if is_background:
+        try:
+            if os.path.exists(target_path):
+                os.remove(target_path)
+        except:
+            pass
+            
+    return sent_success
+
 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 = []
     
     # ==========================================
     # فاز 1: تلاش برای آپلود در بله (ساخت لینک)
@@ -741,56 +799,27 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                     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
+                                    
+                                    if send_res:
+                                        # 🟢 ایجاد تسک کارگر برای آپلود مستقیم و بی‌صدا در روبیکا در پس‌زمی��ه (بدون اختلال)
+                                        asyncio.create_task(background_rubika_upload(client, chat_id, abs_path, file_type, caption, True))
+                                        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)
+                                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)
 
     # ==========================================
     # فاز 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:
-            sent_success = False
-            
-            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
-                
-        except Exception as e:
-            error_logs.append(f"Rubpy Send Err: {str(e)[:100]}")
-            await asyncio.sleep(3)
+    success = await background_rubika_upload(client, chat_id, abs_path, file_type, caption, False)
+    if success:
+        return True
 
     return "\n".join(error_logs[-5:])