Update main.py

main.py

@@ -18,18 +18,12 @@ from huggingface_hub import AsyncInferenceClient, HfApi, hf_hub_download
 from PIL import Image
 from pydub import AudioSegment
 
-# --- رمز عبور ادمین برای فعال‌سازی اشتراک‌ها ---
-ADMIN_SECRET = "hmd744_admin"
+# --- رمز عبور ادمین برای ارتقای کاربران (اینجا می‌توانید تغییر دهید) ---
+ADMIN_PASSWORD = "alpha2026"
 
 # --- ثبت زمان روشن شدن ربات برای جلوگیری از پرتاب پیام‌های قدیمی ---
 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"
@@ -95,21 +89,28 @@ 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] = {"plan": "free"}
+        # کاربر جدید پیش‌فرض رایگان است
+        user_credits_db[str_chat_id] = {"is_premium": False}
         
     user_data = user_credits_db[str_chat_id]
-    current_plan = user_data.get("plan", "free")
+    is_premium = user_data.get("is_premium", False)
     
-    # بازنشانی خودکار سهمیه روزانه یا در صورت تغییر پلن
-    if user_data.get("last_reset") != today_str or user_data.get("last_plan_set") != current_plan:
+    # تعریف سهمیه‌ها بر اساس وضعیت کاربری (رایگان یا ویژه)
+    limits = {
+        "chat": 100 if is_premium else 10,
+        "image": 50 if is_premium else 5,
+        "edit_image": 10 if is_premium else 1,
+        "podcast": 50 if is_premium else 1,
+        "tts": 100 if is_premium else 5,
+        "file": 20 if is_premium else 1,
+        "stt": 50 if is_premium else 5
+    }
+    
+    # بازنشانی خودکار سهمیه روزانه یا هنگام تغییر اشتراک
+    if user_data.get("last_reset") != today_str:
         user_data["last_reset"] = today_str
-        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
-            
+        for k, v in limits.items():
+            user_data[k] = v
         save_db(user_credits_db)
         
     return user_data
@@ -129,7 +130,7 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (متصل به دیتاست + اشتراک ویژه + پنل ادمین + پادکست + ویرایش عکس) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (متصل به دیتاست + سیستم اشتراکی VIP + پادکست + ویرایش عکس) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -353,7 +354,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, "❌ اعتبار پیام‌های چت شما برای امروز تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار پیام‌های چت شما برای امروز تمام شده است.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
     
@@ -450,7 +451,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, "❌ اعتبار ساخت عکس روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت عکس روزانه شما تمام شده است.", False)
 
     if not HF_TOKENS: return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
     
@@ -526,7 +527,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, "❌ اعتبار ویرایش عکس روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار ویرایش عکس روزانه شما تمام شده است.", False)
 
     if not HF_TOKENS:
         return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
@@ -584,7 +585,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, "❌ اعتبار تبدیل متن به صدای شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل متن به صدای شما تمام شده است.", False)
 
     try:
         proc_msg = await send_with_keyboard(client, chat_id, f"⏳ در حال ساخت صدا با «{speaker_name}»...\n(لطفاً صبور باشید)", False)
@@ -635,7 +636,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, "❌ اعتبار ساخت پادکست روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت پادکست روزانه شما تمام شده است.", False)
 
     proc_msg = await send_with_keyboard(client, chat_id, "📻 در حال بررسی موضوع و نگارش سناریوی پادکست توسط هوش مصنوعی...\n(لطفاً صبور باشید)", False)
     available_speakers = []
@@ -725,7 +726,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, "❌ اعتبار تبدیل صدا به متن شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل صدا به متن شما تمام شده است.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
     
@@ -772,7 +773,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, "❌ اعتبار تحلیل فایل روزانه شما تمام شده است.\nبرای ارتقا به بخش «حساب کاربری 👤» مراجعه کنید.", False)
+        return await send_with_keyboard(client, chat_id, "❌ اعتبار تحلیل فایل روزانه شما تمام شده است.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلید جیمینای تنظیم نیست.", False)
     
@@ -862,54 +863,46 @@ else:
             # تشخیص محتوای متنی
             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
+            if chat_id not in user_states:
+                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None, "is_admin": False}
+
+            # 🛠 --- بخش مدیریت ادمین --- 🛠
+            if user_text_str.startswith("/admin_login"):
+                parts = user_text_str.split()
+                if len(parts) == 2 and parts[1] == ADMIN_PASSWORD:
+                    user_states[chat_id]["is_admin"] = True
+                    await send_with_keyboard(client, chat_id, "✅ با موفقیت به عنوان ادمین وارد شدید.\n\nبرای ارتقای یک کاربر به نسخه پرو:\n`/upgrade <شناسه_کاربر>`\n\nبرای لغو اشتراک یک کاربر:\n`/downgrade <شناسه_کاربر>`", False)
+                else:
+                    await send_with_keyboard(client, chat_id, "❌ رمز عبور اشتباه است.", False)
                 return
 
