Update main.py

main.py

@@ -79,7 +79,6 @@ def load_db():
     print("در حال تلاش برای خواندن دیتابیس کاربران از دیتاست هاگینگ فیس...")
     if HF_TOKEN_DB:
         try:
-            # دانلود دیتابیس از دیتاست خصوصی
             file_path = hf_hub_download(
                 repo_id=DATASET_REPO, 
                 filename=DB_FILE, 
@@ -90,9 +89,8 @@ def load_db():
                 print("✅ دیتابیس با موفقیت از دیتاست هاگینگ فیس لود شد.")
                 return json.load(f)
         except Exception as e:
-            print("⚠️ فایل دیتابیس در هاگینگ فیس یافت نشد یا خطایی رخ داد (ساخت دیتابیس جدید). جزئیات:", str(e)[:100])
+            print("⚠️ فایل دیتابیس در هاگینگ فیس یافت نشد یا خطایی رخ داد (ساخت دیتابیس جدید).")
     
-    # اگر از دیتاست لود نشد، از لوکال بخوان
     if os.path.exists(DB_FILE):
         try:
             with open(DB_FILE, "r", encoding="utf-8") as f:
@@ -102,7 +100,6 @@ def load_db():
     return {}
 
 def _upload_db_background():
-    """تابع کمکی برای آپلود دیتابیس بدون درگیر کردن سرعت ربات"""
     if HF_TOKEN_DB:
         api = HfApi(token=HF_TOKEN_DB)
         try:
@@ -118,24 +115,40 @@ def _upload_db_background():
 def save_db(db_data):
     with db_lock:
         try:
-            # ذخیره فایل به صورت لوکال
             with open(DB_FILE, "w", encoding="utf-8") as f:
                 json.dump(db_data, f, ensure_ascii=False, indent=4)
-            
-            # استارت کردن یک Thread جدید برای آپلود به دیتاست تا ربات منتظر نماند
             threading.Thread(target=_upload_db_background, daemon=True).start()
         except Exception as e:
             print("خطا در ذخیره دیتابیس:", e)
 
 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:
+                user_data["referral_code"] = new_code
+                save_db(user_credits_db)
+                break
+    return user_data["referral_code"]
+
+def find_user_by_referral_code(code):
+    code = code.strip()
+    for uid, data in user_credits_db.items():
+        if data.get("referral_code", "") == code:
+            return uid
+    return None
+
 def get_user_credits(chat_id):
-    # اطمینان از پاک بودن استرینگ از علائم مزاحم
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     today_str = datetime.date.today().isoformat()
     
     if str_chat_id not in user_credits_db:
-        # ساخت کاربر جدید با مقادیر اولیه
         user_credits_db[str_chat_id] = {
             "is_premium": False,
             "expire_date": None,
@@ -147,30 +160,29 @@ def get_user_credits(chat_id):
             "tts": 5,
             "file": 1,
             "stt": 5,
-            "has_joined": False
+            "has_joined": False,
+            "invited_count": 0,
+            "used_referral": False,
+            "referral_code": ""
         }
         save_db(user_credits_db)
         
     user_data = user_credits_db[str_chat_id]
     is_premium = user_data.get("is_premium", False)
     
-    # 🛡 بررسی انقضای اشتراک ماهانه کاربران ویژه 🛡
     if is_premium and user_data.get("expire_date"):
         try:
             expire_date = datetime.datetime.fromisoformat(user_data["expire_date"])
             if datetime.datetime.now() > expire_date:
-                # اگر زمان اشتراک تمام شده باشد، دوباره رایگان می‌شود
                 user_data["is_premium"] = False
                 user_data["expire_date"] = None
                 is_premium = False
         except Exception:
             pass
 
-    # 🔄 سیستم شارژ روزانه (فقط و فقط برای نسخه رایگان) 🔄
     if not is_premium:
         if user_data.get("last_reset") != today_str:
             user_data["last_reset"] = today_str
-            # شارژ روزانه نسخه رایگان
             user_data["chat"] = 10
             user_data["image"] = 5
             user_data["edit_image"] = 1
@@ -197,7 +209,7 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو + مدیریت اشتراک نامحدود + دیتاست + سیستم ضد باگ) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (نسخه پرو + مدیریت اشتراک نامحدود + دعوت دوستان + سیستم ضد باگ) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -240,6 +252,12 @@ MAIN_KEYPAD_DICT = {
                 {"id": "buy_btn", "type": "Simple", "button_text": "خرید اشتراک 💎"}
             ]
         },
