Update main.py

main.py

@@ -29,14 +29,12 @@ BOT_GUID = None
 
 # =======================================================
 # 🔥 سیستم کنترل سرعت و ضد رگبار (Burst Controller) 🔥
-# جلوگیری از هنگ کردن ربات هنگام دریافت پیام‌های تلنبار شده
 # =======================================================
 global_burst_count = 0
 global_burst_time = time.time()
 burst_lock = threading.Lock()
 
 def is_backlog_burst():
-    """اگر بیشتر از 5 پیام در 1 ثانیه بیاید، یعنی رگبار از سمت سرور است و رد می‌شود"""
     global global_burst_count, global_burst_time
     with burst_lock:
         now = time.time()
@@ -46,17 +44,16 @@ def is_backlog_burst():
             return False
         else:
             global_burst_count += 1
-            if global_burst_count > 5:
+            if global_burst_count > 60:
                 return True
             return False
 
 # --- سیستم دیتابیس حساب کاربری متصل به دیتاست هاگینگ فیس ---
 DB_FILE = "users_db.json"
 DATASET_REPO = "Opera8/Karbaran-rayegan-tedad"
-HF_TOKEN_DB = os.environ.get("HF_TOKEN") # توکنی که برای دیتاست تنظیم کرده‌اید
+HF_TOKEN_DB = os.environ.get("HF_TOKEN")
 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]
     gy2 = (gy + 1) if (gm > 2) else gy
@@ -124,12 +121,10 @@ def save_db(db_data):
 
 user_credits_db = load_db()
 
-# --- سیستم تولید و یافتن کد هدیه 8 رقمی یکتا ---
 def get_or_create_referral_code(chat_id):
     user_data = user_credits_db[chat_id]
     if not user_data.get("referral_code"):
         while True:
-            # فقط از اعداد برای تولید کد 8 رقمی استفاده می‌کنیم
             new_code = ''.join(random.choices(string.digits, k=8))
             exists = any(u.get("referral_code") == new_code for u in user_credits_db.values())
             if not exists:
@@ -195,7 +190,6 @@ def get_user_credits(chat_id):
             
     return user_data
 
-# --- تابع تبدیل اعداد فارسی/عربی به انگلیسی ---
 def to_english_digits(text):
     if not text:
         return text
@@ -215,7 +209,6 @@ def home():
 def run_flask():
     app.run(host="0.0.0.0", port=7860, threaded=True)
 
-
 # --- توابع کمکی برای جلوگیری از بلاک شدن Event Loop ---
 def sync_save_image(image, file_name):
     rgb_im = image.convert('RGB')
@@ -236,8 +229,7 @@ def sync_combine_audio(current_audio, new_bytes):
 def sync_export_audio(audio_obj, file_name):
     audio_obj.export(file_name, format="mp3")
 
