Update main.py

main.py

@@ -18,9 +18,18 @@ from huggingface_hub import AsyncInferenceClient, HfApi, hf_hub_download
 from PIL import Image
 from pydub import AudioSegment
 
+# --- رمز عبور ادمین برای فعال‌سازی اشتراک‌ها ---
+ADMIN_SECRET = "hmd744_admin"
+
 # --- ثبت زمان روشن شدن ربات برای جلوگیری از پرتاب پیام‌های قدیمی ---
 BOT_STARTUP_TIME = time.time()
 
+# --- تنظیمات سهمیه اکانت‌ها ---
+PLAN_LIMITS = {
+    "free": {"chat": 10, "image": 5, "edit_image": 1, "podcast": 1, "tts": 5, "file": 1, "stt": 5},
+    "premium": {"chat": 100, "image": 50, "edit_image": 10, "podcast": 50, "tts": 100, "file": 20, "stt": 50}
+}
+
 # --- سیستم دیتابیس حساب کاربری متصل به دیتاست هاگینگ فیس ---
 DB_FILE = "users_db.json"
 DATASET_REPO = "Opera8/Karbaran-rayegan-tedad"
@@ -86,20 +95,21 @@ def get_user_credits(chat_id):
     today_str = datetime.date.today().isoformat()
     
     if str_chat_id not in user_credits_db:
-        user_credits_db[str_chat_id] = {}
+        user_credits_db[str_chat_id] = {"plan": "free"}
         
     user_data = user_credits_db[str_chat_id]
+    current_plan = user_data.get("plan", "free")
     
-    # بازنشانی خودکار سهمیه روزانه
-    if user_data.get("last_reset") != today_str:
+    # بازنشانی خودکار سهمیه روزانه یا در صورت تغییر پلن
+    if user_data.get("last_reset") != today_str or user_data.get("last_plan_set") != current_plan:
         user_data["last_reset"] = today_str
-        user_data["chat"] = 30
-        user_data["image"] = 5
-        user_data["edit_image"] = 3
-        user_data["podcast"] = 1
-        user_data["tts"] = 5
-        user_data["file"] = 5
-        user_data["stt"] = 5
+        user_data["last_plan_set"] = current_plan
+        
+        # اعمال محدودیت‌ها بر اساس نوع اکانت
+        limits = PLAN_LIMITS.get(current_plan, PLAN_LIMITS["free"])
+        for key, val in limits.items():
+            user_data[key] = val
+            
         save_db(user_credits_db)
         
     return user_data
@@ -119,7 +129,7 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (متصل به دیتاست + نسخه پرو + پادکست + ویرایش عکس) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (متصل به دیتاست + اشتراک ویژه + پنل ادمین + پادکست + ویرایش عکس) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -343,7 +353,7 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
     str_chat_id = str(chat_id)
     creds = get_user_credits(str_chat_id)
     if creds["chat"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار پیام‌های چت شما برای امروز تمام شده است.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار پیام‌های چت شما برای امروز تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
     
@@ -440,7 +450,7 @@ async def process_image(client, chat_id, prompt):
     str_chat_id = str(chat_id)
     creds = get_user_credits(str_chat_id)
     if creds["image"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت عکس روزانه شما تمام شده است.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت عکس روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
 
     if not HF_TOKENS: return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
     
@@ -516,7 +526,7 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     str_chat_id = str(chat_id)
     creds = get_user_credits(str_chat_id)
     if creds["edit_image"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار ویرایش عکس روزانه شما تمام شده است.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار ویرایش عکس روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
 
     if not HF_TOKENS:
         return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
@@ -574,7 +584,7 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
     str_chat_id = str(chat_id)
     creds = get_user_credits(str_chat_id)
     if creds["tts"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل متن به صدای شما تمام شده است.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل متن به صدای شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
 
     try:
         proc_msg = await send_with_keyboard(client, chat_id, f"⏳ در حال ساخت صدا با «{speaker_name}»...\n(لطفاً صبور باشید)", False)
@@ -625,7 +635,7 @@ async def process_podcast(client, chat_id, prompt):
     str_chat_id = str(chat_id)
     creds = get_user_credits(str_chat_id)
     if creds["podcast"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت پادکست روزانه شما تمام شده است.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت پادکست روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
 
     proc_msg = await send_with_keyboard(client, chat_id, "📻 در حال بررسی موضوع و نگارش سناریوی پادکست توسط هوش مصنوعی...\n(لطفاً صبور باشید)", False)
     available_speakers = []
@@ -715,7 +725,7 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     str_chat_id = str(chat_id)
     creds = get_user_credits(str_chat_id)
     if creds["stt"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل صدا به متن شما تمام شده است.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل صدا به متن شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
     
@@ -762,7 +772,7 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     str_chat_id = str(chat_id)
     creds = get_user_credits(str_chat_id)
     if creds["file"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار تحلیل فایل روزانه شما تمام شده است.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار تحلیل فایل روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلید جیمینای تنظیم نیست.", False)
     
@@ -845,14 +855,62 @@ else:
             
             # 🛡 سیستم محدودکننده سرعت کاربر (Rate Limit) 🛡
             last_req_time = user_last_request_time.get(chat_id, 0)
-            if current_time - last_req_time < 1.5:
+            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 ""
+            str_chat_id = str(chat_id)
+
+            # ==========================================
+            # 🔐 بخش ادمین و مدیریت اشتراک‌ها 🔐
+            # ==========================================
             
+            # ۱. گرفتن دسترسی ادمین با رمز
+            if user_text_str == f"/adminlogin {ADMIN_SECRET}":
+                if "_ADMINS_" not in user_credits_db:
+                    user_credits_db["_ADMINS_"] = []
+                if str_chat_id not in user_credits_db["_ADMINS_"]:
+                    user_credits_db["_ADMINS_"].append(str_chat_id)
+                    save_db(user_credits_db)
+                await send_with_keyboard(client, chat_id, "✅ دسترسی مدیریت (پنل ادمین) برای شما فعال شد.\n\nدستورات مجاز:\n`/premium [شناسه_کاربر]` 👈 فعال‌سازی اشتراک\n`/free [شناسه_کاربر]` 👈 لغو اشتراک", True)
+                return
+
+            is_admin = str_chat_id in user_credits_db.get("_ADMINS_", [])
+
+            # ۲. دستور فعال‌سازی اشتراک توسط ادمین
+            if is_admin and user_text_str.startswith("/premium "):
+                target_id = user_text_str.replace("/premium ", "").strip()
+                if target_id not in user_credits_db:
+                    user_credits_db[target_id] = {"plan": "free"}
+                user_credits_db[target_id]["plan"] = "premium"
+                user_credits_db[target_id]["last_reset"] = "" # ریست اجباری سهمیه‌ها برای اعمال آنی
+                save_db(user_credits_db)
+                await send_with_keyboard(client, chat_id, f"✅ اشتراک پرو برای کاربر {target_id} با موفقیت فعال شد.", True)
+                try:
+                    await client.send_message(target_id, "🎉 **تبریک!**\nاشتراک ویژه (نسخه پرو) شما توسط مدیریت فعال شد. هم‌اکنون می‌توانید از تمام امکانات پیشرفته و نامحدود ربات لذت ببرید.")
+                except: pass
+                return
+
+            # ۳. دستور لغو اشتراک توسط ادمین
+            if is_admin and user_text_str.startswith("/free "):
+                target_id = user_text_str.replace("/free ", "").strip()
+                if target_id in user_credits_db:
+                    user_credits_db[target_id]["plan"] = "free"
+                    user_credits_db[target_id]["last_reset"] = ""
+                    save_db(user_credits_db)
+                    await send_with_keyboard(client, chat_id, f"✅ اکانت کاربر {target_id} به حالت رایگان برگشت داده شد.", True)
+                    try:
+                        await client.send_message(target_id, "⚠️ **توجه:**\nاشتراک ویژه شما به پایان رسید و حساب شما به نسخه رایگان (محدود) تغییر یافت.")
+                    except: pass
+                return
+
+            # ==========================================
+            # پایان بخش ادمین
+            # ==========================================
+
             # تشخیص فایل
             is_file = False
             file_name = "unknown_file.jpg"
@@ -885,10 +943,12 @@ else:
             # --- حساب کاربری پویا و متصل به دیتاست ---
             if user_text_str in ["/account", "حساب کاربری 👤"]:
                 creds = get_user_credits(chat_id)
+                plan_status = "🌟 نسخه پرو (اشتراک ویژه فعال)" if creds.get("plan") == "premium" else "🥉 نسخه رایگان (آزمایشی)"
+                
                 account_profile = f"""👤 **اطلاعات حساب کاربری شما**
 
 🔹 **شناسه یکتا:** `{chat_id}`
-🔹 **وضعیت اشتراک:** 🥉 نسخه رایگان (آزمایشی)
+🔹 **وضعیت اشتراک:** {plan_status}
 
 📊 **سهمیه باقی‌مانده شما در امروز:**
 - 💬 چت هوشمند: {creds['chat']} پیام
@@ -897,9 +957,21 @@ else:
 - 🎙 ساخت پادکست: {creds['podcast']} برنامه
 - 🗣 تبدیل متن به صدا: {creds['tts']} فایل
 - 📁 تحلیل فایل: {creds['file']} فایل
-- 📝 تبدیل صدا به متن: {creds['stt']} فایل
+- 📝 تبدیل صدا به متن: {creds['stt']} فایل"""
+
+                if creds.get("plan") != "premium":
+                    account_profile += """
+
+💎 **ارتقا به نسخه پرو (نامحدود):**
+با ارتقا به نسخه پرو امکانات شما بی‌نهایت افزایش یافته و محدودیت‌ها برداشته می‌شود.
 
-*نکته: اطلاعات شما به صورت امن در سرور ابری ذخیره می‌شود و اعتبار شما هر روز ساعت ۰۰:۰۰ مجدداً شارژ می‌گردد.* 🚀"""
+💰 **هزینه اشتراک ماهانه:** 199 هزار تومان
+💳 **شماره کارت:** `6104338667771394`
+
+جهت فعال‌سازی، لطفاً «رسید واریز» و «شناسه یکتا» خود را به پشتیبانی تلگرام/روبیکا ارسال کنید:
+👨‍💻 **پشتیبانی:** @H_a_m_e_d100
+*(پس از ارسال رسید، اشتراک شما سریعاً فعال می‌گردد)*"""
+                
                 await send_with_keyboard(client, chat_id, account_profile, True)
                 return
 
@@ -1052,5 +1124,5 @@ else:
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
     if bot_token:
-        print("ربات آلفا با موفقیت به همراه دیتابیس ابری (HF Datasets) روشن شد...")
+        print("ربات آلفا با موفقیت به همراه دیتابیس ابری (HF Datasets) و پنل ادمین روشن شد...")
         bot.run()