-            # ==========================================
-            # پایان بخش ادمین
-            # ==========================================
+            if user_states[chat_id].get("is_admin"):
+                if user_text_str.startswith("/upgrade"):
+                    parts = user_text_str.split()
+                    if len(parts) == 2:
+                        target_id = parts[1]
+                        if target_id not in user_credits_db:
+                            user_credits_db[target_id] = {"is_premium": False}
+                        
+                        user_credits_db[target_id]["is_premium"] = True
+                        user_credits_db[target_id]["last_reset"] = "" # برای اینکه در درخواست بعدی محدودیت‌ها ۱۰۰ برابری شوند
+                        save_db(user_credits_db)
+                        await send_with_keyboard(client, chat_id, f"✅ اشتراک ویژه برای کاربر `{target_id}` فعال شد.", False)
+                    else:
+                        await send_with_keyboard(client, chat_id, "⚠️ فرمت صحیح: `/upgrade ID`", False)
+                    return
+                    
+                if user_text_str.startswith("/downgrade"):
+                    parts = user_text_str.split()
+                    if len(parts) == 2:
+                        target_id = parts[1]
+                        if target_id in user_credits_db:
+                            user_credits_db[target_id]["is_premium"] = False
+                            user_credits_db[target_id]["last_reset"] = "" 
+                            save_db(user_credits_db)
+                        await send_with_keyboard(client, chat_id, f"✅ اشتراک کاربر `{target_id}` لغو و به رایگان تبدیل شد.", False)
+                    return
 
             # تشخیص فایل
             is_file = False
@@ -932,23 +925,36 @@ else:
                             file_name = file_attr.get('file_name', file_name) if isinstance(file_attr, dict) else file_name
                             break
 
-            if chat_id not in user_states:
-                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
-
             if user_text_str in ["/start", "سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
-                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
+                user_states[chat_id]["mode"] = None
+                user_states[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)
-                plan_status = "🌟 نسخه پرو (اشتراک ویژه فعال)" if creds.get("plan") == "premium" else "🥉 نسخه رایگان (آزمایشی)"
+                is_prem = creds.get("is_premium", False)
+                status_text = "🌟 نسخه پرو (اشتراک ویژه)" if is_prem else "🥉 نسخه رایگان (آزمایشی)"
                 
+                upgrade_text = ""
+                if not is_prem:
+                    upgrade_text = """
+🎁 **ارتقا به نسخه پرو (ویژه):**
+با تهیه اشتراک ویژه، محدودیت‌های شما به شکل زیر افزایش می‌یابد:
+- ۱۰۰ پیام چت، ۲۰ تحلیل فایل، ۵۰ پادکست، ۱۰۰ تبدیل متن به صدا، ۱۰ ویرایش تصویر در روز!
+
+💳 **هزینه اشتراک ماهانه:** ۱۹۹,۰۰۰ تومان
+شماره کارت: `6104338667771394`
+
+پس از واریز، رسید خود را به همراه **شناسه یکتا** (که در بالا نوشته شده) به پشتیبانی زیر ارسال کنید تا اشتراکتان فعال شود:
+🆔 پشتیبانی: @H_a_m_e_d100
+"""
+
                 account_profile = f"""👤 **اطلاعات حساب کاربری شما**
 
 🔹 **شناسه یکتا:** `{chat_id}`
-🔹 **وضعیت اشتراک:** {plan_status}
+🔹 **وضعیت اشتراک:** {status_text}
 
 📊 **سهمیه باقی‌مانده شما در امروز:**
 - 💬 چت هوشمند: {creds['chat']} پیام
@@ -957,56 +963,46 @@ else:
 - 🎙 ساخت پادکست: {creds['podcast']} برنامه
 - 🗣 تبدیل متن به صدا: {creds['tts']} فایل
 - 📁 تحلیل فایل: {creds['file']} فایل
-- 📝 تبدیل صدا به متن: {creds['stt']} فایل"""
-
-                if creds.get("plan") != "premium":
-                    account_profile += """
-
-💎 **ارتقا به نسخه پرو (نامحدود):**
-با ارتقا به نسخه پرو امکانات شما بی‌نهایت افزایش یافته و محدودیت‌ها برداشته می‌شود.
-
-💰 **هزینه اشتراک ماهانه:** 199 هزار تومان
-💳 **شماره کارت:** `6104338667771394`
-
-جهت فعال‌سازی، لطفاً «رسید واریز» و «شناسه یکتا» خود را به پشتیبانی تلگرام/روبیکا ارسال کنید:
-👨‍💻 **پشتیبانی:** @H_a_m_e_d100
-*(پس از ارسال رسید، اشتراک شما سریعاً فعال می‌گردد)*"""
-                
+- 📝 تبدیل صدا به متن: {creds['stt']} فایل
+{upgrade_text}"""
                 await send_with_keyboard(client, chat_id, account_profile, True)
                 return
 
             if user_text_str in ["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
-                user_states[chat_id] = {"mode": "chat", "text": "", "history": []}
+                user_states[chat_id]["mode"] = "chat"
+                user_states[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", "text": "", "history": []}
+                user_states[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", "text": "", "history": [], "file_bytes": None, "file_name": None}
+                user_states[chat_id]["mode"] = "image_edit_waiting_for_image"
+                user_states[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", "text": "", "history": []}
+                user_states[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", "text": "", "history": []}
+                user_states[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", "text": "", "history": [], "file_bytes": None, "file_name": None}
+                user_states[chat_id]["mode"] = "file_waiting_for_file"
+                user_states[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", "text": "", "history": []}
+                user_states[chat_id]["mode"] = "stt_waiting_for_audio"
                 await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل خود (ویس، آهنگ، ویدیو و...) را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
@@ -1124,5 +1120,5 @@ else:
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
     if bot_token:
-        print("ربات آلفا با موفقیت به همراه دیتابیس ابری (HF Datasets) و پنل ادمین روشن شد...")
+        print("ربات آلفا پرو با سیستم اشتراک VIP روشن شد...")
         bot.run()