Update main.py

main.py

@@ -1,3 +1,6 @@
+# ==============================================================================
+# 🟢 پارت 1: وارد کردن کتابخانه‌ها و ماژول‌ها (Imports)
+# ==============================================================================
 import os
 import time
 import threading
@@ -14,12 +17,15 @@ import string
 import uuid
 import concurrent.futures
 from flask import Flask
-from rubpy.bot import BotClient, filters
+from rubpy.bot import BotClient, filters  # بازگردانی به نسخه اصلی شما
 
 from huggingface_hub import AsyncInferenceClient, HfApi, hf_hub_download
 from PIL import Image
 from pydub import AudioSegment
 
+# ==============================================================================
+# 🟢 پارت 2: تنظیمات سرورها، مدل‌های تغییر صدا و کلون صدا
+# ==============================================================================
 # --- تنظیمات آدرس سرورهای تغییر صدا ---
 VC_BASE_URL = "https://sada8888-sada.hf.space"
 LEGACY_BASE_URL = "https://ezmarynoori-rvc.hf.space"
@@ -42,6 +48,9 @@ STANDARD_MODELS = {
     "12": {"name": "مریم", "ref": "https://huggingface.co/datasets/Hamed744/mp3/resolve/main/%D8%B5%D8%AF%D8%A7%DB%8C%20%D8%AE%D8%A7%D9%86%D9%85.wav?download=true"}
 }
 
+# ==============================================================================
+# 🟢 پارت 3: متغیرهای ادمین، سیستم ضد اسپم و توابع تبدیل تاریخ
+# ==============================================================================
 # --- کد مدیریت ---
 ADMIN_CODE = "3011"
 BOT_GUID = None
@@ -54,19 +63,13 @@ user_message_times = {}
 def is_user_spamming(chat_id):
     now = time.time()
     times = user_message_times.get(chat_id,[])
-    times = [t for t in times if now - t < 3.0]
+    times =[t for t in times if now - t < 3.0]
     times.append(now)
     user_message_times[chat_id] = times
     if len(times) > 4:
         return True
     return False
 
-# --- سیستم دیتابیس حساب کاربری מתصل به دیتاست هاگینگ فیس ---
-DB_FILE = "users_db.json"
-DATASET_REPO = "opera8/Karbaran-rayegan-tedad"
-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
@@ -86,6 +89,15 @@ def gregorian_to_jalali(gy, gm, gd):
         jd = 1 + ((days - 186) % 30)
     return jy, jm, jd
 
+# ==============================================================================
+# 🟢 پارت 4: راه‌اندازی دیتابیس و همگام‌سازی با دیتاست هاگینگ فیس
+# ==============================================================================
+# --- سیستم دیتابیس حساب کاربری מתصل به دیتاست هاگینگ فیس ---
+DB_FILE = "users_db.json"
+DATASET_REPO = "opera8/Karbaran-rayegan-tedad"
+HF_TOKEN_DB = os.environ.get("HF_TOKEN")
+db_lock = threading.Lock()
+
 def load_db():
     print("در حال تلاش برای خواندن دیتابیس کاربران از دیتاست هاگینگ فیس...")
     if not HF_TOKEN_DB:
@@ -151,6 +163,9 @@ async def background_db_uploader():
 
 user_credits_db = load_db()
 
+# ==============================================================================
+# 🟢 پارت 5: توابع کد معرف، مدیریت سهمیه حساب کاربری و تبدیل اعداد
+# ==============================================================================
 def get_or_create_referral_code(chat_id):
     user_data = user_credits_db[chat_id]
     if not user_data.get("referral_code"):
@@ -238,6 +253,9 @@ def to_english_digits(text):
     translation_table = str.maketrans(persian_digits + arabic_digits, english_digits * 2)
     return str(text).translate(translation_table)
 