+        {
+            "buttons":[
+                {"id": "invite_btn", "type": "Simple", "button_text": "دعوت دوستان 🎁"},
+                {"id": "referral_btn", "type": "Simple", "button_text": "ثبت کد هدیه 🎫"}
+            ]
+        },
         {
             "buttons":[
                 {"id": "transfer_btn", "type": "Simple", "button_text": "انتقال اکانت از برنامه به ربات"}
@@ -254,7 +272,6 @@ MAIN_KEYPAD_DICT = {
     "resize_keyboard": True
 }
 
-# --- ⛔️ سیستم بررسی عضویت اجباری کانال (Force Join) ⛔️ ---
 CHANNEL_USERNAME = "aialpha"
 CHANNEL_GUID = None
 
@@ -271,7 +288,6 @@ JOIN_KEYPAD_DICT = {
 
 async def check_channel_membership(client, user_id):
     global CHANNEL_GUID
-    # جلوگیری از بلاک شدن خود ربات
     if str(user_id) == str(BOT_GUID): return True
     
     try:
@@ -293,7 +309,7 @@ async def check_channel_membership(client, user_id):
                     CHANNEL_GUID = getattr(res.exist.chat, 'object_guid', None)
 
         if not CHANNEL_GUID:
-            return True # در صورت عدم یافتن آیدی کانال، اجازه عبور داده می‌شود
+            return True
 
         payload = {
             "channel_guid": CHANNEL_GUID,
@@ -312,10 +328,9 @@ async def check_channel_membership(client, user_id):
         
         return False
     except Exception:
-        return False # فرض بر این است که عضو نیست
+        return False
 
 
-# --- تابع ارسال منحصراً برای کیبورد پایین صفحه ---
 async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
     try:
         if not use_keyboard:
@@ -335,7 +350,6 @@ async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
             return None
 
 
-# --- 🚨 تابع هوشمند دانلود فایل (با استفاده از UUID جهت جلوگیری از تداخل حریم خصوصی) 🚨 ---
 async def helper_download_file(client, msg_obj):
     errors =[]
     file_id = None
@@ -348,7 +362,6 @@ async def helper_download_file(client, msg_obj):
             elif isinstance(val, dict) and 'file_id' in val: file_id = val['file_id']
             if file_id: break
 
-    # 🛡 استفاده از UUID به جای اعداد تصادفی برای تضمین ۱۰۰ درصدی عدم تداخل نام فایل‌ها 🛡
     temp_name = f"temp_dl_{uuid.uuid4().hex}.tmp"
 
     if hasattr(client, "download_file"):
@@ -389,7 +402,6 @@ async def helper_download_file(client, msg_obj):
     raise Exception("دانلود ناموفق. لاگ:\n" + "\n".join(errors))
 
 
-# --- تنظیمات کلیدها ---
 GEMINI_KEYS_STR1 = os.environ.get("GEMINI_API_KEYS1", "")
 GEMINI_KEYS_STR2 = os.environ.get("GEMINI_API_KEYS2", "")
 
@@ -399,21 +411,13 @@ if GEMINI_KEYS_STR1:
 if GEMINI_KEYS_STR2:
     _raw_keys.extend(GEMINI_KEYS_STR2.split(","))
 
-# حذف کلیدهای تکراری و خالی
 GEMINI_KEYS = list(set([k.strip() for k in _raw_keys if k.strip()]))
 print(f"✅ تعداد {len(GEMINI_KEYS)} کلید جیمینای با موفقیت شناسایی شد.")
 
-# --- 🚀 سیستم چرخشی پیشرفته (Round-Robin) برای کلیدهای جیمینای ---
 current_gemini_key_index = 0
 gemini_key_lock = threading.Lock()
 
 def get_next_gemini_keys(count=100):
-    """
-    این تابع تضمین می‌کند که به ترتیب از کلیدها استفاده شود.
-    مثلاً درخواست اول کلیدهای ۱ تا ۱۰۰ را می‌گیرد.
-    درخواست دوم کلیدهای ۱۰۱ تا ۲۰۰ را می‌گیرد.
-    و وقتی به آخر لیست برسد، دوباره از اول شروع می‌کند.
-    """
     global current_gemini_key_index
     with gemini_key_lock:
         total_keys = len(GEMINI_KEYS)
@@ -435,25 +439,21 @@ 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"
     
-    # اگر قرار است ویس ارسال شود اما فایل پسوند مناسب ندارد، برای جلوگیری از ارور فوراً به Music تبدیل می‌کنیم
     if api_file_type == "Voice" and not abs_path.lower().endswith('.ogg'):
         api_file_type = "Music"
         
-    # سیستم ۳ بار تلاش مجدد برای جلوگیری از قطعی سرور روبیکا
     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}")
@@ -465,17 +465,14 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                     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))
                                 
-                                # ۲. ارسال فایل به لینک آپلود
                                 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)
@@ -488,7 +485,6 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                         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):
@@ -497,7 +493,6 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                             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),
@@ -517,7 +512,6 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                             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}")
@@ -529,7 +523,6 @@ 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(1.5)
 
-    # تلاش نهایی از طریق روب‌پای به عنوان راه جایگزین
     try:
         if hasattr(client, "send_file"):
             await client.send_file(chat_id, abs_path)
@@ -567,7 +560,6 @@ processed_message_ids = set()
 user_last_request_time = {}
 
 
-# --- ۱. پردازش چت متنی و چندرسانه‌ای ---
 async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=None):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -604,7 +596,6 @@ 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
     
@@ -622,7 +613,6 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
                         except (KeyError, IndexError): continue 
             except Exception: continue
 
