Update main.py

main.py

@@ -142,54 +142,78 @@ async def helper_download_file(client, msg_obj):
     raise Exception("دانلود ناموفق. لاگ:\n" + "\n".join(errors))
 
 
-# --- 🚨 تابع هوشمند آپلود فایل (نسخه فوق‌پیشرفته + دیباگر) 🚨 ---
-async def helper_upload_file(client, chat_id, file_name, file_type="photo", caption=""):
-    abs_path = os.path.abspath(file_name) # پیدا کردن مسیر قطعی در سرور
-    error_logs =[]
-    
-    # 1. تلاش اول: ارسال پیام با پارامتر فایل
-    try:
-        await client.send_message(chat_id, text=caption, file=abs_path)
-        return True
-    except Exception as e:
-        error_logs.append(f"MsgFile: {e}")
+# --- تنظیمات کلیدها ---
+GEMINI_KEYS_STR = os.environ.get("GEMINI_API_KEYS", "")
+GEMINI_KEYS =[k.strip() for k in GEMINI_KEYS_STR.split(",") if k.strip()]
 
-    # 2. تلاش دوم: استفاده مستقیم بدون نام پارامتر
-    try:
-        if file_type == "photo":
-            await client.send_photo(chat_id, abs_path, caption=caption)
-        else:
-            await client.send_voice(chat_id, abs_path)
-        return True
-    except Exception as e:
-        error_logs.append(f"DirectArg: {e}")
+HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
+HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
-    # 3. تلاش سوم: استفاده با نام پارامتر
-    try:
-        if file_type == "photo":
-            await client.send_photo(chat_id, photo=abs_path, caption=caption)
-        else:
-            await client.send_voice(chat_id, voice=abs_path)
-        return True
-    except Exception as e:
-        error_logs.append(f"Kwarg: {e}")
+# 🔴 تنظیم کردن توکن ربات به صورت سراسری برای متد آپلود 🔴
+bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
-    # 4. تلاش چهارم: ارسال به عنوان سند (Document)
+
+# --- 🚨 تابع اختصاصی آپلود مستقیم به هسته روبیکا (دور زدن محدودیت‌های Rubpy) 🚨 ---
+async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
+    abs_path = os.path.abspath(file_name)
+    error_logs =[]
+    
+    # تنظیم نوع فایل برای API رسمی روبیکا
+    api_file_type = "Image" if file_type in ["photo", "Image", "image"] else "Voice" if file_type in["voice", "Voice", "audio"] else "File"
+    
     try:
-        await client.send_document(chat_id, document=abs_path, caption=caption)
-        return True
-    except Exception as e:
-        error_logs.append(f"Doc: {e}")
+        # روش تضمینی: استفاده از HTTP Request خام به Bot API روبیکا 
+        url_request = f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile"
+        url_send = f"https://botapi.rubika.ir/v3/{bot_token}/sendFile"
         
-    # 5. تلاش پنجم: تبدیل به بایت و ارسال مستقیم حافظه
+        async with aiohttp.ClientSession() as session:
+            # مرحله 1: درخواست ایجاد فضای آپلود
+            async with session.post(url_request, json={"type": api_file_type}, timeout=20) as resp:
+                req_data = await resp.json()
+                
+                if req_data.get("status") == "OK":
+                    upload_url = req_data["data"]["upload_url"]
+                    file_id = req_data["data"]["file_id"]
+                    
+                    # مرحله 2: آپلود بایت‌های فایل
+                    with open(abs_path, "rb") as f:
+                        form = aiohttp.FormData()
+                        form.add_field('file', f, filename=os.path.basename(abs_path))
+                        headers = {"file-id": str(file_id)}
+                        
+                        async with session.post(upload_url, data=form, headers=headers, timeout=60) as up_resp:
+                            up_data = await up_resp.json()
+                            
+                            if up_data.get("status") == "OK":
+                                # مرحله 3: ارسال فایل آپلود شده به چت کاربر به همراه کیبورد
+                                send_payload = {
+                                    "chat_id": chat_id,
+                                    "file_id": file_id,
+                                    "text": caption,
+                                    "chat_keypad_type": "New",
+                                    "chat_keypad": MAIN_KEYPAD_DICT
+                                }
+                                async with session.post(url_send, json=send_payload, timeout=20) as send_resp:
+                                    s_data = await send_resp.json()
+                                    if s_data.get("status") == "OK":
+                                        return True
+                                    else:
+                                        error_logs.append(f"SendFile API Error: {s_data}")
+                            else:
+                                error_logs.append(f"Upload API Error: {up_data}")
+                else:
+                    error_logs.append(f"Request API Error: {req_data}")
+    except Exception as e:
+        error_logs.append(f"Raw HTTP Exception: {e}")
+
+    # پشتیبان: اگر API بالا به هر دلیلی محدود شده بود
     try:
-        with open(abs_path, 'rb') as f:
-            await client.send_message(chat_id, text=caption, file=f)
-        return True
+        if hasattr(client, "send_file"):
+            await client.send_file(chat_id, abs_path, caption=caption)
+            return True
     except Exception as e:
-        error_logs.append(f"ByteObj: {e}")
+        error_logs.append(f"Rubpy Send_file: {e}")
 
-    # در صورت شکست، لیست ارورها را خروجی می‌دهد
     return "\n".join(error_logs)
 
 
@@ -219,13 +243,6 @@ SPEAKERS = {
     "28": ("مهتاب (زن)", "Kore"), "29": ("سام (مرد)", "Fenrir"), "30": ("لیدا (زن)", "Leda")
 }
 
-# --- تنظیمات کلیدها ---
-GEMINI_KEYS_STR = os.environ.get("GEMINI_API_KEYS", "")
-GEMINI_KEYS =[k.strip() for k in GEMINI_KEYS_STR.split(",") if k.strip()]
-
-HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
-HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
-
 user_states = {}
 
 
@@ -357,12 +374,10 @@ async def process_image(client, chat_id, prompt):
         await asyncio.sleep(1)
         caption_text = "🎨 تصویر شما آماده شد!\n(مدل: Z-Image-Turbo)"
         
-        # استفاده از تابع هوشمند برای آپلود
-        upload_result = await helper_upload_file(client, chat_id, file_name, "photo", caption_text)
+        # آپلود مستقیم به سرور اصلی روبیکا
+        upload_result = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
         
-        if upload_result is True:
-            await send_with_keyboard(client, chat_id, "✅ تصویر با موفقیت ارسال شد.", True)
-        else:
+        if upload_result is not True:
             err_msg = str(upload_result)[:800]
             await send_with_keyboard(client, chat_id, f"❌ تصویر در سرور ساخته شد، اما روبیکا در آپلود آن ارور داد.\n\n🔴 لاگ ارور برنامه‌نویس:\n`{err_msg}`", True)
             
@@ -418,8 +433,8 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
             
             await asyncio.sleep(1)
             
-            # استفاده از تابع هوشمند برای آپلود
-            upload_result = await helper_upload_file(client, chat_id, file_name, "voice", "")
+            # آپلود مستقیم به سرور اصلی روبیکا
+            upload_result = await helper_upload_file(client, chat_id, file_name, "Voice", "")
             
             if upload_result is True:
                 await send_with_keyboard(client, chat_id, "✅ صدای شما با موفقیت ارسال شد.", True)
@@ -525,8 +540,6 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
 
 
 # --- تنظیمات ربات روبیکا ---
-bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
-
 if not bot_token:
     print("خطا: توکن ربات روبیکا وارد نشده است!")
 else: