Update main.py

main.py

@@ -13,7 +13,7 @@ import datetime
 import string
 import uuid
 from flask import Flask
-from rubpy import BotClient, filters
+from rubpy.bot import BotClient, filters
 
 # ایمپورت‌های جدید برای هوش مصنوعی، کار با فایل صوتی و دیتابیس هاگینگ فیس
 from huggingface_hub import AsyncInferenceClient, HfApi, hf_hub_download
@@ -229,6 +229,11 @@ MAIN_KEYPAD_DICT = {
                 {"id": "file_btn", "type": "Simple", "button_text": "تحلیل فایل 📁"}
             ]
         },
+        {
+            "buttons":[
+                {"id": "create_file_btn", "type": "Simple", "button_text": "ساخت فایل 📄"}
+            ]
+        },
         {
             "buttons":[
                 {"id": "account_btn", "type": "Simple", "button_text": "حساب کاربری 👤"},
@@ -430,82 +435,114 @@ HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
 
-# --- 🚨 تابع اختصاصی آپلود فایل به روبیکا (نسخه اصلاح شده) 🚨 ---
+# --- 🚨 تابع اختصاصی آپلود فایل به روبیکا (مجهز به سیستم Fallback هوشمند صوتی و Music) 🚨 ---
 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"
     
-    try:
-        url_request = f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile"
-        url_send = f"https://botapi.rubika.ir/v3/{bot_token}/sendFile"
+    # اگر قرار است ویس ارسال شود اما فایل پسوند مناسب ندارد، برای جلوگیری از ارور فوراً به Music تبدیل می‌کنیم
+    if api_file_type == "Voice" and not abs_path.lower().endswith('.ogg'):
+        api_file_type = "Music"
         
-        async with aiohttp.ClientSession() as session:
-            # ۱. درخواست آدرس آپلود از روبیکا
-            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":
-                    data_dict = req_data.get("data", {})
-                    upload_url = data_dict.get("upload_url")
+    # سیستم ۳ بار تلاش مجدد برای جلوگیری از قطعی سرور روبیکا
+    for attempt in range(3):
+        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:
+                    if resp.status != 200:
+                        error_logs.append(f"Request HTTP {resp.status}")
+                        await asyncio.sleep(1.5)
+                        continue
+                        
+                    req_data = await resp.json()
                     
-                    # اگر آدرس آپلود با موفقیت دریافت شد
-                    if upload_url:
-                        with open(abs_path, "rb") as f:
-                            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:
-                                up_data = await up_resp.json()
+                    if req_data.get("status") == "OK":
+                        upload_url = req_data.get("data", {}).get("upload_url")
+                        
+                        # اگر آدرس آپلود با موفقیت دریافت شد
+                        if upload_url:
+                            with open(abs_path, "rb") as f:
+                                form = aiohttp.FormData()
+                                form.add_field('file', f, filename=os.path.basename(abs_path))
                                 
-                                # ۳. گرفتن file_id از پاسخی که سرور آپلود برمی‌گرداند
-                                final_file_id = None
-                                if up_data.get("status") == "OK" or up_data.get("status_det") == "OK":
-                                    if "data" in up_data:
-                                        final_file_id = up_data["data"].get("file_id") or up_data["data"].get("id")
-                                    elif "file_id" in up_data:
-                                        final_file_id = up_data.get("file_id")
+                                # ۲. ارسال فایل به لینک آپلود
+                                async with session.post(upload_url, data=form, timeout=60) as up_resp:
+                                    if up_resp.status != 200:
+                                        error_logs.append(f"Upload HTTP {up_resp.status}")
+                                        # 🔴 اگر ارور 502 داد یعنی روبیکا فرمت ویس را پس زده است، پس تبدیل به حالت Music می‌کنیم
+                                        if up_resp.status == 502 and api_file_type == "Voice":
+                                            api_file_type = "Music" 
+                                        await asyncio.sleep(1.5)
+                                        continue
                                         
-                                if final_file_id:
-                                    # ۴. ارسال نهایی پیام (به همراه فایل) به کاربر
-                                    send_payload = {
-                                        "chat_id": str(chat_id),
-                                        "file_id": str(final_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 (No file_id): {up_data}")
+                                    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)
+                                        continue
+                                    
+                                    # ۳. گرفتن file_id از پاسخی که سرور آپلود برمی‌گرداند
+                                    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):
+                                            final_file_id = up_data["data"].get("file_id") or up_data["data"].get("id")
+                                        elif "file_id" in up_data:
+                                            final_file_id = up_data.get("file_id")
+                                            
+                                    if final_file_id:
+                                        # ۴. ارسال نهایی پیام
+                                        send_payload = {
+                                            "chat_id": str(chat_id),
+                                            "file_id": str(final_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:
+                                            if send_resp.status != 200:
+                                                error_logs.append(f"Send HTTP {send_resp.status}")
+                                                await asyncio.sleep(1.5)
+                                                continue
+                                                
+                                            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:
+                                        # 🔴 اگر فرمت فایل نامعتبر بود، به جای ویس، به صورت موزیک (دارای دکمه پخش) ارسال کن
+                                        if up_data.get("status") == "INVALID_INPUT" and api_file_type == "Voice":
+                                            api_file_type = "Music"
+                                        error_logs.append(f"Upload API Error: {up_data}")
+                        else:
+                            error_logs.append(f"Missing upload_url in Response: {req_data}")
                     else:
-                        error_logs.append(f"Missing upload_url in Response: {data_dict}")
-                else:
-                    error_logs.append(f"Request API Error: {req_data}")
-    except Exception as e:
-        error_logs.append(f"Raw HTTP Error: {e}")
+                        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)
 
-    # در صورت شکست، تلاش از طریق متدهای کتابخانه rubpy (Fallback)
+    # تلاش نهایی از طریق روب‌پای به عنوان راه جایگزین
     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"): # استفاده از متدهای مدرن‌تر در آپدیت‌های جدید rubpy
+        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: {e}")
+        error_logs.append(f"Rubpy Send Error: {str(e)[:100]}")
 
-    return "\n".join(error_logs)
+    return "\n".join(error_logs[-4:])
 
 
 WORKER_URLS =[
@@ -906,20 +943,20 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
                 user_credits_db[str_chat_id]["tts"] -= 1
                 save_db(user_credits_db)
             
-            file_name = f"audio_{uuid.uuid4().hex}.mp3"
-            with open(file_name, "wb") as f: f.write(audio_bytes)
-            await asyncio.sleep(1)
+            # فایل صوتی را فقط به صورت MP3 ذخیره می‌کنیم تا از مشکل 502 روبیکا جلوگیری شود
+            file_name_mp3 = f"audio_{uuid.uuid4().hex}.mp3"
             
-            # ارسال به صورت ویس
-            upload_result_voice = await helper_upload_file(client, chat_id, file_name, "Voice", "✅ صدای شما (به صورت ویس) آماده شد.")
+            with open(file_name_mp3, "wb") as f: f.write(audio_bytes)
             
-            # ارسال به صورت فایل MP3
-            upload_result_file = await helper_upload_file(client, chat_id, file_name, "File", "✅ فایل MP3 صدای شما:")
+            await asyncio.sleep(1)
+            
+            # ارسال فقط به صورت فایل دانلودشونده MP3 (حالت Voice حذف شد)
+            upload_result_file = await helper_upload_file(client, chat_id, file_name_mp3, "File", "✅ صدای شما با موفقیت آماده شد (فایل MP3):")
             
-            if upload_result_voice is not True and upload_result_file is not True: 
-                await send_with_keyboard(client, chat_id, f"❌ ارور آپلود:\n`{str(upload_result_voice)[:800]}`", True)
+            if upload_result_file is not True: 
+                await send_with_keyboard(client, chat_id, f"❌ ارور آپلود فایل:\n`{str(upload_result_file)[:800]}`", True)
                 
-            if os.path.exists(file_name): os.remove(file_name)
+            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)
     except Exception: traceback.print_exc()
 
@@ -968,7 +1005,7 @@ async def process_podcast(client, chat_id, prompt):
             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: pass
         
         proc_msg = await send_with_keyboard(client, chat_id, f"🎙 سناریو نوشته شد! در حال ریکورد و میکس دیالوگ‌های گویندگان (شامل {len(script_data)} نوبت صحبت)...\nاین مرحله زمان‌بر است...", False)
@@ -999,8 +1036,9 @@ async def process_podcast(client, chat_id, prompt):
             user_credits_db[str_chat_id]["podcast"] -= 1
             save_db(user_credits_db)
 
-        file_name = f"final_podcast_{uuid.uuid4().hex}.mp3"
-        combined_audio.export(file_name, format="mp3")
+        # فقط با فرمت MP3 خروجی می‌دهیم (بدون تبدیل OGG)
+        file_name_mp3 = f"final_podcast_{uuid.uuid4().hex}.mp3"
+        combined_audio.export(file_name_mp3, format="mp3")
         
         try:
             if proc_msg:
@@ -1009,16 +1047,15 @@ async def process_podcast(client, chat_id, prompt):
                 if msg_id: await client.delete_messages(chat_id,[msg_id])
         except: pass
 
-        caption_voice = f"🎧 پادکست شما با موفقیت میکس و ساخته شد! (ویس)\n\n💡 موضوع شما: {prompt}"
         caption_file = f"🎧 فایل پادکست شما با فرمت MP3:\n\n💡 موضوع شما: {prompt}"
         
-        upload_result_voice = await helper_upload_file(client, chat_id, file_name, "Voice", caption_voice)
-        upload_result_file = await helper_upload_file(client, chat_id, file_name, "File", caption_file)
+        # ارسال فقط به صورت فایل MP3
+        upload_result_file = await helper_upload_file(client, chat_id, file_name_mp3, "File", caption_file)
         
-        if upload_result_voice is not True and upload_result_file is not True: 
-            await send_with_keyboard(client, chat_id, f"❌ پادکست ساخته شد اما روبیکا خطای آپلود داد.\n\n`{str(upload_result_voice)[:800]}`", True)
+        if upload_result_file is not True: 
+            await send_with_keyboard(client, chat_id, f"❌ پادکست ساخته شد اما روبیکا خطای آپلود داد.\n\n`{str(upload_result_file)[:800]}`", True)
             
-        if os.path.exists(file_name): os.remove(file_name)
+        if os.path.exists(file_name_mp3): os.remove(file_name_mp3)
 
 
 # --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن ---
@@ -1116,13 +1153,142 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
         await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", True)
 
 
+# --- ۵.۵. پردازش ساخت مقاله و تبدیل به فایل (PDF & DOCX) ---
+async def process_create_file(client, chat_id, topic):
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
+    creds = get_user_credits(str_chat_id)
+    
+    # بررسی محدودیت بخش ساخت فایل (کسر از اعتبار پیام چت)
+    if creds["chat"] <= 0:
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار چت شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
+
+    if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
+    
+    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
+
+    # Fallback to Hugging Face
+    if not article_text and HF_TOKENS:
+        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 not article_text:
+        return await send_with_keyboard(client, chat_id, "❌ سرور تولید ��تن شلوغ است، لطفاً بعداً تلاش کنید.", True)
+
+    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])
+    except Exception: pass
+
+    proc_msg = await send_with_keyboard(client, chat_id, "📄 مقاله با موفقیت نوشته شد! در حال ارتباط با سرور و تبدیل به فایل‌های PDF و Word...\n(لطفا صبور باشید)", False)
+
+    pdf_bytes = None
+    docx_bytes = None
+    converter_url = "https://opera8-texttopdf.hf.space/"
+
+    async with aiohttp.ClientSession() as session:
+        # دریافت خروجی PDF
+        try:
+            form_data_pdf = aiohttp.FormData()
+            form_data_pdf.add_field('content', article_text)
+            form_data_pdf.add_field('format', 'pdf')
+            async with session.post(converter_url, data=form_data_pdf, timeout=60) as resp:
+                if resp.status == 200:
+                    pdf_bytes = await resp.read()
+        except Exception as e: print("PDF creation error:", e)
+
+        # دریافت خروجی DOCX
+        try:
+            form_data_docx = aiohttp.FormData()
+            form_data_docx.add_field('content', article_text)
+            form_data_docx.add_field('format', 'docx')
+            async with session.post(converter_url, data=form_data_docx, timeout=60) as resp:
+                if resp.status == 200:
+                    docx_bytes = await resp.read()
+        except Exception as e: print("DOCX creation error:", e)
+
+    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])
+    except Exception: pass
+
+    if not pdf_bytes and not docx_bytes:
+        return await send_with_keyboard(client, chat_id, "❌ متاسفانه در تبدیل متن به فایل خطایی رخ داد.", True)
+
+    # کسر اعتبار ساخت فایل از سهمیه "چت" کاربر در صورت غیر پرمیوم بودن
+    if not creds.get("is_premium"):
+        user_credits_db[str_chat_id]["chat"] -= 1
+        save_db(user_credits_db)
+
+    uid = uuid.uuid4().hex
+    pdf_filename = f"Article_{uid}.pdf"
+    docx_filename = f"Article_{uid}.docx"
+
+    uploaded_any = False
+    
+    if pdf_bytes:
+        with open(pdf_filename, "wb") as f:
+            f.write(pdf_bytes)
+        res = await helper_upload_file(client, chat_id, pdf_filename, "File", f"📄 فایل PDF مقاله شما:\n\n💡 موضوع: {topic}")
+        if res is True: uploaded_any = True
+        if os.path.exists(pdf_filename): os.remove(pdf_filename)
+        await asyncio.sleep(1.5)
+
+    if docx_bytes:
+        with open(docx_filename, "wb") as f:
+            f.write(docx_bytes)
+        res = await helper_upload_file(client, chat_id, docx_filename, "File", f"📝 فایل Word (DOCX) مقاله شما:\n\n💡 موضوع: {topic}")
+        if res is True: uploaded_any = True
+        if os.path.exists(docx_filename): os.remove(docx_filename)
+
+    if uploaded_any:
+        await send_with_keyboard(client, chat_id, "✅ مقاله شما با موفقیت به صورت فایل تحویل داده شد!", True)
+    else:
+        await send_with_keyboard(client, chat_id, "❌ فایل‌ها ساخته شدند اما روبیکا در ارسال آن‌ها دچار مشکل شد.", True)
+
+
 # --- تنظیمات ربات روبیکا ---
 if not bot_token:
     print("خطا: توکن ربات روبیکا وارد نشده است!")
 else:
     bot = BotClient(bot_token)
 
-    @bot.on_update()
+    @bot.on_update(filters.private)
     async def main_handler(client, update):
         global BOT_GUID
         
@@ -1476,6 +1642,11 @@ else:
                 await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل خود (ویس، آهنگ، ویدیو و...) را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
+            if user_text_str in["/create_file", "ساخت فایل 📄"]:
+                user_states[chat_id]["mode"] = "create_file_waiting_for_topic"
+                await send_with_keyboard(client, chat_id, "📄 شما وارد بخش **ساخت فایل** شدید.\n\nلطفاً موضوع مقاله‌ای که می‌خواهید را کامل بفرستید.\nمثال: نحوه مدیریت زمان\n\nهوش مصنوعی یک مقاله کامل و طولانی نوشته و در نهایت فایل PDF و Word آن را به شما تحویل می‌دهد.\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                return
+
             current_mode = user_states[chat_id].get("mode")
 
             if current_mode is None:
@@ -1585,6 +1756,13 @@ else:
                 else: await send_with_keyboard(client, chat_id, "⚠️ لطفاً درخواست خود را متنی بنویسید.", False)
                 return
 
+            elif current_mode == "create_file_waiting_for_topic":
+                if user_text_str:
+                    asyncio.create_task(process_create_file(client, chat_id, user_text_str))
+                else: 
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً موضوع مقاله خود را به صورت متنی بفرستید.", False)
+                return
+
         except Exception: traceback.print_exc()
 
 if __name__ == "__main__":