-    # --- Fallback to Hugging Face Gemma 4 ---
     if not final_answer and HF_TOKENS:
         hf_messages =[]
         for msg in history:
@@ -673,7 +663,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)
@@ -704,7 +693,6 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد.", False)
 
 
-# --- ۲. پردازش ساخت عکس ---
 async def process_image(client, chat_id, prompt):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -718,7 +706,6 @@ async def process_image(client, chat_id, prompt):
     gemini_sys_prompt = f"You are an expert AI image generation prompt engineer. Translate the following user input to English, and enhance it with high-quality, highly detailed, 4k resolution, cinematic lighting, and visually striking descriptive keywords. Return ONLY the final English prompt string.\nUser input: {prompt}"
     
     if GEMINI_KEYS:
-        # دریافت ۱۰۰ کلید به صورت چرخشی
         keys_to_try_gemini = get_next_gemini_keys(100)
         async with aiohttp.ClientSession() as session:
             for key in keys_to_try_gemini:
@@ -732,7 +719,6 @@ async def process_image(client, chat_id, prompt):
                             break
                 except Exception: continue
 
-    # Fallback to HF Router if Gemini fails
     if enhanced_prompt == prompt and HF_TOKENS:
         keys_to_try_hf = HF_TOKENS.copy()
         random.shuffle(keys_to_try_hf)
@@ -758,7 +744,7 @@ async def process_image(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 Exception: pass
     
     short_preview = enhanced_prompt[:150] + "..." if len(enhanced_prompt) > 150 else enhanced_prompt
@@ -787,7 +773,6 @@ async def process_image(client, chat_id, prompt):
 
     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)
 
@@ -802,7 +787,7 @@ async def process_image(client, chat_id, prompt):
         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)
 
-# --- تابع جدید برای ارسال متن به اسپیس مترجم ---
+
 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"
@@ -814,12 +799,10 @@ async def translate_text_aloha(prompt_text):
     
     try:
         async with aiohttp.ClientSession() as session:
-            # پیوستن به صف ترجمه
             async with session.post(join_url, json=payload, timeout=20) as resp:
                 if resp.status != 200:
                     return prompt_text
                     
-            # دریافت داده‌ها به صورت جریانی (Stream)
             data_url = f"https://hamed744-translate-tts-aloha.hf.space/gradio_api/queue/data?session_hash={session_hash}"
             async with session.get(data_url, timeout=60) as resp:
                 async for line_bytes in resp.content:
@@ -838,7 +821,7 @@ async def translate_text_aloha(prompt_text):
         
     return prompt_text
 
-# --- ۲.۵. پردازش ویرایش عکس با FLUX.2-dev ---
+
 async def process_image_edit(client, chat_id, image_bytes, prompt):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -850,7 +833,6 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
 
     proc_msg = await send_with_keyboard(client, chat_id, "🪄 در حال ترجمه دستور شما توسط اسپیس و اعمال جادوی FLUX.2...\n(این فرآیند ممکن است کمی طول بکشد)", False)
     
-    # گرفتن ترجمه از اسپیس درخواستی
     translated_prompt = await translate_text_aloha(prompt)
     if not translated_prompt or translated_prompt.strip() == "":
         translated_prompt = prompt
@@ -883,7 +865,6 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     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)
 
@@ -901,7 +882,6 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
         await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره عکس ویرایش شده:\n{str(e)[:150]}", True)
 
 
-# --- ۳. پردازش ساخت صدا ---
 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)
@@ -938,19 +918,15 @@ 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)
             
-            # فایل صوتی را فقط به صورت MP3 ذخیره می‌کنیم تا از مشکل 502 روبیکا جلوگیری شود
             file_name_mp3 = f"audio_{uuid.uuid4().hex}.mp3"
             
             with open(file_name_mp3, "wb") as f: f.write(audio_bytes)
             
             await asyncio.sleep(1)
-            
-            # ارسال فقط به صورت فایل دانلودشونده MP3 (حالت Voice حذف شد)
             upload_result_file = await helper_upload_file(client, chat_id, file_name_mp3, "File", "✅ صدای شما با موفقیت آماده شد (فایل MP3):")
             
             if upload_result_file is not True: 
@@ -961,7 +937,6 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
     except Exception: traceback.print_exc()
 
 
-# --- ۳.۵. پردازش پادکست ---
 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)
@@ -1031,12 +1006,10 @@ async def process_podcast(client, chat_id, prompt):
                 combined_audio += audio_segment
             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)
 
-        # فقط با فرمت MP3 خروجی می‌دهیم (بدون تبدیل OGG)
         file_name_mp3 = f"final_podcast_{uuid.uuid4().hex}.mp3"
         combined_audio.export(file_name_mp3, format="mp3")
         
@@ -1048,8 +1021,6 @@ async def process_podcast(client, chat_id, prompt):
         except: pass
 
         caption_file = f"🎧 فایل پادکست شما با فرمت MP3:\n\n💡 موضوع شما: {prompt}"
