Update main.py

main.py

@@ -58,13 +58,14 @@ ADMIN_CODE = "3011"
 BOT_GUID = None
 
 # =======================================================
-# سیستم ضد اسپم کاربر-محور
+# سیستم ضد اسپم کاربر-محور (بهینه شده برای جلوگیری از نشت مموری)
 # =======================================================
 user_message_times = {}
 
 def is_user_spamming(chat_id):
     now = time.time()
-    times = user_message_times.get(chat_id,[])
+    times = user_message_times.get(chat_id, [])
+    # فقط زمان‌های ۳ ثانیه اخیر رو نگه دار
     times =[t for t in times if now - t < 3.0]
     times.append(now)
     user_message_times[chat_id] = times
@@ -92,30 +93,35 @@ def gregorian_to_jalali(gy, gm, gd):
     return jy, jm, jd
 
 # ==============================================================================
-# 🟢 پارت 4: دیتابیس SQLite (نسخه نجات همه‌جانبه: اعتبارها + پیام‌های قدیمی)
+# 🟢 پارت 4: دیتابیس SQLite (نسخه فوق‌سریع Non-Blocking با سیستم Threading)
 # ==============================================================================
 import os
 import sqlite3
 import json
 import copy
+import threading
 
-# استفاده از نام جدید v3 برای شروع کاملاً پاک و بدون قفل
 DB_FILE = "/data/users_v3.db" 
-CORRUPT_FILE = "/data/users_db.db" # فایل ۱۱.۷ مگابایتی شما
-BACKUP_JSON = "/data/users_db.json.bak" # فایل ۴.۷ مگابایتی اعتبارها
+CORRUPT_FILE = "/data/users_db.db" 
+BACKUP_JSON = "/data/users_db.json.bak" 
 
 last_saved_state = {}
+# حافظه کش برای پیام‌های پردازش شده جهت جلوگیری از قفل شدن دیتابیس
+recent_messages_dict = {}
+# قفل‌ها برای جلوگیری از تداخل رشته‌ها (Threads)
+db_lock = threading.Lock()
+msg_cache_lock = threading.Lock()
 
 def init_sqlite_db():
     try:
         os.makedirs(os.path.dirname(DB_FILE), exist_ok=True)
-        
-        # اگر فایل v3 وجود ندارد، یعنی بار اول است و باید نجات را شروع کنیم
         is_first_run = not os.path.exists(DB_FILE)
 
         conn = sqlite3.connect(DB_FILE, timeout=60.0)
         cursor = conn.cursor()
-        cursor.execute('PRAGMA journal_mode=DELETE;') # امن‌ترین حالت برای شبکه
+        # 🚀 تغییر بسیار مهم: استفاده از WAL به جای DELETE برای پشتیبانی از هزاران کاربر همزمان
+        cursor.execute('PRAGMA journal_mode=WAL;') 
+        cursor.execute('PRAGMA synchronous=NORMAL;')
         cursor.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
         cursor.execute('CREATE TABLE IF NOT EXISTS processed_messages (message_id TEXT PRIMARY KEY)')
         conn.commit()
@@ -123,9 +129,7 @@ def init_sqlite_db():
 
         if is_first_run:
             print("🚨 عملیات نجات آغاز شد...")
-            # ۱. بازیابی اعتبارها از JSON (۱۰۰ درصد مطمئن)
             restore_users_from_json()
-            # ۲. نجات شناسه‌های پیام از فایل ۱۱.۷ مگابایتی (تا جایی که راه بدهد)
             salvage_messages_from_corrupt_db()
 
     except Exception as e:
@@ -139,6 +143,7 @@ def restore_users_from_json():
                 data = json.load(f)
             if data:
                 conn = sqlite3.connect(DB_FILE, timeout=60.0)
+                conn.execute('PRAGMA journal_mode=WAL;')
                 for cid, udata in data.items():
                     conn.execute("INSERT OR REPLACE INTO users VALUES (?, ?)", (str(cid), json.dumps(udata)))
                 conn.commit()