-
-# --- ساختار کیبورد آپدیت شده ---
+# --- ساختار کیبورد ---
 MAIN_KEYPAD_DICT = {
     "rows":[
         {
@@ -352,7 +344,6 @@ async def check_channel_membership(client, user_id):
     except Exception:
         return False
 
-
 async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
     try:
         if not use_keyboard:
@@ -372,56 +363,112 @@ async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
             return None
 
 
+bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
+
+# ==================================================================
+# 🔥 تابع دانلود کاملاً ضد بمب اتم (35 بار پافشاری و کنترل 502)
+# ==================================================================
 async def helper_download_file(client, msg_obj):
     errors =[]
-    file_id = None
     file_obj = None
+    file_id = None
+    
+    # 1. استخراج آبجکت و آیدی فایل با دقت بالا
     for attr in['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
         val = getattr(msg_obj, attr, None)
         if val:
             file_obj = val
-            if hasattr(val, 'file_id'): file_id = val.file_id
-            elif isinstance(val, dict) and 'file_id' in val: file_id = val['file_id']
-            if file_id: break
+            if hasattr(val, 'file_id'): 
+                file_id = val.file_id
+            elif isinstance(val, dict) and 'file_id' in val: 
+                file_id = val['file_id']
+            break
+            
+    if not file_obj and hasattr(msg_obj, "file_id"):
+        file_id = msg_obj.file_id
+        file_obj = msg_obj
+
+    if not file_id:
+        raise Exception("خطا: هیچ فایلی در پیام یافت نشد.")
 
     temp_name = f"temp_dl_{uuid.uuid4().hex}.tmp"
 
-    if hasattr(client, "download_file"):
+    # روش 1: استفاده مستقیم از API قدرتمند روبیکا با پافشاری بسیار بالا (20 بار)
+    for attempt in range(20):
         try:
-            await client.download_file(msg_obj, temp_name)
-            if os.path.exists(temp_name):
-                data = await asyncio.to_thread(sync_read_file, temp_name)
-                os.remove(temp_name)
-                return data
-        except Exception as e: errors.append(f"تلاش ۱: {e}")
-
-        if file_obj:
-            try:
-                await client.download_file(file_obj, temp_name)
-                if os.path.exists(temp_name):
-                    data = await asyncio.to_thread(sync_read_file, temp_name)
-                    os.remove(temp_name)
-                    return data
-            except Exception as e: errors.append(f"تلاش ۲: {e}")
+            url_get_file = f"https://botapi.rubika.ir/v3/{bot_token}/getFile"
+            payload = {"file_id": str(file_id)}
+            headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"}
+            
+            async with aiohttp.ClientSession() as session:
+                async with session.post(url_get_file, json=payload, headers=headers, timeout=30) as resp:
+                    if resp.status == 200:
+                        res_data = await resp.json()
+                        if res_data.get("status") == "OK" or res_data.get("status_det") == "OK":
+                            download_url = res_data.get("data", {}).get("download_url") or res_data.get("data", {}).get("file_url")
+                            if download_url:
+                                # اگر لینک داد، 3 بار هم برای خود لینک تلاش میکنیم تا اگر قطع شد جبران شود
+                                for dl_attempt in range(3):
+                                    try:
+                                        async with session.get(download_url, headers=headers, timeout=60) as dl_resp:
+                                            if dl_resp.status == 200:
+                                                data = await dl_resp.read()
+                                                if data and len(data) > 0:
+                                                    return data
+                                            else:
+                                                errors.append(f"DL HTTP {dl_resp.status}")
+                                            await asyncio.sleep(2)
+                                    except Exception as dl_err:
+                                        errors.append(f"DL Error: {str(dl_err)[:30]}")
+                                        await asyncio.sleep(2)
+                            else:
+                                errors.append("No download URL in response.")
+                        else:
+                            errors.append(f"API Not OK: {res_data.get('status')}")
+                    elif resp.status in[502, 503, 500]:
+                        errors.append(f"GetFile HTTP {resp.status}")
+                        await asyncio.sleep(3.5) # صبر بیشتر برای خطای 502
+                        continue
+                    else:
+                        errors.append(f"GetFile HTTP {resp.status}")
+                        await asyncio.sleep(2)
+        except Exception as e:
+            errors.append(f"API Error: {str(e)[:50]}")
+            await asyncio.sleep(2)
 
-        if file_id:
+    # روش 2: کتابخانه rubpy (بک‌آپ با 10 بار پافشاری)
+    if file_obj:
+        for attempt in range(10):
             try:
+                if hasattr(client, "download"):
+                    result = await client.download(file_obj, save_as=temp_name)
+                    if isinstance(result, bytes) and len(result) > 0:
+                        return result
+                    if os.path.exists(temp_name):
+                        data = await asyncio.to_thread(sync_read_file, temp_name)
+                        os.remove(temp_name)
+                        if data and len(data) > 0:
+                            return data
+            except Exception as e:
+                err_str = str(e)
+                errors.append(f"Rubpy Obj Error: {err_str[:50]}")
+                await asyncio.sleep(3)
+
+    # روش 3: متد قدیمی rubpy (بک‌آپ نهایی با 5 بار پافشاری)
+    for attempt in range(5):
+        try:
+            if hasattr(client, "download_file"):
                 await client.download_file(file_id, file_name=temp_name)
                 if os.path.exists(temp_name):
                     data = await asyncio.to_thread(sync_read_file, temp_name)
                     os.remove(temp_name)
-                    return data
-            except Exception as e: errors.append(f"تلاش ۳: {e}")
-
-            try:
-                await client.download_file(file_id, save_as=temp_name)
-                if os.path.exists(temp_name):
-                    data = await asyncio.to_thread(sync_read_file, temp_name)
-                    os.remove(temp_name)
-                    return data
-            except Exception as e: errors.append(f"تلاش ۴: {e}")
-
-    raise Exception("دانلود ناموفق. لاگ:\n" + "\n".join(errors))
+                    if data and len(data) > 0:
+                        return data
+        except Exception as e:
+            errors.append(f"Rubpy FileId Error: {str(e)[:50]}")
+            await asyncio.sleep(3)
+            
+    raise Exception(f"سرورهای دانلود روبیکا پس از ۳۵ بار تلاش پاسخ ندادند!\nلاگ خطاها: {str(errors[-5:])}")
 
 
 GEMINI_KEYS_STR1 = os.environ.get("GEMINI_API_KEYS1", "")
@@ -458,28 +505,35 @@ def get_next_gemini_keys(count=100):
 HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
-bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
-
 
+# ==================================================================
+# 🔥 تابع آپلود فایل فوق قدرتمند و ضدضربه (75 بار مقاومت در برابر 502)
+# ==================================================================
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
     error_logs =[]
     
-    api_file_type = "Image" if file_type in["photo", "Image", "image"] else "Voice" if file_type in["voice", "Voice", "audio"] else "File"
+    # تعیین خودکار و هوشمندانه نوع فرمت ارسالی به روبیکا
+    api_file_type = "Image" if file_type in["photo", "Image", "image"] else "Voice" if file_type in ["voice", "Voice", "audio"] else file_type
     
+    # روبیکا با ارسال فایل‌های ogg غیر ویس و فایل‌های mp3 به عنوان File مشکل دارد و 502 میدهد.
     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"
         
-    for attempt in range(3):
+    # روش اول: استفاده مستقیم از API روبیکا با 15 بار پافشاری قدرتمند
+    for attempt in range(15):
         try:
             url_request = f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile"
             url_send = f"https://botapi.rubika.ir/v3/{bot_token}/sendFile"
             
             async with aiohttp.ClientSession() as session:
-                async with session.post(url_request, json={"type": api_file_type}, timeout=20) as resp:
+                # 1. درخواست لینک آپلود
+                async with session.post(url_request, json={"type": api_file_type}, timeout=30) as resp:
                     if resp.status != 200:
                         error_logs.append(f"Request HTTP {resp.status}")
-                        await asyncio.sleep(1.5)
+                        await asyncio.sleep(2)
                         continue
                         
                     req_data = await resp.json()
@@ -492,21 +546,24 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                 form = aiohttp.FormData()
                                 form.add_field('file', f, filename=os.path.basename(abs_path))
                                 
-                                async with session.post(upload_url, data=form, timeout=60) as up_resp:
+                                # 2. آپلود فایل در سرور (با تایم‌اوت 300 ثانیه مخصوص فایل‌های حجیم مثل پادکست)
+                                async with session.post(upload_url, data=form, timeout=300) as up_resp:
                                     if up_resp.status != 200:
                                         error_logs.append(f"Upload HTTP {up_resp.status}")
-                                        if up_resp.status == 502 and api_file_type == "Voice":
-                                            api_file_type = "Music" 
-                                        await asyncio.sleep(1.5)
+                                        if up_resp.status in[502, 503, 500]:
+                                            await asyncio.sleep(3)
+                                        else:
+                                            await asyncio.sleep(1.5)
                                         continue
                                         
                                     try:
                                         up_data = await up_resp.json()
                                     except Exception as json_err:
                                         error_logs.append(f"JSON Decode Error: {json_err}")
-                                        await asyncio.sleep(1.5)
+                                        await asyncio.sleep(2)
                                         continue
                                     
+                                    # 3. دریافت آیدی و ارسال نهایی پیام
                                     final_file_id = None
                                     if up_data.get("status") == "OK" or up_data.get("status_det") == "OK":
                                         if "data" in up_data and isinstance(up_data["data"], dict):
@@ -522,10 +579,10 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                             "chat_keypad_type": "New",
                                             "chat_keypad": MAIN_KEYPAD_DICT
                                         }
-                                        async with session.post(url_send, json=send_payload, timeout=20) as send_resp:
+                                        async with session.post(url_send, json=send_payload, timeout=30) as send_resp:
                                             if send_resp.status != 200:
                                                 error_logs.append(f"Send HTTP {send_resp.status}")
-                                                await asyncio.sleep(1.5)
+                                                await asyncio.sleep(2)
                                                 continue
                                                 
                                             s_data = await send_resp.json()
@@ -543,21 +600,29 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                         error_logs.append(f"Request API Error: {req_data}")
         except Exception as e:
             error_logs.append(f"Raw HTTP Error: {str(e)[:100]}")
-            await asyncio.sleep(1.5)
+            await asyncio.sleep(2.5)
 
-    try:
-        if hasattr(client, "send_file"):
-            await client.send_file(chat_id, abs_path)
-            if caption:
-                await send_with_keyboard(client, chat_id, caption, True)
-            return True
-        elif hasattr(client, "send_document"): 
-            await client.send_document(chat_id, abs_path, caption=caption)
-            return True
-    except Exception as e:
-        error_logs.append(f"Rubpy Send Error: {str(e)[:100]}")
+    # روش دوم: بک‌آپ قدرتمند با کتابخانه rubpy با 10 بار پافشاری 
+    for attempt in range(10):
+        try:
+            if hasattr(client, "send_file"):
+                await client.send_file(chat_id, abs_path)
+                if caption:
+                    await send_with_keyboard(client, chat_id, caption, True)
+                return True
+            elif hasattr(client, "send_document"): 
+                await client.send_document(chat_id, abs_path, caption=caption)
+                return True
+        except Exception as e:
+            err_msg = str(e)[:100]
+            error_logs.append(f"Rubpy Send Error: {err_msg}")
+            if "502" in err_msg or "timeout" in err_msg.lower() or "time" in err_msg.lower():
+                await asyncio.sleep(4)
+                continue
+            else:
+                await asyncio.sleep(2)
 
-    return "\n".join(error_logs[-4:])
+    return "\n".join(error_logs[-6:])
 
 
 WORKER_URLS =[
@@ -626,26 +691,27 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         history = history[-40:]
         if history[0]["role"] == "model": history = history[1:]
 
-    keys_to_try = get_next_gemini_keys(100)
+    # --- سیستم پافشاری (تلاش مجدد) برای دریافت جواب از هوش مصنوعی ---
     final_answer = None
-    
-    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": history, "generationConfig": {"temperature": 0.7, "maxOutputTokens": 8192}}
-            try:
-                async with session.post(url, json=payload, timeout=60) as response:
-                    if response.status == 200:
-                        data = await response.json()
-                        try:
-                            final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
-                            break
-                        except (KeyError, IndexError): continue 
-            except Exception: continue
+    for attempt in range(3):
+        keys_to_try = get_next_gemini_keys(100)
+        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": history, "generationConfig": {"temperature": 0.7, "maxOutputTokens": 8192}}
+                try:
+                    async with session.post(url, json=payload, timeout=60) as response:
+                        if response.status == 200:
+                            data = await response.json()
+                            try:
+                                final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
+                                break
+                            except (KeyError, IndexError): continue 
+                except Exception: continue
+        if final_answer: break
+        await asyncio.sleep(2)
 
-    # اگر جیمینای جواب نداد
     if not final_answer:
-        # اگر فایل ارسال شد�� عکس نبود، خطای اختصاصی می‌دهیم
         if has_non_image_file:
             if history and history[-1]["role"] == "user": history.pop()
             try:
@@ -657,7 +723,6 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
             await send_with_keyboard(client, chat_id, "❌ متأسفانه سرور اصلی در حال حاضر شلوغ است و در این شرایط **فعلاً تنها امکان تحلیل تصاویر** توسط سرور جایگزین در دسترس است.", False)
             return
             
-        # اگر فایل عکس بود یا فقط متن بود، می‌رویم سراغ مدل جایگزین (هاگینگ فیس)
         if HF_TOKENS:
             hf_messages =[]
             for msg in history:
@@ -677,26 +742,28 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
                 if content_parts:
                     hf_messages.append({"role": role, "content": content_parts})
 
-            keys_to_try_hf = HF_TOKENS.copy()
-            random.shuffle(keys_to_try_hf)
-            
-            async with aiohttp.ClientSession() as session:
-                for hf_key in keys_to_try_hf:
-                    url = "https://router.huggingface.co/v1/chat/completions"
-                    headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
-                    payload = {
-                        "model": "google/gemma-4-31B-it:novita",
-                        "messages": hf_messages,
-                        "max_tokens": 4096
-                    }
-                    try:
-                        async with session.post(url, headers=headers, json=payload, timeout=60) as response:
-                            if response.status == 200:
-                                data = await response.json()
-                                final_answer = data["choices"][0]["message"]["content"]
-                                break
-                    except Exception:
-                        continue
+            for attempt in range(3):
+                keys_to_try_hf = HF_TOKENS.copy()
+                random.shuffle(keys_to_try_hf)
+                async with aiohttp.ClientSession() as session:
+                    for hf_key in keys_to_try_hf:
+                        url = "https://router.huggingface.co/v1/chat/completions"
+                        headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
+                        payload = {
+                            "model": "google/gemma-4-31B-it:novita",
+                            "messages": hf_messages,
+                            "max_tokens": 4096
+                        }
+                        try:
+                            async with session.post(url, headers=headers, json=payload, timeout=60) as response:
+                                if response.status == 200:
+                                    data = await response.json()
+                                    final_answer = data["choices"][0]["message"]["content"]
+                                    break
+                        except Exception:
+                            continue
+                if final_answer: break
+                await asyncio.sleep(2)
 
     try:
         if proc_msg:
@@ -710,10 +777,6 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         await send_with_keyboard(client, chat_id, "❌ تمامی سرورها شلوغ هستند. لطفاً بعداً امتحان کنید.", False)
         return
 
-    if not creds.get("is_premium"):
-        user_credits_db[str_chat_id]["chat"] -= 1
-        save_db(user_credits_db)
-
     history.append({"role": "model", "parts":[{"text": final_answer}]})
     user_states[chat_id]["history"] = history
 
@@ -729,12 +792,19 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
             temp_text = temp_text[split_idx:].strip()
         if temp_text: chunks.append(temp_text)
         
+        success_sent = False
         for idx, chunk in enumerate(chunks):
             if idx != len(chunks) - 1: chunk += "\n\n⏳ *(ادامه در پیام بعدی)...* 👇"
             try:
-                await send_with_keyboard(client, chat_id, chunk, False)
+                res = await send_with_keyboard(client, chat_id, chunk, False)
+                if res: success_sent = True
                 await asyncio.sleep(2.5) 
             except Exception: await asyncio.sleep(2.5)
+            
+        # --- کسر اعتبار فقط در صورتی که حداقل یک بخش از پیام موفقیت‌آمیز ارسال شده باشد ---
+        if success_sent and not creds.get("is_premium"):
+            user_credits_db[str_chat_id]["chat"] -= 1
+            save_db(user_credits_db)
                 
     except Exception:
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد.", False)
@@ -794,7 +864,6 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
             if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
     
-    # تنظیم ابعاد انتخابی کاربر
     w, h = 1024, 1024
     size_name = "مربع (1:1) ⬛"
     if size_choice == "2":
@@ -810,24 +879,28 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
     short_preview = enhanced_prompt[:150] + "..." if len(enhanced_prompt) > 150 else enhanced_prompt
     proc_msg = await send_with_keyboard(client, chat_id, f"🎨 در حال طراحی عکس...\n\n📏 **ابعاد:** {size_name}\n📝 **پرامپت ساخته شده:**\n`{short_preview}`\n\n(ممکن است چند ثانیه زمان ببرد)", False)
 
-    keys_to_try = HF_TOKENS.copy()
-    random.shuffle(keys_to_try)
+    # --- سیستم پافشاری برای ساخت عکس ---
     generated_image = None
     last_error_log = "هیچ اتصالی برقرار نشد."
     
-    for token in keys_to_try:
-        try:
-            hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
-            generated_image = await hf_client.text_to_image(
-                enhanced_prompt, 
-                model="Tongyi-MAI/Z-Image-Turbo",
-                width=w,
-                height=h
-            )
-            break 
-        except Exception as e:
-            last_error_log = str(e)
-            continue
+    for attempt in range(5):
+        keys_to_try = HF_TOKENS.copy()
+        random.shuffle(keys_to_try)
+        for token in keys_to_try:
+            try:
+                hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
+                generated_image = await hf_client.text_to_image(
+                    enhanced_prompt, 
+                    model="Tongyi-MAI/Z-Image-Turbo",
+                    width=w,
+                    height=h
+                )
+                break 
+            except Exception as e:
+                last_error_log = str(e)
+                continue
+        if generated_image: break
+        await asyncio.sleep(2)
             
     try:
         if proc_msg:
@@ -836,21 +909,38 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
             if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
 
-    if not generated_image: return await send_with_keyboard(client, chat_id, f"❌ عکس ساخته نشد.\n\n⚠️ خطا:\n{last_error_log[:200]}", True)
-
-    user_credits_db[str_chat_id]["image"] -= 1
-    save_db(user_credits_db)
+    if not generated_image: 
+        return await send_with_keyboard(client, chat_id, f"❌ عکس ساخته نشد.\n\n⚠️ خطا:\n{last_error_log[:200]}", True)
 
     try:
         file_name = f"image_{uuid.uuid4().hex}.jpg"
-        # استفاده از ترد برای جلوگیری از مسدود شدن برنامه حین کار با PIL
         await asyncio.to_thread(sync_save_image, generated_image, file_name)
         await asyncio.sleep(1)
         caption_text = f"🎨 تصویر شما با پرامپت هوشمند آماده شد!\n\n📏 ابعاد تصویر: {size_name}\n✨ ایده اولیه: {prompt}"
-        upload_result = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
-        if upload_result is not True: await send_with_keyboard(client, chat_id, f"❌ خطا در آپلود روبیکا:\n`{str(upload_result)[:800]}`", True)
+        
+        # آپلود قدرتمند با 3 بار تلاش مجدد بیرونی 
+        upload_result = False
+        error_log_img = ""
+        for up_att in range(3):
+            res = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
+            if res is True:
+                upload_result = True
+                break
+            else:
+                error_log_img = res
+                await asyncio.sleep(4)
+        
+        # --- کسر اعتبار فقط در صورت موفقیت کامل آپلود به کاربر ---
+        if upload_result is True:
+            if not creds.get("is_premium"):
+                user_credits_db[str_chat_id]["image"] -= 1
+                save_db(user_credits_db)
+        else:
+            await send_with_keyboard(client, chat_id, f"❌ عکس ساخته شد اما آپلود در روبیکا با خطا مواجه شد:\n`{str(error_log_img)[:800]}`", True)
+            
         if os.path.exists(file_name): os.remove(file_name)
-    except Exception as e: await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره عکس:\n{str(e)[:150]}", True)
+    except Exception as e: 
+        await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره و ارسال عکس:\n{str(e)[:150]}", True)
 
 
 async def translate_text_aloha(prompt_text):
@@ -902,46 +992,63 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     if not translated_prompt or translated_prompt.strip() == "":
         translated_prompt = prompt
 
-    keys_to_try = HF_TOKENS.copy()
-    random.shuffle(keys_to_try)
+    # --- سیستم پافشاری برای ویرایش عکس ---
     generated_image = None
     last_error_log = "سرور پاسخ نداد."
     
-    for token in keys_to_try:
-        try:
-            hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
-            generated_image = await hf_client.image_to_image(
-                image_bytes,
-                prompt=translated_prompt,
-                model="black-forest-labs/FLUX.2-dev"
-            )
-            break 
-        except Exception as e:
-            last_error_log = str(e)
-            continue
+    for attempt in range(5):
+        keys_to_try = HF_TOKENS.copy()
+        random.shuffle(keys_to_try)
+        for token in keys_to_try:
+            try:
+                hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
+                generated_image = await hf_client.image_to_image(
+                    image_bytes,
+                    prompt=translated_prompt,
+                    model="black-forest-labs/FLUX.2-dev"
+                )
+                break 
+            except Exception as e:
+                last_error_log = str(e)
+                continue
+        if generated_image: break
+        await asyncio.sleep(2)
             
     try:
         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 not generated_image:
         return await send_with_keyboard(client, chat_id, f"❌ متأسفانه ویرایش عکس انجام نشد.\n\n⚠️ علت:\n{last_error_log[:200]}", True)
 
-    user_credits_db[str_chat_id]["edit_image"] -= 1
-    save_db(user_credits_db)
-
     try:
         file_name = f"edited_flux_{uuid.uuid4().hex}.jpg"
-        # استفاده از ترد برای ذخیره عکس
         await asyncio.to_thread(sync_save_image, generated_image, file_name)
         await asyncio.sleep(1)
         caption_text = f"🪄 ویرایش عکس با هوش مصنوعی Flux.2 انجام شد!\n\n✨ تغییرات خواسته شده: {prompt}\n🔤 متن ارسال شده به هوش: {translated_prompt}"
-        upload_result = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
-        if upload_result is not True:
-            await send_with_keyboard(client, chat_id, f"❌ عکس ساخته شد اما خطا در ارسال به روبیکا رخ داد:\n`{str(upload_result)[:800]}`", True)
+        
+        upload_result = False
+        error_log_edit = ""
+        for up_att in range(3):
+            res = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
+            if res is True:
+                upload_result = True
+                break
+            else:
+                error_log_edit = res
+                await asyncio.sleep(4)
+        
+        # --- کسر اعتبار فقط پس از تحویل موفقیت آمیز ---
+        if upload_result is True:
+            if not creds.get("is_premium"):
+                user_credits_db[str_chat_id]["edit_image"] -= 1
+                save_db(user_credits_db)
+        else:
+            await send_with_keyboard(client, chat_id, f"❌ عکس ساخته شد اما خطا در ارسال به روبیکا رخ داد:\n`{str(error_log_edit)[:800]}`", True)
+            
         if os.path.exists(file_name): os.remove(file_name)
     except Exception as e:
         await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره عکس ویرایش شده:\n{str(e)[:150]}", True)
@@ -957,23 +1064,30 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
         proc_msg = await send_with_keyboard(client, chat_id, f"⏳ در حال ساخت صدا با «{speaker_name}»...\n(لطفاً صبور باشید)", False)
         payload = {"text": user_text, "speaker": speaker_id, "temperature": 1.5, "prompt": "", "use_live_model": True}
         headers = {"User-Agent": "Mozilla/5.0", "Content-Type": "application/json"}
+        
+        # --- سیستم پافشاری برای ساخت صدا ---
         audio_bytes = None
         last_error = "پاسخی دریافت نشد"
-        workers = WORKER_URLS.copy()
-        random.shuffle(workers)
         
-        async with aiohttp.ClientSession(headers=headers, timeout=aiohttp.ClientTimeout(total=600)) as session:
-            for worker_url in workers[:3]:
-                try:
-                    async with session.post(worker_url, json=payload) as response:
-                        if response.status == 200:
-                            content_type = response.headers.get('Content-Type', '')
-                            if 'audio' in content_type or response.content_length > 1000:
-                                audio_bytes = await response.read()
-                                break
-                            else: last_error = "فایل نامعتبر"
-                        else: last_error = f"ارور ({response.status})"
-                except Exception as e: last_error = f"خطا: {str(e)}"; continue
+        for attempt in range(10):
+            workers = WORKER_URLS.copy()
+            random.shuffle(workers)
+            async with aiohttp.ClientSession(headers=headers, timeout=aiohttp.ClientTimeout(total=600)) as session:
+                for worker_url in workers[:3]:
+                    try:
+                        async with session.post(worker_url, json=payload) as response:
+                            if response.status == 200:
+                                content_type = response.headers.get('Content-Type', '')
+                                if 'audio' in content_type or response.content_length > 1000:
+                                    audio_bytes = await response.read()
+                                    break
+                                else: last_error = "فایل نامعتبر"
+                            else: last_error = f"ارور ({response.status})"
+                    except Exception as e: 
+                        last_error = f"خطا: {str(e)}"
+                        continue
+            if audio_bytes: break
+            await asyncio.sleep(2)
 
         try:
             if proc_msg:
@@ -983,23 +1097,34 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
         except Exception: pass
 
         if audio_bytes:
-            if not creds.get("is_premium"):
-                user_credits_db[str_chat_id]["tts"] -= 1
-                save_db(user_credits_db)
-            
             file_name_mp3 = f"audio_{uuid.uuid4().hex}.mp3"
-            
-            # ذخیره فایل در ترد جداگانه
             await asyncio.to_thread(sync_write_file, file_name_mp3, audio_bytes)
-            
             await asyncio.sleep(1)
-            upload_result_file = await helper_upload_file(client, chat_id, file_name_mp3, "File", "✅ صدای شما با موفقیت آماده شد (فایل MP3):")
             
-            if upload_result_file is not True: 
-                await send_with_keyboard(client, chat_id, f"❌ ارور آپلود فایل:\n`{str(upload_result_file)[:800]}`", True)
+            # ارسال با فرمت Music و مقاومت ۳ برابری
+            upload_result_file = False
+            error_log_tts = ""
+            for up_att in range(3):
+                # 🔴 ارسال متن به صدا بصورت موزیک 🔴
+                res = await helper_upload_file(client, chat_id, file_name_mp3, "Music", "✅ صدای شما با موفقیت آماده شد (فایل MP3):")
+                if res is True:
+                    upload_result_file = True
+                    break
+                else:
+                    error_log_tts = res
+                    await asyncio.sleep(4)
+            
+            # --- کسر اعتبار فقط پس از تحویل موفق فایل ---
+            if upload_result_file is True: 
+                if not creds.get("is_premium"):
+                    user_credits_db[str_chat_id]["tts"] -= 1
+                    save_db(user_credits_db)
+            else:
+                await send_with_keyboard(client, chat_id, f"❌ فایل صدا ساخته شد اما سرور روبیکا اجازه آپلود نداد:\n`{str(error_log_tts)[:800]}`", True)
                 
             if os.path.exists(file_name_mp3): os.remove(file_name_mp3)
-        else: await send_with_keyboard(client, chat_id, f"❌ سرورها درگیر هستند.\nدلیل: {last_error}", True)
+        else: 
+            await send_with_keyboard(client, chat_id, f"❌ سرورها درگیر هستند.\nدلیل: {last_error}", True)
     except Exception: traceback.print_exc()
 
 
@@ -1057,7 +1182,9 @@ async def process_podcast(client, chat_id, prompt):
         for index, turn in enumerate(script_data):
             payload_gen = {"text": turn["dialogue"], "speaker": turn["speaker_id"], "temperature": 0.9}
             chunk_audio_bytes = None
-            for attempt in range(3):
+            
+            # --- پافشاری قدرتمندتر برای دریافت هر بخش پادکست ---
+            for attempt in range(10):
                 try:
                     async with session.post(url_generate, json=payload_gen, timeout=120) as resp:
                         if resp.status == 200:
@@ -1065,19 +1192,15 @@ async def process_podcast(client, chat_id, prompt):
                             break
                 except Exception: await asyncio.sleep(2)
             
-            if not chunk_audio_bytes: return await send_with_keyboard(client, chat_id, f"❌ خطا در تولید صدای بخش {index+1} از سرور. عملیات متوقف شد.", True)
+            if not chunk_audio_bytes: 
+                return await send_with_keyboard(client, chat_id, f"❌ خطا در تولید صدای بخش {index+1} از سرور. عملیات متوقف شد.", True)
                 
             try:
-                # جلوگیری از گیر کردن لوپ در هنگام ترکیب فایل‌های صوتی سنگین
                 combined_audio = await asyncio.to_thread(sync_combine_audio, combined_audio, chunk_audio_bytes)
-            except Exception as e: return await send_with_keyboard(client, chat_id, f"❌ خطا در پردازش صدا (آیا pydub و ffmpeg نصب است؟):\n{str(e)}", True)
-
-        if not creds.get("is_premium"):
-            user_credits_db[str_chat_id]["podcast"] -= 1
-            save_db(user_credits_db)
+            except Exception as e: 
+                return await send_with_keyboard(client, chat_id, f"❌ خطا در پردازش صدا (آیا pydub و ffmpeg نصب است؟):\n{str(e)}", True)
 
         file_name_mp3 = f"final_podcast_{uuid.uuid4().hex}.mp3"
-        # اکسپورت فایل صوتی در ترد جدا
         await asyncio.to_thread(sync_export_audio, combined_audio, file_name_mp3)
         
         try:
@@ -1088,10 +1211,28 @@ async def process_podcast(client, chat_id, prompt):
         except: pass
 
         caption_file = f"🎧 فایل پادکست شما با فرمت MP3:\n\n💡 موضوع شما: {prompt}"
-        upload_result_file = await helper_upload_file(client, chat_id, file_name_mp3, "File", caption_file)
         
-        if upload_result_file is not True: 
-            await send_with_keyboard(client, chat_id, f"❌ پادکست ساخته شد اما روبیکا خطای آپلود داد.\n\n`{str(upload_result_file)[:800]}`", True)
+        # ⚠️ آپلود بسیار قدرتمند پادکست (ارسال با هویت Music برای جلوگیری از 502) 
+        # همراه با 3 مرحله تکرار خارجی، که خود شامل 15 + 10 بار تکرار داخلی است!
+        upload_result_file = False
+        error_log_pod = ""
+        for up_att in range(3):
+            # 🔴 ارسال پادکست بصورت موزیک 🔴
+            res = await helper_upload_file(client, chat_id, file_name_mp3, "Music", caption_file)
+            if res is True:
+                upload_result_file = True
+                break
+            else:
+                error_log_pod = res
+                await asyncio.sleep(5)
+        
+        # --- کسر اعتبار فقط در صورتی که فایل پادکست با موفقیت در روبیکا آپلود شده باشد ---
+        if upload_result_file is True: 
+            if not creds.get("is_premium"):
+                user_credits_db[str_chat_id]["podcast"] -= 1
+                save_db(user_credits_db)
+        else:
+            await send_with_keyboard(client, chat_id, f"❌ پادکست ساخته شد اما روبیکا پس از ده‌ها تلاش خطای آپلود داد.\n\n`{str(error_log_pod)[:800]}`", True)
             
         if os.path.exists(file_name_mp3): os.remove(file_name_mp3)
 
@@ -1109,21 +1250,25 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     mime_type, _ = mimetypes.guess_type(file_name)
     if not mime_type: mime_type = "audio/ogg" 
     
-    keys_to_try = get_next_gemini_keys(100)
-    transcribed_text = None
     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.2}}
-            try:
-                async with session.post(url, json=payload, timeout=60) as response:
-                    if response.status == 200:
-                        data = await response.json()
-                        transcribed_text = data["candidates"][0]["content"]["parts"][0]["text"]
-                        break
-            except Exception: continue
+    # --- سیستم پافشاری برای تبدیل صدا به متن ---
+    transcribed_text = None
+    for attempt in range(5):
+        keys_to_try = get_next_gemini_keys(100)
+        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}}
+                try:
+                    async with session.post(url, json=payload, timeout=60) as response:
+                        if response.status == 200:
+                            data = await response.json()
+                            transcribed_text = data["candidates"][0]["content"]["parts"][0]["text"]
+                            break
+                except Exception: continue
+        if transcribed_text: break
+        await asyncio.sleep(2)
 
     try:
         if proc_msg:
@@ -1133,10 +1278,11 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     except Exception: pass
 
     if transcribed_text:
-        if not creds.get("is_premium"):
+        sent = await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
+        # --- کسر اعتبار فقط اگر پیام با موفقیت در روبیکا ارسال شد ---
+        if sent and not creds.get("is_premium"):
             user_credits_db[str_chat_id]["stt"] -= 1
             save_db(user_credits_db)
-        await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
     else: 
         await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", True)
 
@@ -1164,70 +1310,76 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
         
     is_image = mime_type.startswith('image/')
     
-    keys_to_try = get_next_gemini_keys(100)
+    # --- سیستم پافشاری برای تحلیل فایل ---
     final_answer = None
-    
-    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}}
-            try:
-                async with session.post(url, json=payload, timeout=45) as response:
-                    if response.status == 200:
-                        data = await response.json()
-                        final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
-                        break
-            except Exception: continue
+    for attempt in range(5):
+        keys_to_try = get_next_gemini_keys(100)
+        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}}
+                try:
+                    async with session.post(url, json=payload, timeout=45) as response:
+                        if response.status == 200:
+                            data = await response.json()
+                            final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
+                            break
+                except Exception: continue
+        if final_answer: break
+        await asyncio.sleep(2)
 
-    # جایگزین تصویر اگر جیمینای جواب نداد
     if not final_answer:
         if is_image and HF_TOKENS:
-            keys_to_try_hf = HF_TOKENS.copy()
-            random.shuffle(keys_to_try_hf)
-            
-            hf_messages =[
-                {
-                    "role": "user",
-                    "content":[
-                        {"type": "text", "text": prompt},
-                        {
-                            "type": "image_url",
-                            "image_url": {"url": f"data:{mime_type};base64,{base64_data}"}
-                        }
-                    ]
-                }
-            ]
-            
-            async with aiohttp.ClientSession() as session:
-                for hf_key in keys_to_try_hf:
-                    url = "https://router.huggingface.co/v1/chat/completions"
-                    headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
-                    payload = {
-                        "model": "google/gemma-4-31B-it:novita",
-                        "messages": hf_messages,
-                        "max_tokens": 4096
+            for attempt in range(5):
+                keys_to_try_hf = HF_TOKENS.copy()
+                random.shuffle(keys_to_try_hf)
+                
+                hf_messages =[
+                    {
+                        "role": "user",
+                        "content":[
+                            {"type": "text", "text": prompt},
+                            {
+                                "type": "image_url",
+                                "image_url": {"url": f"data:{mime_type};base64,{base64_data}"}
+                            }
+                        ]
                     }
-                    try:
-                        async with session.post(url, headers=headers, json=payload, timeout=60) as response:
-                            if response.status == 200:
-                                data = await response.json()
-                                final_answer = data["choices"][0]["message"]["content"]
-                                break
-                    except Exception:
-                        continue
+                ]
+                
+                async with aiohttp.ClientSession() as session:
+                    for hf_key in keys_to_try_hf:
+                        url = "https://router.huggingface.co/v1/chat/completions"
+                        headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
+                        payload = {
+                            "model": "google/gemma-4-31B-it:novita",
+                            "messages": hf_messages,
+                            "max_tokens": 4096
+                        }
+                        try:
+                            async with session.post(url, headers=headers, json=payload, timeout=60) as response:
+                                if response.status == 200:
+                                    data = await response.json()
+                                    final_answer = data["choices"][0]["message"]["content"]
+                                    break
+                        except Exception:
+                            continue
+                if final_answer: break
+                await asyncio.sleep(2)
 
     try:
         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 final_answer: 
-        if not creds.get("is_premium"):
+        sent = await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
+        # --- کسر اعتبار فقط پس از ارسال موفقیت‌آمیز نتیجه ---
+        if sent and not creds.get("is_premium"):
             user_credits_db[str_chat_id]["file"] -= 1
             save_db(user_credits_db)
-        await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
     else: 
         if not is_image:
             await send_with_keyboard(client, chat_id, "❌ متأسفانه سرور اصلی در حال حاضر شلوغ است و در این شرایط **فعلاً تنها امکان تحلیل تصاویر** توسط سرور جایگزین در دسترس است.", True)
@@ -1247,40 +1399,47 @@ async def process_create_file(client, chat_id, topic):
     proc_msg = await send_with_keyboard(client, chat_id, "✍️ در حال تحقیق و نگارش مقاله جامع و حرفه‌ای...\n(این عملیات با توجه به طولانی بودن متن ممکن است کمی طول بکشد)", False)
 
     ai_prompt = f"یک مقاله بسیار جامع، کاملا حرفه‌ای، بسیار طولانی و با جزئیات کامل درباره موضوع زیر به زبان فارسی بنویس. فقط متن مقاله را بده و هیچ توضیح اضافه‌ای نده:\n\nموضوع: {topic}"
+    
+    # --- سیستم پافشاری برای تولید متن مقاله ---
     article_text = None
-
-    keys_to_try = get_next_gemini_keys(100)
-    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": ai_prompt}]}], "generationConfig": {"temperature": 0.7, "maxOutputTokens": 8192}}
-            try:
-                async with session.post(url, json=payload, timeout=60) as response:
-                    if response.status == 200:
-                        data = await response.json()
-                        article_text = data["candidates"][0]["content"]["parts"][0]["text"]
-                        break
-            except Exception: continue
-
-    if not article_text and HF_TOKENS:
-        keys_to_try_hf = HF_TOKENS.copy()
-        random.shuffle(keys_to_try_hf)
+    for attempt in range(5):
+        keys_to_try = get_next_gemini_keys(100)
         async with aiohttp.ClientSession() as session:
-            for hf_key in keys_to_try_hf:
-                url = "https://router.huggingface.co/v1/chat/completions"
-                headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
-                payload = {
-                    "model": "google/gemma-4-31B-it:novita",
-                    "messages":[{"role": "user", "content":[{"type": "text", "text": ai_prompt}]}],
-                    "max_tokens": 4096
-                }
+            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": ai_prompt}]}], "generationConfig": {"temperature": 0.7, "maxOutputTokens": 8192}}
                 try:
-                    async with session.post(url, headers=headers, json=payload, timeout=60) as response:
+                    async with session.post(url, json=payload, timeout=60) as response:
                         if response.status == 200:
                             data = await response.json()
-                            article_text = data["choices"][0]["message"]["content"]
+                            article_text = data["candidates"][0]["content"]["parts"][0]["text"]
                             break
                 except Exception: continue
+        if article_text: break
+        await asyncio.sleep(2)
+
+    if not article_text and HF_TOKENS:
+        for attempt in range(5):
+            keys_to_try_hf = HF_TOKENS.copy()
+            random.shuffle(keys_to_try_hf)
+            async with aiohttp.ClientSession() as session:
+                for hf_key in keys_to_try_hf:
+                    url = "https://router.huggingface.co/v1/chat/completions"
+                    headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
+                    payload = {
+                        "model": "google/gemma-4-31B-it:novita",
+                        "messages":[{"role": "user", "content":[{"type": "text", "text": ai_prompt}]}],
+                        "max_tokens": 4096
+                    }
+                    try:
+                        async with session.post(url, headers=headers, json=payload, timeout=60) as response:
+                            if response.status == 200:
+                                data = await response.json()
+                                article_text = data["choices"][0]["message"]["content"]
+                                break
+                    except Exception: continue
+            if article_text: break
+            await asyncio.sleep(2)
 
     if not article_text:
         return await send_with_keyboard(client, chat_id, "❌ سرور تولید متن شلوغ است، لطفاً بعداً تلاش کنید.", True)
@@ -1292,78 +1451,116 @@ async def process_create_file(client, chat_id, topic):
             if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
 
+    proc_msg = await send_with_keyboard(client, chat_id, "📄 مقاله با موفقیت نوشته شد! در حال ارتباط با سرور و تبدیل به فایل‌های PDF و Word...\n(لطفا صبور باشید)", False)
+
     converter_url = "https://opera8-texttopdf.hf.space/"
     uid = uuid.uuid4().hex
-    
-    # پیام اطلاع رسانی قبل از ساخت فایل‌ها
-    proc_msg = await send_with_keyboard(client, chat_id, "📄 مقاله با موفقیت نوشته شد!\n\n⏳ در حال ارتباط همزمان با هوش مصنوعی برای دریافت فایل‌های PDF و Word...\n", False)
 
-    # تابع کمکی برای درخواست همزمان به سرور
-    async def fetch_document(fmt):
-        for attempt in range(3):
-            try:
-                async with aiohttp.ClientSession() as session:
-                    form_data = aiohttp.FormData()
-                    form_data.add_field('content', article_text)
-                    form_data.add_field('format', fmt)
-                    # تایم‌اوت 180 ثانیه برای فرصت کافی به Weasyprint
-                    async with session.post(converter_url, data=form_data, timeout=180) as resp:
-                        if resp.status == 200:
-                            data = await resp.read()
-                            if data and len(data) > 0:
-                                return data
-            except Exception:
-                pass
-            await asyncio.sleep(2)
-        return None
+    # =======================================================================
+    # ---- 1. پردازش و ارسال فایل PDF (دونه دونه و با پافشاری شدید) ----
+    # =======================================================================
+    pdf_success = False
+    for attempt in range(30):  # تا 30 بار تلاش (تقریبا ول کن نیست)
+        try:
+            pdf_bytes = None
+            async with aiohttp.ClientSession() as session:
+                form_data = aiohttp.FormData()
+                form_data.add_field('content', article_text)
+                form_data.add_field('format', 'pdf')
+                async with session.post(converter_url, data=form_data, timeout=90) as resp:
+                    if resp.status == 200:
+                        data_bytes = await resp.read()
+                        if data_bytes and len(data_bytes) > 100:
+                            pdf_bytes = data_bytes
+            
+            if pdf_bytes:
+                filename = f"Article_{uid}.pdf"
+                await asyncio.to_thread(sync_write_file, filename, pdf_bytes)
+                
+                res_upload = False
+                for _ in range(3):
+                    r = await helper_upload_file(client, chat_id, filename, "File", f"📄 فایل PDF مقاله شما:\n\n💡 موضوع: {topic}")
+                    if r is True:
+                        res_upload = True
+                        break
+                    await asyncio.sleep(4)
+
+                if os.path.exists(filename): 
+                    os.remove(filename)
+                
+                if res_upload is True:
+                    pdf_success = True
+                    break  # با موفقیت ساخته و ارسال شد -> خروج از حلقه
+        except Exception as e:
+            print(f"PDF error (attempt {attempt+1}):", e)
+        
+        await asyncio.sleep(4)  # در صورت خطا 4 ثانیه مکث و تلاش مجدد
+
+    # =======================================================================
+    # استراحت بین دو فایل برای هضم شدن فایل قبلی توسط سیستم روبیکا
+    # =======================================================================
+    await asyncio.sleep(3.5)
+
+    # =======================================================================
+    # ---- 2. پردازش و ارسال فایل Word (دونه دونه و با پافشاری شدید) ----
+    # =======================================================================
+    docx_success = False
+    for attempt in range(30):  # تا 30 بار تلاش 
+        try:
+            docx_bytes = None
+            async with aiohttp.ClientSession() as session:
+                form_data = aiohttp.FormData()
+                form_data.add_field('content', article_text)
+                form_data.add_field('format', 'docx')
+                async with session.post(converter_url, data=form_data, timeout=90) as resp:
+                    if resp.status == 200:
+                        data_bytes = await resp.read()
+                        if data_bytes and len(data_bytes) > 100:
+                            docx_bytes = data_bytes
+            
+            if docx_bytes:
+                filename = f"Article_{uid}.docx"
+                await asyncio.to_thread(sync_write_file, filename, docx_bytes)
+                
+                res_upload = False
+                for _ in range(3):
+                    r = await helper_upload_file(client, chat_id, filename, "File", f"📝 فایل Word (DOCX) مقاله شما:\n\n💡 موضوع: {topic}")
+                    if r is True:
+                        res_upload = True
+                        break
+                    await asyncio.sleep(4)
+
+                if os.path.exists(filename): 
+                    os.remove(filename)
+                
+                if res_upload is True:
+                    docx_success = True
+                    break  # با موفقیت ساخته و ارسال شد -> خروج از حلقه
+        except Exception as e:
+            print(f"DOCX error (attempt {attempt+1}):", e)
+        
+        await asyncio.sleep(4)  # در صورت خطا 4 ثانیه مکث و تلاش مجدد
 
-    # اجرای همزمان هر دو درخواست برای سرعت بیشتر و جلوگیری از تایم‌اوت متوالی
-    pdf_bytes, docx_bytes = await asyncio.gather(
-        fetch_document('pdf'),
-        fetch_document('docx')
-    )
-    
     try:
         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
 
-    # --- آپلود فایل‌ها در روبیکا ---
-    uploaded_pdf = False
-    uploaded_docx = False
-    
-    if pdf_bytes:
-        pdf_filename = f"Article_{uid}.pdf"
-        await asyncio.to_thread(sync_write_file, pdf_filename, pdf_bytes)
-        res = await helper_upload_file(client, chat_id, pdf_filename, "File", f"📄 فایل PDF مقاله شما:\n\n💡 موضوع: {topic}")
-        if res is True: 
-            uploaded_pdf = True
-        if os.path.exists(pdf_filename): 
-            os.remove(pdf_filename)
-            
-    await asyncio.sleep(1.5)
-
-    if docx_bytes:
-        docx_filename = f"Article_{uid}.docx"
-        await asyncio.to_thread(sync_write_file, docx_filename, docx_bytes)
-        res = await helper_upload_file(client, chat_id, docx_filename, "File", f"📝 فایل Word (DOCX) مقاله شما:\n\n💡 موضوع: {topic}")
-        if res is True: 
-            uploaded_docx = True
-        if os.path.exists(docx_filename): 
-            os.remove(docx_filename)
-
-    if not creds.get("is_premium"):
-        user_credits_db[str_chat_id]["chat"] -= 1
-        save_db(user_credits_db)
-
-    if uploaded_pdf and uploaded_docx:
-        await send_with_keyboard(client, chat_id, "✅ هر دو فایل (PDF و Word) مقاله شما با موفقیت تحویل داده شد!", True)
-    elif uploaded_pdf or uploaded_docx:
-        await send_with_keyboard(client, chat_id, "⚠️ یکی از فایل‌ها با موفقیت ارسال شد، اما در ساخت یا ارسال فایل دیگر مشکلی از سمت سرور پیش آمد.", True)
+    # --- کسر اعتبار فقط در صورتی که حداقل یکی از فایل‌ها با موفقیت کامل آپلود شده باشد ---
+    if pdf_success and docx_success:
+        if not creds.get("is_premium"):
+            user_credits_db[str_chat_id]["chat"] -= 1
+            save_db(user_credits_db)
+        await send_with_keyboard(client, chat_id, "✅ مقاله شما با موفقیت به صورت هر دو فایل (PDF و Word) تحویل داده شد!", True)
+    elif pdf_success or docx_success:
+        if not creds.get("is_premium"):
+            user_credits_db[str_chat_id]["chat"] -= 1
+            save_db(user_credits_db)
+        await send_with_keyboard(client, chat_id, "⚠️ یکی از فایل‌ها با موفقیت ارسال شد اما دیگری پس از تلاش‌های مکرر سرور با خطا مواجه شد.", True)
     else:
-        await send_with_keyboard(client, chat_id, "❌ فایل‌ها ساخته شدند اما متأسفانه روبیکا در ارسال آن‌ها دچار مشکل شد.", True)
+        await send_with_keyboard(client, chat_id, "❌ متأسفانه پس از تلاش‌های مکرر، سرور قادر به ساخت و ارسال هیچ‌یک از فایل‌ها نبود و اعتباری از شما کسر نشد.", True)
 
 
 if not bot_token:
@@ -1723,7 +1920,7 @@ else:
 🔑 **شناسه یکتای شما:** `{chat_id}`
 
 👨‍💻 **ارتباط با پشتیبانی:**
-🆔 @aialpha_admin"""
+🆔 @H_a_m_e_d100"""
                 await send_with_keyboard(client, chat_id, buy_text, True)
                 return
 
@@ -1737,7 +1934,7 @@ else:
 🔑 **شناسه یکتای ربات شما:** `{chat_id}`
 
 👨‍💻 **دقت کنید شناسه ربات رو به پشتیبانی داخل خود برنامه هوش مصنوعی آلفا ارسال کنید:**
-🤖"""
+🆔 @H_a_m_e_d100"""
                 await send_with_keyboard(client, chat_id, transfer_text, True)
                 return
 
@@ -1796,7 +1993,6 @@ else:
             # ===============================================
             elif current_mode == "waiting_for_referral_code":
                 if user_text_str:
-                    # تبدیل به اعداد انگلیسی در صورت استفاده از کیبورد فارسی
                     normalized_code = to_english_digits(user_text_str).strip()
                     
                     if not normalized_code.isdigit() or len(normalized_code) != 8:
@@ -1815,11 +2011,9 @@ else:
                     
                     user_states[str_chat_id]["mode"] = None
                     
-                    # 1. پاداش کاربر دعوت شده (اضافه شدن 10 تبدیل صدا)
                     user_credits_db[str_chat_id]["used_referral"] = True
                     user_credits_db[str_chat_id]["tts"] = user_credits_db[str_chat_id].get("tts", 0) + 10
                     
-                    # 2. پاداش شخص دعوت کننده
                     user_credits_db[inviter_id]["invited_count"] = user_credits_db[inviter_id].get("invited_count", 0) + 1
                     current_invites = user_credits_db[inviter_id]["invited_count"]
                     
@@ -1864,18 +2058,37 @@ else:
                     await send_with_keyboard(client, chat_id, "✅ کد هدیه با موفقیت ثبت شد!\n🎁 **10 سهمیه تبدیل رایگان متن به صدا** به حساب شما اضافه گردید.", True)
                 return
 
+            # ===============================================
+            # آپدیت جدید: در چت ابتدا فایل را می‌گیرد و می‌پرسد چه کار کند
+            # ===============================================
             elif current_mode == "chat":
                 if is_file:
                     await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل...", False)
                     try:
                         file_bytes = await helper_download_file(client, msg_obj)
-                        asyncio.create_task(process_gemini(client, chat_id, user_text_str, file_bytes=file_bytes, file_name=file_name))
-                    except Exception as dl_err: await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n{str(dl_err)}", False)
-                elif user_text_str: asyncio.create_task(process_gemini(client, chat_id, user_text_str))
+                        user_states[str_chat_id]["file_bytes"] = file_bytes
+                        user_states[str_chat_id]["file_name"] = file_name
+                        user_states[str_chat_id]["mode"] = "chat_waiting_for_prompt"
+                        await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً متنی بفرستید و بگویید **چه کاری با این فایل انجام دهم؟**\n(مثلاً: این تصویر را توصیف کن، یا متن این سند را خلاصه کن)", False)
+                    except Exception as dl_err: 
+                        await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n{str(dl_err)}", False)
+                elif user_text_str: 
+                    asyncio.create_task(process_gemini(client, chat_id, user_text_str))
+                return
+
+            elif current_mode == "chat_waiting_for_prompt":
+                if user_text_str:
+                    saved_bytes = user_states[str_chat_id].get("file_bytes")
+                    saved_name = user_states[str_chat_id].get("file_name", "file.jpeg")
+                    user_states[str_chat_id]["mode"] = "chat"
+                    user_states[str_chat_id]["file_bytes"] = None
+                    asyncio.create_task(process_gemini(client, chat_id, user_text_str, file_bytes=saved_bytes, file_name=saved_name))
+                else: 
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً درخواست خود را متنی بنویسید.", False)
                 return
 
             # ===============================================
-            # آپدیت جدید: دریافت سایز قبل از ساخت عکس
+            # دریافت سایز قبل از ساخت عکس
             # ===============================================
             elif current_mode == "image_waiting_for_text":
                 if user_text_str:
@@ -2006,10 +2219,8 @@ if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
     
     if bot_token:
-        # افزایش ظرفیت کارگرهای (Workers) پس‌زمینه در Asyncio برای افزایش شدید سرعت پاسخگویی
         loop = asyncio.get_event_loop()
-        # تنظیم ترد پول روی 16 کارگر به جای 1 پیش‌فرض که باعث کندی میشد
         loop.set_default_executor(concurrent.futures.ThreadPoolExecutor(max_workers=16))
         
-        print("ربات آلفا پرو با سیستم اشتراک نامحدود + سپر امنیتی + 16 Worker پس‌زمینه روشن شد...")
+        print("ربات آلفا پرو با سیستم اشتراک نامحدود + سپر امنیتی + دانلود پیشرفته فایل + 16 Worker پس‌زمینه روشن شد...")
         bot.run()