-        
-        # ارسال فقط به صورت فایل MP3
         upload_result_file = await helper_upload_file(client, chat_id, file_name_mp3, "File", caption_file)
         
         if upload_result_file is not True: 
@@ -1058,7 +1029,6 @@ async def process_podcast(client, chat_id, prompt):
         if os.path.exists(file_name_mp3): os.remove(file_name_mp3)
 
 
-# --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن ---
 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)
@@ -1072,7 +1042,6 @@ 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 = "لطفاً این فایل صوتی/تصویری را با دقت کامل گوش بده و صحبت‌های داخل آن را کلمه به کلمه به متن تبدیل کن. هیچ توضیح اضافه‌ای نده."
@@ -1080,7 +1049,7 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     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}}
+            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:
@@ -1097,7 +1066,6 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     except Exception: pass
 
     if transcribed_text:
-        # کسر اعتبار در دیتاست (اگر نامحدود نیست)
         if not creds.get("is_premium"):
             user_credits_db[str_chat_id]["stt"] -= 1
             save_db(user_credits_db)
@@ -1106,7 +1074,6 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
         await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", True)
 
 
-# --- ۵. پردازش تحلیل فایل مستقل ---
 async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
@@ -1120,14 +1087,13 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     mime_type, _ = mimetypes.guess_type(file_name)
     if not mime_type: mime_type = "image/jpeg"
     
-    # دریافت ۱۰۰ کلید به صورت چرخشی
     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}}
+            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:
@@ -1144,7 +1110,6 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     except Exception: pass
 
     if final_answer: 
-        # کسر اعتبار در دیتاست (اگر نامحدود نیست)
         if not creds.get("is_premium"):
             user_credits_db[str_chat_id]["file"] -= 1
             save_db(user_credits_db)
@@ -1153,12 +1118,10 @@ 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)
 
@@ -1182,7 +1145,6 @@ async def process_create_file(client, chat_id, topic):
                         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)
@@ -1220,7 +1182,6 @@ async def process_create_file(client, chat_id, topic):
     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)
@@ -1230,7 +1191,6 @@ async def process_create_file(client, chat_id, topic):
                     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)
@@ -1250,7 +1210,6 @@ async def process_create_file(client, chat_id, topic):
     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)
@@ -1282,7 +1241,6 @@ async def process_create_file(client, chat_id, topic):
         await send_with_keyboard(client, chat_id, "❌ فایل‌ها ساخته شدند اما روبیکا در ارسال آن‌ها دچار مشکل شد.", True)
 
 
-# --- تنظیمات ربات روبیکا ---
 if not bot_token:
     print("خطا: توکن ربات روبیکا وارد نشده است!")
 else:
@@ -1292,13 +1250,10 @@ else:
     async def main_handler(client, update):
         global BOT_GUID
         
-        # 🔥 سیستم ضد رگبار و تخلیه صف (Backlog) سرور 🔥
-        # در صورت هجوم پیام‌های تلنبار شده سرور روبیکا، پیام‌ها رد می‌شود
         if is_backlog_burst():
             return
             
         try:
-            # 🛡 دریافت آیدی یکتای ربات برای جلوگیری از لوپ 🛡
             if not BOT_GUID:
                 try:
                     me_info = await client.get_me()
@@ -1309,25 +1264,20 @@ else:
 
             msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
             
-            # --- استخراج دقیق شناسه فرستنده (برای جلوگیری از حرف زدن ربات با خودش) ---
             author_id = getattr(update, 'author_guid', None)
             if not author_id and msg_obj:
                 author_id = msg_obj.get('author_object_guid') if isinstance(msg_obj, dict) else getattr(msg_obj, 'author_object_guid', None)
             
-            # ⛔️ حیاتی‌ترین بخش: اگر نویسنده پیام خودِ ربات است، هرگز پردازش نشود!
             if BOT_GUID and author_id == BOT_GUID:
                 return
             
-            # --- استخراج دقیق شناسه چت ---
             chat_id = getattr(update, 'object_guid', None) or getattr(update, 'author_guid', None) or getattr(update, "chat_id", None)
             if not chat_id: return
             
-            # استخراج آیدی پیام
             msg_id = getattr(update, "message_id", None)
             if not msg_id and msg_obj:
                 msg_id = msg_obj.get("message_id") if isinstance(msg_obj, dict) else getattr(msg_obj, "message_id", None)
             
-            # 🛡 کلید ترکیبی ضد تداخل: آیدی چت + آیدی پیام 🛡
             unique_msg_key = f"{chat_id}_{msg_id}" if msg_id else None
             if unique_msg_key:
                 if unique_msg_key in processed_message_ids: return
@@ -1336,21 +1286,22 @@ else:
             
             current_time = time.time()
             
-            # 🛡 سیستم محدودکننده سرعت کاربر (Rate Limit) 🛡
             last_req_time = user_last_request_time.get(chat_id, 0)
             if current_time - last_req_time < 1.0:
                 return
             user_last_request_time[chat_id] = current_time
 
-            # تشخیص محتوای متنی
             user_text = getattr(update, "text", "") or getattr(msg_obj, "text", "")
             user_text_str = str(user_text).strip() if user_text else ""
             user_text_lower = user_text_str.lower()
 
-            if chat_id not in user_states:
-                user_states[chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None}
+            str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
+
+            if str_chat_id not in user_states:
+                user_states[str_chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None}
+
+            creds = get_user_credits(str_chat_id)
 
-            # 🛠 --- سیستم پنل مدیریت (بدون نیاز به لاگین) --- 🛠
             if user_text_lower.startswith(f"{ADMIN_CODE} pro=") or user_text_lower.startswith(f"{ADMIN_CODE}pro="):
                 parts = user_text_str.split("=", 1)
                 if len(parts) >= 2:
@@ -1368,7 +1319,10 @@ else:
                                 "tts": 5,
                                 "file": 1,
                                 "stt": 5,
-                                "has_joined": True
+                                "has_joined": True,
+                                "invited_count": 0,
+                                "used_referral": False,
+                                "referral_code": ""
                             }
                         
                         user_credits_db[target_id]["is_premium"] = True
@@ -1437,10 +1391,12 @@ else:
                             stt_rem = "نامحدود ∞" if is_prem else t_creds.get('stt', 0)
                             image_rem = t_creds.get('image', 0)
                             edit_image_rem = t_creds.get('edit_image', 0)
+                            invs = t_creds.get('invited_count', 0)
 
                             info_msg = f"""🔍 **اطلاعات کاربر `{target_id}`:**
 
 🔹 **وضعیت:** {status_text}
+🎁 **تعداد افراد دعوت شده:** {invs} نفر
 
 📊 **سهمیه:**
 - چت: {chat_rem}
@@ -1455,7 +1411,6 @@ else:
                             await send_with_keyboard(client, chat_id, f"❌ کاربری با شناسه `{target_id}` در دیتابیس یافت نشد.", False)
                 return
 
-            # تشخیص فایل
             is_file = False
             file_name = f"unknown_file_{uuid.uuid4().hex[:6]}.jpg"
             
@@ -1476,13 +1431,8 @@ else:
                             file_name = file_attr.get('file_name', file_name) if isinstance(file_attr, dict) else file_name
                             break
 
-            # ⛔️ سیستم قفل اجباری کانال (Force Join) - حالت اعتمادی (یک بار کلیک) ⛔️
-            str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
-            creds = get_user_credits(str_chat_id)
-
             if user_text_str == "✅ عضو شدم":
                 if not creds.get("has_joined", False):
-                    # با همان کلیک اول تایید می‌شود
                     user_credits_db[str_chat_id]["has_joined"] = True
                     save_db(user_credits_db)
                     await send_with_keyboard(client, chat_id, "🎉 **عضویت شما با موفقیت تایید شد! خیلی خوش آمدید.**\n\nحالا می‌توانید از تمامی امکانات ربات استفاده کنید.\nلطفاً یکی از گزینه‌های منو را انتخاب کنید:", True)
@@ -1500,15 +1450,19 @@ else:
                     await client.send_message(chat_id, join_msg)
                 return
 
-            if user_text_str in["/start", "سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
-                user_states[chat_id]["mode"] = None
-                user_states[chat_id]["file_bytes"] = None
+            if user_text_lower.startswith("/start"):
+                user_states[str_chat_id]["mode"] = None
+                user_states[str_chat_id]["file_bytes"] = None
+                await send_with_keyboard(client, chat_id, "سلام! به ربات هوش مصنوعی آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:", True)
+                return
+
+            if user_text_str in["سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
+                user_states[str_chat_id]["mode"] = None
+                user_states[str_chat_id]["file_bytes"] = None
                 await send_with_keyboard(client, chat_id, "سلام! به ربات هوش مصنوعی آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:", True)
                 return
 
-            # --- حساب کاربری ---
             if user_text_str in["/account", "حساب کاربری 👤"]:
-                creds = get_user_credits(chat_id)
                 is_prem = creds.get("is_premium", False)
                 
                 if is_prem:
@@ -1538,11 +1492,15 @@ else:
                 stt_rem = "نامحدود ∞" if is_prem else creds['stt']
                 image_rem = creds['image']
                 edit_image_rem = creds['edit_image']
+                invited_count = creds.get('invited_count', 0)
+                my_code = get_or_create_referral_code(str_chat_id)
 
                 account_profile = f"""👤 **اطلاعات حساب کاربری شما**
 
 🔹 **شناسه یکتا:** `{chat_id}`
+🎫 **کد هدیه شما:** `{my_code}`
 🔹 **وضعیت اشتراک:** {status_text}{expire_info}
+🎁 **تعداد افراد دعوت شده:** {invited_count} نفر
 
 📊 **سهمیه باقی‌مانده شما:**
 - 💬 چت هوشمند: {chat_rem}
@@ -1557,7 +1515,58 @@ else:
                 await send_with_keyboard(client, chat_id, account_profile, True)
                 return
 
-            # --- دکمه خرید اشتراک ---
+            # ===============================================
+            # 🎁 سیستم جدید دعوت دوستان (دو سر سود) 🎁
+            # ===============================================
+            if user_text_str in["/invite", "دعوت دوستان 🎁"]:
+                invites = creds.get("invited_count", 0)
+                remains = 10 - (invites % 10)
+                my_code = get_or_create_referral_code(str_chat_id)
+                
+                invite_text = f"""🎁 **سیستم دعوت دوستان (دو سر سود)**
+
+با دعوت دوستان خود به ربات آلفا، هم شما و هم دوستتان هدیه می‌گیرید!
+به دوست خود بگویید پس از ورو�� به ربات، دکمه **«ثبت کد هدیه 🎫»** را بزند و کد زیر را وارد کند.
+✨ **سود دوست شما:** در همان لحظه 10 تبدیل رایگان متن به صدا دریافت می‌کند.
+✨ **سود شما:** به آمار دعوت‌هایتان اضافه می‌شود و به ازای هر **10 نفر**، **3 روز اشتراک پرو نامحدود** می‌گیرید.
+
+📊 **آمار شما:**
+- تعداد دعوت‌های موفق: {invites} نفر
+- دعوت‌های باقی‌مانده تا جایزه بعدی: {remains} نفر
+
+کد هدیه اختصاصی شما:
+`{my_code}`
+
+(متن زیر را کپی کرده و برای دوستانتان بفرستید 👇)"""
+                await send_with_keyboard(client, chat_id, invite_text, True)
+                
+                forward_text = f"""🤖 **ربات هوش مصنوعی آلفا پرو**
+
+✨ با این ربات می‌تونی کارهای زیر رو به راحتی انجام بدی:
+💬 چت با پیشرفته‌ترین هوش مصنوعی
+🎨 ساخت و ویرایش حرفه‌ای عکس
+🎙 ساخت پادکست اختصاصی
+🗣 تبدیل متن به صدا (30 گوینده مختلف)
+📝 تبدیل صدا و ویدیو به متن
+📁 تحلیل فایل‌ها و ساخت مقاله
+
+👇 اول وارد ربات زیر شو:
+@aialphabot
+
+سپس دکمه **«ثبت کد هدیه 🎫»** را بزن و کد 8 رقمی زیر رو وارد کن تا همون اول **10 تا تبدیل صدا هدیه بگیری**:
+`{my_code}`"""
+                await send_with_keyboard(client, chat_id, forward_text, False)
+                return
+
+            if user_text_str in["/referral", "ثبت کد هدیه 🎫"]:
+                if creds.get("used_referral", False):
+                    await send_with_keyboard(client, chat_id, "❌ شما قبلاً کد هدیه یک نفر را ثبت کرده‌اید و فقط یک‌بار مجاز به استفاده از این امکان هستید.", True)
+                    return
+                user_states[str_chat_id]["mode"] = "waiting_for_referral_code"
+                msg = "🎫 **ثبت کد هدیه**\n\nکد هدیه 8 رقمی (اعداد) که از دوست خود دریافت کرده‌اید را اینجا وارد کنید تا در همان لحظه **10 سهمیه تبدیل رایگان متن به صدا** هدیه بگیرید!\n\n(برای انصراف دکمه «برگشت♻️» را بزنید)"
+                await send_with_keyboard(client, chat_id, msg, True)
+                return
+
             if user_text_str in["/buy", "خرید اشتراک 💎"]:
                 buy_text = f"""💎 **خرید اشتراک ویژه آلفا پرو (یک ماهه)**
 
@@ -1589,7 +1598,6 @@ else:
                 await send_with_keyboard(client, chat_id, buy_text, True)
                 return
 
-            # --- دکمه انتقال اکانت ---
             if user_text_str in["/transfer", "انتقال اکانت از برنامه به ربات"]:
                 transfer_text = f"""🔄 **انتقال اکانت از برنامه به ربات**
 
@@ -1605,55 +1613,128 @@ else:
                 return
 
             if user_text_str in["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
-                user_states[chat_id]["mode"] = "chat"
-                user_states[chat_id]["history"] =[]
+                user_states[str_chat_id]["mode"] = "chat"
+                user_states[str_chat_id]["history"] =[]
                 await send_with_keyboard(client, chat_id, "💬 شما وارد بخش **چت با هوش مصنوعی** شدید.\n\nهر سوالی دارید بفرستید تا جواب بدم:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/image", "🎨 عکس", "ساخت تصاویر🎨"]:
-                user_states[chat_id]["mode"] = "image_waiting_for_text"
+                user_states[str_chat_id]["mode"] = "image_waiting_for_text"
                 await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس پیشرفته** شدید.\n\nمتن خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/edit_image", "ویرایش تصاویر 🪄"]:
-                user_states[chat_id]["mode"] = "image_edit_waiting_for_image"
-                user_states[chat_id]["file_bytes"] = None
+                user_states[str_chat_id]["mode"] = "image_edit_waiting_for_image"
+                user_states[str_chat_id]["file_bytes"] = None
                 await send_with_keyboard(client, chat_id, "🪄 به بخش **ویرایش عکس (Flux.2)** خوش آمدید.\n\nلطفاً ابتدا عکسی که می‌خواهید ویرایش کنید را بفرستید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/tts", "🎙️ صدا", "تبدیل متن به صدا🗣️"]:
-                user_states[chat_id]["mode"] = "tts_waiting_for_text"
+                user_states[str_chat_id]["mode"] = "tts_waiting_for_text"
                 await send_with_keyboard(client, chat_id, "🎙️ شما وارد بخش **تبدیل متن به صدا** شدید.\n\nلطفاً متنی که می‌خواهید به صدا تبدیل شود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
             
             if user_text_str in["/podcast", "📻 پادکست", "ساخت پادکست 🎙️"]:
-                user_states[chat_id]["mode"] = "podcast_waiting_for_topic"
+                user_states[str_chat_id]["mode"] = "podcast_waiting_for_topic"
                 await send_with_keyboard(client, chat_id, "📻 شما وارد بخش **ساخت پادکست** شدید.\n\nلطفاً موضوع پادکست خود را بفرستید.\nمثال: درباره تاریخچه پیدایش قهوه با ۳ گوینده یک پادکست جذاب بساز . همچنین این قسمت متصل به مدل زبانی است و درخواست هارو قبل از ساخت درک میکنه. میتوانید مقاله کامل یک سایت بفرستید با تبلیغات یا هرچی، هوش مصنوعی متن مقاله رو استخراج و پادکست براتون میسازه . در توضیحات امکان مشخص کردن تعداد گوینده به همراه اسم شون نیز از سمت شما امکان پذیر است.\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/file", "تحلیل فایل 📁"]:
-                user_states[chat_id]["mode"] = "file_waiting_for_file"
-                user_states[chat_id]["file_bytes"] = None
+                user_states[str_chat_id]["mode"] = "file_waiting_for_file"
+                user_states[str_chat_id]["file_bytes"] = None
                 await send_with_keyboard(client, chat_id, "📁 شما وارد بخش **تحلیل فایل اختصاصی** شدید.\n\nلطفاً فایل خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/stt", "فایل صوتی به متن 📝"]:
-                user_states[chat_id]["mode"] = "stt_waiting_for_audio"
+                user_states[str_chat_id]["mode"] = "stt_waiting_for_audio"
                 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"
+                user_states[str_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")
+            current_mode = user_states[str_chat_id].get("mode")
 
             if current_mode is None:
                 if is_file: pass 
                 elif user_text_str: await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا از کیبورد پایین، بخش مورد نظرتان را انتخاب کنید:", True)
                 return
 
+            # ===============================================
+            # بخش دریافت کد هدیه 8 رقمی از کاربر
+            # ===============================================
+            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:
+                        await send_with_keyboard(client, chat_id, "❌ کد وارد شده نامعتبر است. لطفاً یک کد 8 رقمی صحیح بفرستید.", False)
+                        return
+                    
+                    inviter_id = find_user_by_referral_code(normalized_code)
+                    
+                    if not inviter_id:
+                        await send_with_keyboard(client, chat_id, "❌ کد هدیه یافت نشد. لطفاً کد را با دقت بررسی کنید.", False)
+                        return
+                        
+                    if str(inviter_id) == str(str_chat_id):
+                        await send_with_keyboard(client, chat_id, "❌ شما نمی‌توانید کد خودتان را ثبت کنید!", False)
+                        return
+                    
+                    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"]
+                    
+                    if current_invites > 0 and current_invites % 10 == 0:
+                        inviter_data = user_credits_db[inviter_id]
+                        inviter_data["is_premium"] = True
+                        now = datetime.datetime.now()
+                        if inviter_data.get("expire_date"):
+                            try:
+                                current_exp = datetime.datetime.fromisoformat(inviter_data["expire_date"])
+                                if current_exp > now:
+                                    new_exp = current_exp + datetime.timedelta(days=3)
+                                else:
+                                    new_exp = now + datetime.timedelta(days=3)
+                            except:
+                                new_exp = now + datetime.timedelta(days=3)
+                        else:
+                            new_exp = now + datetime.timedelta(days=3)
+                            
+                        inviter_data["expire_date"] = new_exp.isoformat()
+                        inviter_data["chat"] = 999999
+                        inviter_data["image"] = 20
+                        inviter_data["edit_image"] = 10
+                        inviter_data["podcast"] = 999999
+                        inviter_data["tts"] = 999999
+                        inviter_data["file"] = 999999
+                        inviter_data["stt"] = 999999
+                        
+                        save_db(user_credits_db)
+                        try:
+                            msg_text = f"🎉 **تبریک ویژه!**\nیک دوست کد هدیه شما را ثبت کرد.\nتعداد کل دعوت‌های شما: {current_invites} نفر\n\n🎁 **هدیه شما:** ۳ روز اشتراک پرو نامحدود به حساب شما افزوده شد!"
+                            asyncio.create_task(send_with_keyboard(client, inviter_id, msg_text, True))
+                        except: pass
+                    else:
+                        save_db(user_credits_db)
+                        remains = 10 - (current_invites % 10)
+                        try:
+                            msg_text = f"🎉 **تبریک!**\nیک دوست کد هدیه شما را ثبت کرد.\nتعداد کل دعوت‌های شما: {current_invites} نفر\n\n⏳ با دعوت {remains} نفر دیگر، ۳ روز اشتراک رایگان دریافت می‌کنید!"
+                            asyncio.create_task(send_with_keyboard(client, inviter_id, msg_text, True))
+                        except: pass
+
+                    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)
@@ -1673,17 +1754,17 @@ else:
                 await send_with_keyboard(client, chat_id, "📥 در حال دانلود عکس...", False)
                 try:
                     file_bytes = await helper_download_file(client, msg_obj)
-                    user_states[chat_id]["file_bytes"] = file_bytes
-                    user_states[chat_id]["mode"] = "image_edit_waiting_for_prompt"
+                    user_states[str_chat_id]["file_bytes"] = file_bytes
+                    user_states[str_chat_id]["mode"] = "image_edit_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)
                 return
 
             elif current_mode == "image_edit_waiting_for_prompt":
                 if user_text_str:
-                    saved_bytes = user_states[chat_id].get("file_bytes")
-                    user_states[chat_id]["mode"] = None 
-                    user_states[chat_id]["file_bytes"] = None
+                    saved_bytes = user_states[str_chat_id].get("file_bytes")
+                    user_states[str_chat_id]["mode"] = None 
+                    user_states[str_chat_id]["file_bytes"] = None
                     asyncio.create_task(process_image_edit(client, chat_id, saved_bytes, user_text_str))
                 else: await send_with_keyboard(client, chat_id, "⚠️ لطفاً درخواست خود را متنی بنویسید.", False)
                 return
@@ -1691,8 +1772,8 @@ else:
             elif current_mode == "tts_waiting_for_text":
                 if user_text_str:
                     if len(user_text_str) > 2500: return await send_with_keyboard(client, chat_id, "⚠️ لطفاً متنی کوتاه‌تر از 2500 کاراکتر بفرستید.", False)
-                    user_states[chat_id]["text"] = user_text_str
-                    user_states[chat_id]["mode"] = "tts_waiting_for_speaker"
+                    user_states[str_chat_id]["text"] = user_text_str
+                    user_states[str_chat_id]["mode"] = "tts_waiting_for_speaker"
                     
                     speakers_menu = """✅ متن شما ذخیره شد.
 لطفاً **شماره** گوینده مورد نظر خود را بفرستید:
@@ -1714,8 +1795,8 @@ else:
                 
                 if normalized_text.isdigit() and normalized_text in SPEAKERS:
                     spk_name, spk_id = SPEAKERS[normalized_text]
-                    txt = user_states[chat_id]["text"]
-                    user_states[chat_id]["mode"] = "tts_waiting_for_text"
+                    txt = user_states[str_chat_id]["text"]
+                    user_states[str_chat_id]["mode"] = "tts_waiting_for_text"
                     asyncio.create_task(process_tts(client, chat_id, txt, spk_id, spk_name))
                 else: 
                     await send_with_keyboard(client, chat_id, "❌ شماره نامعتبر است! لطفاً فقط یک عدد بین 1 تا 30 بفرستید.", False)
@@ -1740,18 +1821,18 @@ else:
                 await send_with_keyboard(client, chat_id, "📥 در حال دریافت فایل...", False)
                 try:
                     file_bytes = await helper_download_file(client, msg_obj)
-                    user_states[chat_id]["file_bytes"] = file_bytes
-                    user_states[chat_id]["file_name"] = file_name 
-                    user_states[chat_id]["mode"] = "file_waiting_for_prompt"
+                    user_states[str_chat_id]["file_bytes"] = file_bytes
+                    user_states[str_chat_id]["file_name"] = file_name 
+                    user_states[str_chat_id]["mode"] = "file_waiting_for_prompt"
                     await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً متنی بگویید **چگونه تحلیل شود؟**", False)
                 except Exception as dl_err: await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n{str(dl_err)}", False)
                 return
 
             elif current_mode == "file_waiting_for_prompt":
                 if user_text_str:
-                    saved_bytes = user_states[chat_id].get("file_bytes")
-                    saved_name = user_states[chat_id].get("file_name", "file.jpeg")
-                    user_states[chat_id]["mode"] = "file_waiting_for_file" 
+                    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"] = "file_waiting_for_file" 
                     asyncio.create_task(process_file_analysis(client, chat_id, saved_bytes, saved_name, user_text_str))
                 else: await send_with_keyboard(client, chat_id, "⚠️ لطفاً درخواست خود را متنی بنویسید.", False)
                 return
@@ -1768,5 +1849,5 @@ else:
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
     if bot_token:
-        print("ربات آلفا پرو با سیستم اشتراک نامحدود + سپر امنیتی Burst Controller روشن شد...")
+        print("ربات آلفا پرو با سیستم اشتراک نامحدود + کد هدیه یکتا + سپر امنیتی روشن شد...")
         bot.run()