@@ -151,75 +156,102 @@ def salvage_messages_from_corrupt_db():
     if os.path.exists(CORRUPT_FILE):
         print(f"🔍 در حال تلاش برای نجات شناسه‌های پیام از فایل {CORRUPT_FILE} ...")
         try:
-            # وصل کردن فایل خراب به دیتابیس جدید
             conn = sqlite3.connect(DB_FILE, timeout=60.0)
+            conn.execute('PRAGMA journal_mode=WAL;')
             conn.execute(f"ATTACH DATABASE '{CORRUPT_FILE}' AS old_db")
-            
-            # کپی کردن شناسه‌ها (با استفاده از INSERT OR IGNORE که اگر خراب بود رد شود)
-            # این دستور تلاش می‌کند حتی اگر فایل Malformed باشد، ردیف‌های سالم را بردارد
             conn.execute("INSERT OR IGNORE INTO processed_messages SELECT message_id FROM old_db.processed_messages")
-            
             conn.commit()
             conn.close()
-            print("✅ شناسه‌های پیام قدیمی با موفقیت نجات یافتند و به دیتابیس جدید منتقل شدند.")
+            print("✅ شناسه‌های پیام قدیمی نجات یافتند.")
         except Exception as e:
-            print(f"⚠��� هشدار: برخی پیام‌ها به دلیل خرابی شدید فایل اصلی قابل نجات نبودند، اما نگران نباشید، ربات دوباره آن‌ها را شناسایی می‌کند. ارور: {e}")
+            print(f"⚠️ هشدار فایل قدیمی: {e}")
 
 def load_db():
-    global last_saved_state
+    global last_saved_state, recent_messages_dict
     init_sqlite_db()
     db_dict = {}
     try:
         conn = sqlite3.connect(DB_FILE, timeout=60.0)
+        conn.execute('PRAGMA journal_mode=WAL;')
         cursor = conn.cursor()
+        
+        # لود کردن کاربران
         cursor.execute("SELECT chat_id, user_data FROM users")
         for row in cursor.fetchall():
             db_dict[row[0]] = json.loads(row[1])
+            
+        # 🚀 لود کردن 15 هزار پیام آخر به داخل RAM برای سرعت فوق العاده
+        cursor.execute("SELECT message_id FROM processed_messages ORDER BY ROWID DESC LIMIT 15000")
+        for row in cursor.fetchall():
+            recent_messages_dict[row[0]] = True
+            
         conn.close()
         last_saved_state = copy.deepcopy(db_dict)
-        print(f"🚀 دیتابیس نهایی آماده شد. تعداد کاربران: {len(db_dict)}")
+        print(f"🚀 دیتابیس آماده شد. تعداد کاربران: {len(db_dict)} | پیام‌های کش شده در رم: {len(recent_messages_dict)}")
         return db_dict
     except Exception as e:
         print(f"⚠️ ارور در لود نهایی: {e}")
         return {}
 
+# --- سیستم ذخیره سازی در پس‌زمینه (بدون معطل کردن ربات) ---
+def background_save_worker(changed_data):
+    with db_lock:
+        try:
+            conn = sqlite3.connect(DB_FILE, timeout=60.0)
+            conn.execute('PRAGMA journal_mode=WAL;')
+            conn.execute('PRAGMA synchronous=NORMAL;')
+            conn.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed_data)
+            conn.commit()
+            conn.close()
+        except Exception as e:
+            print(f"❌ خطا در ذخیره بک‌گراند: {e}")
+
 def save_db(db_data):
     global last_saved_state