+# ==============================================================================
+# 🟢 پارت 6: تنظیمات وب‌سرور (Flask) و توابع کمکی فایل‌ها
+# ==============================================================================
 # --- تنظیمات وب سرور ---
 app = Flask(__name__)
 
@@ -268,6 +286,9 @@ def sync_combine_audio(current_audio, new_bytes):
 def sync_export_audio(audio_obj, file_name):
     audio_obj.export(file_name, format="mp3")
 
+# ==============================================================================
+# 🟢 پارت 7: کیبوردها (دکمه‌های شیشه‌ای / منوها)
+# ==============================================================================
 # --- ساختار کیبورد ---
 MAIN_KEYPAD_DICT = {
     "rows":[
@@ -345,6 +366,9 @@ JOIN_KEYPAD_DICT = {
     "resize_keyboard": True
 }
 
+# ==============================================================================
+# 🟢 پارت 8: تابع بررسی عضویت کانال، ارسال کیبورد و توکن روبیکا
+# ==============================================================================
 async def check_channel_membership(client, user_id):
     global CHANNEL_GUID
     if str(user_id) == str(BOT_GUID): return True
@@ -391,6 +415,9 @@ async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
 
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
+# ==============================================================================
+# 🟢 پارت 9: دانلودر قدرتمند و تابع ضد قطعی روبیکا
+# ==============================================================================
 # ==================================================================
 # تابع دانلود ضد بمب اتم (پافشاری ۳۵ بار)
 # ==================================================================
@@ -491,6 +518,9 @@ async def helper_download_url_to_bytes(url):
                 await asyncio.sleep(2)
     return None
 
+# ==============================================================================
+# 🟢 پارت 10: سیستم چرخش کلیدهای جیمینای و تنظیم توکن‌های هاگینگ فیس
+# ==============================================================================
 # ==================================================================
 # توکن‌ها و کلیدها
 # ==================================================================
@@ -522,17 +552,85 @@ 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()]
 
-# ==================================================================
-# تابع آپلود فایل (قدرتمند و ضدضربه) با حفظ قابلیت ارسال عکس 
-# ==================================================================
+# ==============================================================================
+# 🟢 پارت 11: آپلودر قدرتمند و تابع ارسال فایل ضد خطا (نسخه نهایی ضد بمب اتم 🚀)
+# ==============================================================================
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
-    error_logs =[]
+    error_logs = []
+    
+    api_file_type = "Image" if file_type in["photo", "Image", "image"] else "Voice" if file_type in ["voice", "Voice", "audio"] else file_type
     
-    api_file_type = "Image" if file_type in ["photo", "Image", "image"] else "Voice" if file_type in["voice", "Voice", "audio"] else file_type
-    if api_file_type == "Voice" and not abs_path.lower().endswith('.ogg'): api_file_type = "Music"
-    if api_file_type == "File" and abs_path.lower().endswith('.mp3'): api_file_type = "Music"
+    # روبیکا فرمت‌های خاصی را قبول می‌کند.
+    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"
         
+    # --- فاز اول (فوق سریع): استفاده از متدهای اختصاصی rubpy ---
+    for attempt in range(10):
+        try:
+            sent_success = False
+            
+            # تلاش برای ارسال عکس
+            if api_file_type == "Image" and hasattr(client, "send_photo"):
+                try:
+                    await client.send_photo(chat_id, photo=abs_path, caption=caption)
+                    sent_success = True
+                except TypeError:
+                    await client.send_photo(chat_id, abs_path, caption=caption)
+                    sent_success = True
+                    
+            # تلاش برای ارسال ویس
+            elif api_file_type == "Voice" and hasattr(client, "send_voice"):
+                try:
+                    await client.send_voice(chat_id, abs_path, caption=caption)
+                    sent_success = True
+                except TypeError:
+                    await client.send_voice(chat_id, file=abs_path, caption=caption)
+                    sent_success = True
+                    
+            # تلاش برای ارسال موزیک
+            elif api_file_type == "Music" and hasattr(client, "send_music"):
+                try:
+                    await client.send_music(chat_id, abs_path, caption=caption)
+                    sent_success = True
+                except TypeError:
+                    await client.send_music(chat_id, music=abs_path, caption=caption)
+                    sent_success = True
+
+            # *** سیستم ضد بمب اتم فاز 1 ***
+            # اگر هیچکدام از متدهای بالا وجود نداشت یا به هر دلیلی ارسال نشد، حتما به عنوان فایل (سند) بفرست تا معطل نشود
+            if not sent_success:
+                if hasattr(client, "send_document"):
+                    try:
+                        await client.send_document(chat_id, document=abs_path, caption=caption)
+                        sent_success = True
+                    except TypeError:
+                        await client.send_document(chat_id, abs_path, caption=caption)
+                        sent_success = True
+                elif hasattr(client, "send_file"):
+                    try:
+                        await client.send_file(chat_id, file=abs_path)
+                    except TypeError:
+                        await client.send_file(chat_id, abs_path)
+                    if caption:
+                        await send_with_keyboard(client, chat_id, caption, True)
+                    sent_success = True
+
+            if sent_success:
+                return True
+                    
+        except Exception as e:
+            err_msg = str(e)[:150]
+            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)
+            else: 
+                await asyncio.sleep(2)
+
+    # --- فاز دوم (سنگر آخر): آپلود دستی (Raw HTTP) در صورت مسدودی کامل روب‌پای ---
     for attempt in range(15):
         try:
             url_request = f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile"
@@ -555,20 +653,34 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                 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}")
-                                        await asyncio.sleep(3) if up_resp.status in[502, 503, 500] else 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()
+                                    try: 
+                                        up_data = await up_resp.json()
                                     except Exception as json_err:
                                         error_logs.append(f"JSON Decode Error: {json_err}")
                                         await asyncio.sleep(2)
                                         continue
                                     
                                     final_file_id = None
+                                    
+                                    # *** سیستم ضد بمب اتم فاز 2 ***
+                                    # دور زدن ارور روبیکا: اگر روبیکا به فرمت فایل گیر داد، سریعاً نوع را به File تغییر بده تا بدون بررسی فرمت آپلود کند!
+                                    if up_data.get("status") == "INVALID_INPUT" and "format" in str(up_data).lower():
+                                        api_file_type = "File"
+                                        error_logs.append(f"Format Blocked! Downgrading type to File.")
+                                        await asyncio.sleep(1)
+                                        continue
+
                                     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")
+                                        elif "file_id" in up_data: 
+                                            final_file_id = up_data.get("file_id")
                                             
                                     if final_file_id:
                                         send_payload = {
@@ -588,8 +700,8 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                             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"
+                                        if up_data.get("status") == "INVALID_INPUT" and api_file_type in ["Voice", "Music"]:
+                                            api_file_type = "File" # دور زدن گیر دادن روبیکا
                                         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"Request API Error: {req_data}")
@@ -597,37 +709,11 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
             error_logs.append(f"Raw HTTP Error: {str(e)[:100]}")
             await asyncio.sleep(2.5)
 
-    # روش دوم: استفاده از متدهای اختصاصی rubpy برای ارسال عکس/فایل (جلوگیری از ارسال عکس به صورت داکیومنت)
-    for attempt in range(10):
-        try:
-            if api_file_type == "Image" and hasattr(client, "send_photo"):
-                await client.send_photo(chat_id, photo=abs_path, caption=caption)
-                return True
-            elif api_file_type in ["Voice", "Music"]:
-                if hasattr(client, "send_voice"):
-                    await client.send_voice(chat_id, voice=abs_path, caption=caption)
-                    return True
-                elif hasattr(client, "send_music"):
-                    await client.send_music(chat_id, music=abs_path, caption=caption)
-                    return True
-            
-            # در صورتی که هیچ‌کدام از شروط بالا برقرار نبود به عنوان فایل ارسال شود
-            if hasattr(client, "send_document"): 
-                await client.send_document(chat_id, document=abs_path, caption=caption)
-                return True
-            elif 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
-        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)
-            else: await asyncio.sleep(2)
-
     return "\n".join(error_logs[-6:])
 
+# ==============================================================================
+# 🟢 پارت 12: توابع تغییر صدا و لیست گویندگان
+# ==============================================================================
 # ==================================================================
 # لیست‌های اولیه ربات
 # ==================================================================
@@ -804,6 +890,9 @@ async def process_legacy_vc_job(client, chat_id, src_bytes, model_url, pitch, mo
         
     if os.path.exists(file_name_mp3): os.remove(file_name_mp3)
 
+# ==============================================================================
+# 🟢 پارت 13: پردازش هوش مصنوعی متنی و تصویری جیمینای و هاگینگ‌فیس
+# ==============================================================================
 # ==================================================================
 # سایر توابع پردازشی
 # ==================================================================
@@ -967,6 +1056,9 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
     except Exception:
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد.", False)
 
+# ==============================================================================
+# 🟢 پارت 14: ساخت عکس با هوش مصنوعی
+# ==============================================================================
 async def process_image(client, chat_id, prompt, size_choice="1"):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -1096,6 +1188,9 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
     except Exception as e: 
         await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره و ارسال عکس:\n{str(e)[:150]}", True)
 
+# ==============================================================================
+# 🟢 پارت 15: توابع ترجمه و ویرایش عکس با هوش مصنوعی (Flux.2)
+# ==============================================================================
 async def translate_text_aloha(prompt_text):
     session_hash = ''.join(random.choices(string.ascii_lowercase + string.digits, k=11))
     join_url = "https://hamed744-translate-tts-aloha.hf.space/gradio_api/queue/join"
@@ -1203,6 +1298,9 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     except Exception as e:
         await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره عکس ویرایش شده:\n{str(e)[:150]}", True)
 
+# ==============================================================================
+# 🟢 پارت 16: ساخت صدا از روی متن (تبدیل متن به صدا - TTS)
+# ==============================================================================
 async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -1272,6 +1370,9 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
             await send_with_keyboard(client, chat_id, f"❌ سرورها درگیر هستند.\nدلیل: {last_error}", True)
     except Exception: traceback.print_exc()
 
+# ==============================================================================
+# 🟢 پارت 17: سیستم ساخت پادکست
+# ==============================================================================
 async def process_podcast(client, chat_id, prompt):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -1375,6 +1476,9 @@ async def process_podcast(client, chat_id, prompt):
             
         if os.path.exists(file_name_mp3): os.remove(file_name_mp3)
 
+# ==============================================================================
+# 🟢 پارت 18: استخراج متن از صدا (STT) و تحلیل انواع فایل با هوش مصنوعی
+# ==============================================================================
 async def process_stt(client, chat_id, audio_bytes, file_name):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -1519,6 +1623,9 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
         else:
             await send_with_keyboard(client, chat_id, "❌ تمامی سرورها شلوغ هستند. لطفاً بعداً امتحان کنید.", True)
 
+# ==============================================================================
+# 🟢 پارت 19: ساخت مقاله و خروجی فایل متنی (Word/PDF)
+# ==============================================================================
 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)
@@ -1683,6 +1790,9 @@ async def process_create_file(client, chat_id, topic):
     else:
         await send_with_keyboard(client, chat_id, "❌ متأسفانه پس از تلاش‌های مکرر، سرور قادر به ساخت و ارسال هیچ‌یک از فایل‌ها نبود و اعتباری از شما کسر نشد.", True)
 
+# ==============================================================================
+# 🟢 پارت 20: بدنه اصلی ربات (Handler)، دستورات مدیریتی، حلقه اصلی و استارت
+# ==============================================================================
 # ==================================================================
 # حلقه و مدیریت اصلی ربات
 # ==================================================================