Update main.py

main.py

@@ -704,67 +704,9 @@ 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: تلاش برای آپلود در بله (ساخت لینک)
@@ -799,27 +741,56 @@ 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:
-                                        # 🟢 ایجاد تسک کارگر برای آپلود مستقیم و بی‌صدا در روبیکا در پس‌زمینه (بدون اختلال)
-                                        asyncio.create_task(background_rubika_upload(client, chat_id, abs_path, file_type, caption, True))
-                                        return 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)
+                                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: سیستم نجات! (آپلود مستقیم در روبیکا)
-    # (اگر بله قطع بود، تابع اصلی متوقف می‌شود تا فایل را مستقیماً ارسال کند)
+    # (اگر بله قطع بود، فایل مستقیما تو روبیکا ارسال میشه)
     # ==========================================
-    success = await background_rubika_upload(client, chat_id, abs_path, file_type, caption, False)
-    if success:
-        return True
+    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)
 
     return "\n".join(error_logs[-5:])