-    try:
-        changed = []
-        for cid, data in db_data.items():
-            if cid not in last_saved_state or last_saved_state[cid] != data:
-                changed.append((str(cid), json.dumps(data, ensure_ascii=False)))
-        if not changed: return
-        conn = sqlite3.connect(DB_FILE, timeout=60.0)
-        conn.execute('PRAGMA journal_mode=DELETE;')
-        conn.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed)
-        conn.commit()
-        conn.close()
-        for cid, _ in changed:
-            last_saved_state[cid] = copy.deepcopy(db_data[cid])
-    except Exception as e:
-        print(f"❌ خطا در ذخیره: {e}")
+    changed =[]
+    for cid, data in db_data.items():
+        if cid not in last_saved_state or last_saved_state[cid] != data:
+            changed.append((str(cid), json.dumps(data, ensure_ascii=False)))
+            
+    if not changed: return
+    
+    # آپدیت وضعیت استیت محلی
+    for cid, _ in changed:
+        last_saved_state[cid] = copy.deepcopy(db_data[cid])
+        
+    # 🚀 پرتاب عملیات سنگین به یک پردازشگر دیگر تا ربات متوقف نشود
+    threading.Thread(target=background_save_worker, args=(changed,), daemon=True).start()
 
+# --- سیستم بررسی پیام تکراری (کاملا منتقل شده به RAM سرور) ---
 def is_message_processed(message_id):
-    try:
-        conn = sqlite3.connect(DB_FILE, timeout=30.0)
-        res = conn.execute("SELECT 1 FROM processed_messages WHERE message_id = ?", (str(message_id),)).fetchone()
-        conn.close()
-        return res is not None
-    except: return False
+    # چک کردن داخل RAM (زمان اجرا: 0.0001 ثانیه)
+    return str(message_id) in recent_messages_dict
+
+def background_mark_message(msg_id):
+    with db_lock:
+        try:
+            conn = sqlite3.connect(DB_FILE, timeout=30.0)
+            conn.execute('PRAGMA journal_mode=WAL;')
+            conn.execute("INSERT OR IGNORE INTO processed_messages VALUES (?)", (msg_id,))
+            conn.commit()
+            conn.close()
+        except: pass
 
 def mark_message_processed(message_id):
-    try:
-        conn = sqlite3.connect(DB_FILE, timeout=30.0)
-        conn.execute("INSERT OR IGNORE INTO processed_messages VALUES (?)", (str(message_id),))
-        conn.commit()
-        conn.close()
-    except: pass
+    msg_id_str = str(message_id)
+    with msg_cache_lock:
+        # ثبت در RAM
+        recent_messages_dict[msg_id_str] = True
+        # جلوگیری از پر شدن بیش از حد RAM
+        if len(recent_messages_dict) > 15000:
+            oldest_key = next(iter(recent_messages_dict))
+            del recent_messages_dict[oldest_key]
+            
+    # ذخیره در دیتابیس به صورت پس‌زمینه
+    threading.Thread(target=background_mark_message, args=(msg_id_str,), daemon=True).start()
 
 user_credits_db = load_db()
 
-
 # ==============================================================================
 # 🟢 پارت 5: توابع کد معرف، مدیریت سهمیه حساب کاربری و تبدیل اعداد
 # ==============================================================================
@@ -252,8 +284,8 @@ def get_user_credits(chat_id):
             "expire_date": None,
             "last_reset": "",
             "chat": 10,
-            "image": 5,
-            "edit_image": 1,
+            "image": 0,
+            "edit_image": 0,
             "podcast": 2,
             "tts": 5,
             "file": 1,
@@ -273,6 +305,8 @@ def get_user_credits(chat_id):
     if "voice_conv" not in user_data: user_data["voice_conv"] = 3
     if "voice_clone" not in user_data: user_data["voice_clone"] = 1
     if "last_msg_id" not in user_data: user_data["last_msg_id"] = 0
+    if "image" not in user_data: user_data["image"] = 0
+    if "edit_image" not in user_data: user_data["edit_image"] = 0
 
     is_premium = user_data.get("is_premium", False)
     
@@ -283,21 +317,34 @@ def get_user_credits(chat_id):
                 user_data["is_premium"] = False
                 user_data["expire_date"] = None
                 is_premium = False
+                user_data["image"] = 0
+                user_data["edit_image"] = 0
+                save_db(user_credits_db) 
         except Exception:
             pass
 
     if not is_premium:
+        changed = False
+        # صفر کردن سهمیه عکس کاربرانی که از دیروز اعتبار رایگان جا مانده دارند
+        if user_data.get("image", 0) > 0 or user_data.get("edit_image", 0) > 0:
+            user_data["image"] = 0
+            user_data["edit_image"] = 0
+            changed = True
+
         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
+            user_data["image"] = 0
+            user_data["edit_image"] = 0
             user_data["podcast"] = 2
             user_data["tts"] = 5
             user_data["file"] = 1
             user_data["stt"] = 5
             user_data["voice_conv"] = 3
             user_data["voice_clone"] = 1
+            changed = True
+            
+        if changed:
             save_db(user_credits_db)
             
     return user_data
@@ -1125,8 +1172,28 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
 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)
+    
+    is_prem = creds.get("is_premium", False)
+    
+    # === بررسی کاربران رایگان ===
+    if not is_prem:
+        msg = """کاربر گرامی،
+
+ضمن تشکر از توجه شما، به اطلاع می‌رساند که قابلیت ساخت تصاویر در حال حاضر به صورت رایگان در دسترس نمی‌باشد. این سرویس به دلیل نیاز به منابع پردازشی بالا، جزو خدمات ویژه و نیازمند تهیه اشتراک است.
+
+با این حال، شما می‌توانید از تست رایگان و امکانات سایر بخش‌های پلتفرم ما بهره‌مند شوید.
+
+ما همواره در تلاش برای افزایش منابع و اضافه کردن خدمات رایگان بیشتر به کاربران عزیز هستیم.
+
+در حال حاضر، می‌توانید از سایر بخش‌های رایگان موجود استفاده کنید و یا برای دسترسی کامل و نامحدود به تمامی قابلیت‌ها (از جمله ساخت تصاویر)، اشتراک تهیه نمایید.
+
+از صبوری و همراهی شما سپاسگزاریم."""
+        return await send_with_keyboard(client, chat_id, msg, False)
+
+    # === بررسی اعتبار کاربران پولی ===
     if creds["image"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت عکس شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
+        msg = "⚠️ **اعتبار شما برای این بخش تمام شده است!**\nاز بقیه بخش‌ها می‌توانید نامحدود استفاده کنید. در حال افزایش منابع برای این بخش هستیم."
+        return await send_with_keyboard(client, chat_id, msg, False)
 
     if not HF_TOKENS: return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
     
@@ -1192,7 +1259,6 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
     proc_msg = await send_with_keyboard(client, chat_id, f"🎨 در حال طراحی عکس...\n\n📏 **ابعاد:** {size_name}\n📝 **پرامپت ساخته شده:**\n`{short_preview}`\n\n(ممکن است چند ثانیه زمان ببرد)", False)
 
     generated_image = None
-    last_error_log = "هیچ اتصالی برقرار نشد."
     
     for attempt in range(5):
         keys_to_try = HF_TOKENS.copy()
@@ -1208,7 +1274,6 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
                 )
                 break 
             except Exception as e:
-                last_error_log = str(e)
                 continue
         if generated_image: break
         await asyncio.sleep(2)
@@ -1221,7 +1286,9 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
     except Exception: pass
 
     if not generated_image: 
-        return await send_with_keyboard(client, chat_id, f"❌ عکس ساخته نشد.\n\n⚠️ خطا:\n{last_error_log[:200]}", True)
+        # === پیام اختصاصی جایگزین ارور اسپیس ===
+        error_msg = "⚠️ **منابع این بخش در حال افزایش است!**\nموقتاً این بخش در حال ارتقا است. فعلاً اعتباری از شما کم نشد، به‌زودی این بخش در دسترس قرار خواهد گرفت."
+        return await send_with_keyboard(client, chat_id, error_msg, True)
 
     try:
         file_name = f"image_{uuid.uuid4().hex}.jpg"
@@ -1230,27 +1297,26 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
         caption_text = f"🎨 تصویر شما با پرامپت هوشمند آماده شد!\n\n📏 ابعاد تصویر: {size_name}\n✨ ایده اولیه: {prompt}"
         
         upload_result = False
-        error_log_img = ""
         for up_att in range(3):
             res = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
             if res is True:
                 upload_result = True
                 break
             else:
-                error_log_img = res
                 await asyncio.sleep(4)
         
         if upload_result is True:
-            if not creds.get("is_premium"):
-                user_credits_db[str_chat_id]["image"] -= 1
-                save_db(user_credits_db)
+            # کسر اعتبار از کاربران پولی
+            user_credits_db[str_chat_id]["image"] -= 1
+            save_db(user_credits_db)
         else:
-            await send_with_keyboard(client, chat_id, f"❌ عکس ساخته شد اما آپلود در روبیکا با خطا مواجه شد:\n`{str(error_log_img)[:800]}`", True)
+            await send_with_keyboard(client, chat_id, "❌ عکس ساخته شد اما آپلود در روبیکا با خطا مواجه شد.", True)
             
         if os.path.exists(file_name): os.remove(file_name)
     except Exception as e: 
         await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره و ارسال عکس:\n{str(e)[:150]}", True)
 
+
 # ==============================================================================
 # 🟢 پارت 15: توابع ترجمه و ویرایش عکس با هوش مصنوعی (Flux.2)
 # ==============================================================================
@@ -1290,8 +1356,28 @@ async def translate_text_aloha(prompt_text):
 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)
+    
+    is_prem = creds.get("is_premium", False)
+    
+    # === بررسی کاربران رایگان (جهت اطمینان در لایه دوم) ===
+    if not is_prem:
+        msg = """کاربر گرامی،
+
+ضمن تشکر از توجه شما، به اطلاع می‌رساند که قابلیت ویرایش تصاویر در حال حاضر به صورت رایگان در دسترس نمی‌باشد. این سرویس به دلیل نیاز به منابع پردازشی بالا، جزو خدمات ویژه و نیازمند تهیه اشتراک است.
+
+با این حال، شما می‌توانید از تست رایگان و امکانات سایر بخش‌های پلتفرم ما بهره‌مند شوید.
+
+ما همواره در تلاش برای افزایش منابع و اضافه کردن خدمات رایگان بیشتر به کاربران عزیز هستیم.
+
+در حال حاضر، می‌توانید از سایر بخش‌های رایگان موجود استفاده کنید و یا برای دسترسی کامل و نامحدود به تمامی قابلیت‌ها (از جمله ویرایش تصاویر)، اشتراک تهیه نمایید.
+
+از صبوری و همراهی شما سپاسگزاریم."""
+        return await send_with_keyboard(client, chat_id, msg, False)
+
+    # === بررسی اعتبار کاربران پولی ===
     if creds["edit_image"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار ویرایش عکس شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
+        msg = "⚠️ **اعتبار شما برای این بخش تمام شده است!**\nاز بقیه بخش‌ها می‌توانید نامحدود استفاده کنید. در حال افزایش منابع برای این بخش هستیم."
+        return await send_with_keyboard(client, chat_id, msg, False)
 
     if not HF_TOKENS:
         return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
@@ -1303,7 +1389,6 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
         translated_prompt = prompt
 
     generated_image = None
-    last_error_log = "سرور پاسخ نداد."
     
     for attempt in range(5):
         keys_to_try = HF_TOKENS.copy()
@@ -1318,7 +1403,6 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
                 )
                 break 
             except Exception as e:
-                last_error_log = str(e)
                 continue
         if generated_image: break
         await asyncio.sleep(2)
@@ -1331,7 +1415,9 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     except Exception: pass
 
     if not generated_image:
-        return await send_with_keyboard(client, chat_id, f"❌ متأسفانه ویرایش عکس انجام نشد.\n\n⚠️ علت:\n{last_error_log[:200]}", True)
+        # === پیام اختصاصی جایگزین ارور اسپیس ===
+        error_msg = "⚠️ **منابع این بخش در حال افزایش است!**\nموقتاً این بخش در حال ارتقا است. فعلاً اعتباری از شما کم نشد، به‌زودی این بخش در دسترس قرار خواهد گرفت."
+        return await send_with_keyboard(client, chat_id, error_msg, True)
 
     try:
         file_name = f"edited_flux_{uuid.uuid4().hex}.jpg"
@@ -1340,22 +1426,20 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
         caption_text = f"🪄 ویرایش عکس با هوش مصنوعی Flux.2 انجام شد!\n\n✨ تغییرات خواسته شده: {prompt}\n🔤 متن ارسال شده به هوش: {translated_prompt}"
         
         upload_result = False
-        error_log_edit = ""
         for up_att in range(3):
             res = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
             if res is True:
                 upload_result = True
                 break
             else:
-                error_log_edit = res
                 await asyncio.sleep(4)
         
         if upload_result is True:
-            if not creds.get("is_premium"):
-                user_credits_db[str_chat_id]["edit_image"] -= 1
-                save_db(user_credits_db)
+            # کسر اعتبار از کاربران پولی
+            user_credits_db[str_chat_id]["edit_image"] -= 1
+            save_db(user_credits_db)
         else:
-            await send_with_keyboard(client, chat_id, f"❌ عکس ساخته شد اما خطا در ارسال به روبیکا رخ داد:\n`{str(error_log_edit)[:800]}`", True)
+            await send_with_keyboard(client, chat_id, "❌ عکس ساخته شد اما خطا در ارسال به روبیکا رخ داد.", True)
             
         if os.path.exists(file_name): os.remove(file_name)
     except Exception as e:
@@ -1886,17 +1970,25 @@ async def ram_garbage_collector():
         await asyncio.sleep(1800)  # هر ۳۰ دقیقه یکبار بررسی می‌کند
         now = time.time()
         try:
+            # پاکسازی یوزراستیت‌ها
             for uid in list(user_states.keys()):
-                # اگر کاربر بیش از ۱ ساعت (۳۶۰۰ ثانیه) با ربات کار نکرده بود
                 if now - user_states[uid].get("last_time", 0) > 3600:
-                    # پاک کردن بایت‌های فایل‌های سنگین از رم سرور
                     if user_states[uid].get("file_bytes") is not None:
                         user_states[uid]["file_bytes"] = None
                     if user_states[uid].get("ref_bytes") is not None:
                         user_states[uid]["ref_bytes"] = None
-                    # خالی کردن تاریخچه چت برای آزادسازی رم (نگه داشتن ۵ پیام آخر)
                     if len(user_states[uid].get("history",[])) > 5:
                         user_states[uid]["history"] = user_states[uid]["history"][-5:]
+            
+            # پاکسازی کش زمان پیام‌ها (ضد اسپم) که استفاده نمی‌شوند
+            for uid in list(user_message_times.keys()):
+                times = user_message_times[uid]
+                valid_times =[t for t in times if now - t < 3.0]
+                if not valid_times:
+                    del user_message_times[uid]
+                else:
+                    user_message_times[uid] = valid_times
+                    
         except Exception as e:
             pass
 
@@ -1942,10 +2034,8 @@ else:
             # ===================================================================
             if msg_id:
                 str_msg_id = str(msg_id).strip()
-                # 1. اگر پیام قبلاً در دیتابیس ثبت شده بود، ربات بی‌درنگ کار را متوقف می‌کند
                 if is_message_processed(str_msg_id):
                     return
-                # 2. اگر پیام جدید بود، در همان لحظه در دیتابیس ذخیره می‌شود تا دیگر هرگز خوانده نشود
                 mark_message_processed(str_msg_id)
             # ===================================================================
 
@@ -1956,13 +2046,10 @@ else:
             if is_user_spamming(str_chat_id):
                 return
 
-            # ===================================================================
-            # 🟢 ثبت آخرین زمان فعالیت کاربر برای سیستم پاکسازی رم
-            # ===================================================================
             if str_chat_id not in user_states:
                 user_states[str_chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None, "last_time": time.time()}
             else:
-                user_states[str_chat_id]["last_time"] = time.time()  # آپدیت زمان فعالیت
+                user_states[str_chat_id]["last_time"] = time.time()  
 
             if user_text_lower.startswith(f"{ADMIN_CODE} pro=") or user_text_lower.startswith(f"{ADMIN_CODE}pro="):
                 parts = user_text_str.split("=", 1)
@@ -1975,8 +2062,8 @@ else:
                                 "expire_date": None,
                                 "last_reset": "",
                                 "chat": 10,
-                                "image": 5,
-                                "edit_image": 1,
+                                "image": 0,
+                                "edit_image": 0,
                                 "podcast": 2,
                                 "tts": 5,
                                 "file": 1,
@@ -1995,8 +2082,8 @@ else:
                         user_credits_db[target_id]["expire_date"] = expire_time.isoformat()
                         
                         user_credits_db[target_id]["chat"] = 999999
-                        user_credits_db[target_id]["image"] = 20
-                        user_credits_db[target_id]["edit_image"] = 10
+                        user_credits_db[target_id]["image"] = 10
+                        user_credits_db[target_id]["edit_image"] = 5
                         user_credits_db[target_id]["podcast"] = 999999
                         user_credits_db[target_id]["tts"] = 999999
                         user_credits_db[target_id]["file"] = 999999
@@ -2020,6 +2107,8 @@ else:
                             user_credits_db[target_id]["is_premium"] = False
                             user_credits_db[target_id]["expire_date"] = None
                             user_credits_db[target_id]["last_reset"] = "" 
+                            user_credits_db[target_id]["image"] = 0
+                            user_credits_db[target_id]["edit_image"] = 0
                             save_db(user_credits_db)
                         await send_with_keyboard(client, chat_id, f"✅ اشتراک کاربر `{target_id}` لغو شد و به رایگان تبدیل گشت.", False)
                         try:
@@ -2257,8 +2346,8 @@ else:
 🎙 ساخت پادکست: نامحدود ∞
 📁 تحلیل فایل و سند: نامحدود ∞
 📝 تبدیل فایل صوتی به متن: نامحدود ∞
-🪄 ویرایش تصویر: ۱۰ عدد
-🎨 تولید تصویر: ۲۰ عدد
+🪄 ویرایش تصویر: 5 عدد
+🎨 تولید تصویر: 10 عدد
 ━━━━━━━━━━━━━━━━━━━
 
 💳 **هزینه اشتراک یک ماهه:** 250 هزار تومان
@@ -2306,6 +2395,22 @@ else:
                 return
 
             if user_text_str in["/edit_image", "ویرایش تصاویر 🪄"]:
+                # متوقف کردن کاربران رایگان در همین مرحله و نشان دادن پیام (بدون درخواست عکس)
+                if not creds.get("is_premium", False):
+                    msg = """کاربر گرامی،
+
+ضمن تشکر از توجه شما، به اطلاع می‌رساند که قابلیت ویرایش تصاویر در حال حاضر به صورت رایگان در دسترس نمی‌باشد. این سرویس به دلیل نیاز به منابع پردازشی بالا، جزو خدمات ویژه و نیازمند تهیه اشتراک است.
+
+با این حال، شما می‌توانید از تست رایگان و امکانات سایر بخش‌های پلتفرم ما بهره‌مند شوید.
+
+ما همواره در تلاش برای افزایش منابع و اضافه کردن خدمات رایگان بیشتر به کاربران عزیز هستیم.
+
+در حال حاضر، می‌توانید از سایر بخش‌های رایگان موجود استفاده کنید و یا برای دسترسی کامل و نامحدود به تمامی قابلیت‌ها (از جمله ویرایش تصاویر)، اشتراک تهیه نمایید.
+
+از صبوری و همراهی شما سپاسگزاریم."""
+                    await send_with_keyboard(client, chat_id, msg, False)
+                    return
+                
                 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)
@@ -2321,9 +2426,6 @@ else:
                 await send_with_keyboard(client, chat_id, "📻 شما وارد بخش **ساخت پادکست** شدید.\n\nلطفاً موضوع پادکست خود را بفرستید.\nمثال: درباره تاریخچه پیدایش قهوه با ۳ گوینده یک پادکست جذاب بساز . همچنین این قسمت متصل به مدل زبانی است و درخواست هارو قبل از ساخت درک میکنه. میتوانید مقاله کامل یک سایت بفرستید با تبلیغات یا هرچی، هوش مصنوعی متن مقاله رو استخراج و پادکست براتون میسازه . در توضیحات امکان مشخص کردن تعداد گوینده به همراه اسم شون نیز از سمت شما امکان پذیر است.\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
-            # ===============================================
-            # بخش‌های تغییر صدا و کلون کردن صدا
-            # ===============================================
             if user_text_str in["/vc", "تغییر صدا 🎙️"]:
                 user_states[str_chat_id]["mode"] = "vc_waiting_for_voice"
                 user_states[str_chat_id]["file_bytes"] = None
@@ -2350,7 +2452,6 @@ else:
 *(برای خروج دکمه «برگشت♻️» را بزنید)*"""
                 await send_with_keyboard(client, chat_id, clone_text, True)
                 return
-            # ===============================================
 
             if user_text_str in["/file", "تحلیل فایل 📁"]:
                 user_states[str_chat_id]["mode"] = "file_waiting_for_file"
@@ -2419,8 +2520,8 @@ else:
                             
                         inviter_data["expire_date"] = new_exp.isoformat()
                         inviter_data["chat"] = 999999
-                        inviter_data["image"] = 20
-                        inviter_data["edit_image"] = 10
+                        inviter_data["image"] = 10
+                        inviter_data["edit_image"] = 5
                         inviter_data["podcast"] = 999999
                         inviter_data["tts"] = 999999
                         inviter_data["file"] = 999999
@@ -2591,9 +2692,6 @@ else:
                     await send_with_keyboard(client, chat_id, "⚠️ لطفاً موضوع مقاله خود را به صورت متنی بفرستید.", False)
                 return
 
-            # ===============================================
-            # بخش‌های تغییر صدا و کلون صدا
-            # ===============================================
             elif current_mode == "vc_waiting_for_voice":
                 if not is_file: return await send_with_keyboard(client, chat_id, "⚠️ لطفاً یک فایل صوتی یا ویس ارسال کنید.", False)
                 await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل صوتی شما...", False)
@@ -2631,7 +2729,6 @@ else:
                     asyncio.create_task(process_standard_vc_job(client, chat_id, src_bytes, ref_bytes, model["name"], "voice_conv"))
                 
                 elif choice in LEGACY_MODELS:
-                    # اگر شماره مدل ۸ نبود و اعتبار هم صفر بود، ارور بده
                     if choice != "8" and creds["voice_conv"] <= 0: 
                         return await send_with_keyboard(client, chat_id, "❌ سهمیه تغییر صدای شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
                     
@@ -2700,10 +2797,8 @@ if __name__ == "__main__":
     if bot_token:
         loop = asyncio.get_event_loop()
         
-        # 🟢 تونل‌های پرسرعت: ارتقا از 32 کارگر به 256 کارگر برای پاسخگویی همزمان به هزاران نفر
         loop.set_default_executor(concurrent.futures.ThreadPoolExecutor(max_workers=256))
         
-        # 🟢 روشن کردن سیستم پاکسازی هوشمند رم در پس‌زمینه
         loop.create_task(ram_garbage_collector())
         
         print("🚀 ربات آلفا پرو (نسخه صنعتی + دیتابیس هوشمند + پاکسازی رم + 256 تونل همزمان) روشن شد...")