Update main.py

main.py

@@ -86,33 +86,43 @@ user_credits_db = load_db()
 
 def get_user_credits(chat_id):
     str_chat_id = str(chat_id)
-    today_str = datetime.date.today().isoformat()
     
     if str_chat_id not in user_credits_db:
-        # کاربر جدید پیش‌فرض رایگان است
-        user_credits_db[str_chat_id] = {"is_premium": False}
+        # کاربر جدید پیش‌فرض یک بسته رایگان کوچیک برای تست می‌گیره
+        user_credits_db[str_chat_id] = {
+            "is_premium": False,
+            "expire_date": None,
+            "chat": 20,          # چت رایگان
+            "image": 5,          # عکس رایگان
+            "edit_image": 1,     # ویرایش رایگان
+            "podcast": 1,        # پادکست رایگان
+            "tts": 5,            # تبدیل متن به صدا رایگان
+            "file": 1,           # تحلیل فایل رایگان
+            "stt": 5             # تبدیل صدا به متن رایگان
+        }
+        save_db(user_credits_db)
         
     user_data = user_credits_db[str_chat_id]
-    is_premium = user_data.get("is_premium", False)
     
-    # تعریف سهمیه‌ها بر اساس وضعیت کاربری (رایگان یا ویژه)
-    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
-        for k, v in limits.items():
-            user_data[k] = v
-        save_db(user_credits_db)
-        
+    # بررسی انقضای اشتراک ماهانه
+    if user_data.get("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
+                # صفر کردن اعتبارهای ویژه پس از اتمام یک ماه
+                user_data["chat"] = 0
+                user_data["image"] = 0
+                user_data["edit_image"] = 0
+                user_data["podcast"] = 0
+                user_data["tts"] = 0
+                user_data["file"] = 0
+                user_data["stt"] = 0
+                save_db(user_credits_db)
+        except Exception:
+            pass
+            
     return user_data
 
 # --- تابع تبدیل اعداد فارسی/عربی به انگلیسی ---
@@ -130,13 +140,13 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (متصل به دیتاست + سیستم اشتراکی VIP + پادکست + ویرایش عکس) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (متصل به دیتاست + سیستم اشتراکی تجاری + پادکست + ویرایش عکس) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
 
 
-# --- ساختار کیبورد آپدیت شده (۵ ردیف) ---
+# --- ساختار کیبورد آپدیت شده ---
 MAIN_KEYPAD_DICT = {
     "rows": [
         {
@@ -165,6 +175,11 @@ MAIN_KEYPAD_DICT = {
         {
             "buttons": [
                 {"id": "account_btn", "type": "Simple", "button_text": "حساب کاربری 👤"},
+                {"id": "buy_btn", "type": "Simple", "button_text": "خرید اشتراک 💎"}
+            ]
+        },
+        {
+            "buttons": [
                 {"id": "cancel_btn", "type": "Simple", "button_text": "برگشت♻️"}
             ]
         }
@@ -354,7 +369,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, "❌ اعتبار پیام‌های چت شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
     
@@ -451,7 +466,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, "❌ اعتبار ساخت عکس شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     if not HF_TOKENS: return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
     
@@ -527,7 +542,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, "❌ اعتبار ویرایش عکس شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     if not HF_TOKENS:
         return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
@@ -585,7 +600,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, "❌ اعتبار تبدیل متن به صدای شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     try:
         proc_msg = await send_with_keyboard(client, chat_id, f"⏳ در حال ساخت صدا با «{speaker_name}»...\n(لطفاً صبور باشید)", False)
@@ -636,7 +651,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, "❌ اعتبار پادکست شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     proc_msg = await send_with_keyboard(client, chat_id, "📻 در حال بررسی موضوع و نگارش سناریوی پادکست توسط هوش مصنوعی...\n(لطفاً صبور باشید)", False)
     available_speakers = []
@@ -726,7 +741,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, "❌ اعتبار تبدیل صدا به متن شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
     
@@ -773,7 +788,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, "❌ اعتبار تحلیل فایل شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلید جیمینای تنظیم نیست.", False)
     
@@ -867,12 +882,12 @@ else:
             if chat_id not in user_states:
                 user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None, "is_admin": False}
 
-            # 🛠 --- بخش مدیریت ادمین --- 🛠
+            # 🛠 --- سیستم پنل مدیریت (Admin) --- 🛠
             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)
+                    await send_with_keyboard(client, chat_id, "✅ ورود موفقیت‌آمیز به پنل مدیریت.\n\nبرای ویژه کردن کاربر:\n`/upgrade <ID>`\nبرای لغو اشتراک:\n`/downgrade <ID>`", False)
                 else:
                     await send_with_keyboard(client, chat_id, "❌ رمز عبور اشتباه است.", False)
                 return
@@ -886,9 +901,22 @@ else:
                             user_credits_db[target_id] = {"is_premium": False}
                         
                         user_credits_db[target_id]["is_premium"] = True
-                        user_credits_db[target_id]["last_reset"] = "" # برای اینکه در درخواست بعدی محدودیت‌ها ۱۰۰ برابری شوند
+                        
+                        # تنظیم تاریخ انقضا برای ۳۰ روز آینده
+                        expire_time = datetime.datetime.now() + datetime.timedelta(days=30)
+                        user_credits_db[target_id]["expire_date"] = expire_time.isoformat()
+                        
+                        # شارژ اعتبار ماهانه
+                        user_credits_db[target_id]["chat"] = 1000
+                        user_credits_db[target_id]["image"] = 100
+                        user_credits_db[target_id]["edit_image"] = 50
+                        user_credits_db[target_id]["podcast"] = 100
+                        user_credits_db[target_id]["tts"] = 500
+                        user_credits_db[target_id]["file"] = 50
+                        user_credits_db[target_id]["stt"] = 100
+                        
                         save_db(user_credits_db)
-                        await send_with_keyboard(client, chat_id, f"✅ اشتراک ویژه برای کاربر `{target_id}` فعال شد.", False)
+                        await send_with_keyboard(client, chat_id, f"✅ حساب کاربر `{target_id}` به مدت ۳۰ روز شارژ شد و به پرو ارتقا یافت.", False)
                     else:
                         await send_with_keyboard(client, chat_id, "⚠️ فرمت صحیح: `/upgrade ID`", False)
                     return
@@ -899,9 +927,17 @@ else:
                         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"] = "" 
+                            user_credits_db[target_id]["expire_date"] = None
+                            # صفر کردن اعتبار
+                            user_credits_db[target_id]["chat"] = 0
+                            user_credits_db[target_id]["image"] = 0
+                            user_credits_db[target_id]["edit_image"] = 0
+                            user_credits_db[target_id]["podcast"] = 0
+                            user_credits_db[target_id]["tts"] = 0
+                            user_credits_db[target_id]["file"] = 0
+                            user_credits_db[target_id]["stt"] = 0
                             save_db(user_credits_db)
-                        await send_with_keyboard(client, chat_id, f"✅ اشتراک کاربر `{target_id}` لغو و به رایگان تبدیل شد.", False)
+                        await send_with_keyboard(client, chat_id, f"✅ اشتراک کاربر `{target_id}` لغو شد.", False)
                     return
 
             # تشخیص فایل
@@ -931,43 +967,64 @@ else:
                 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)
-                status_text = "🌟 نسخه پرو (اشتراک ویژه)" if is_prem else "🥉 نسخه رایگان (آزمایشی)"
                 
-                upgrade_text = ""
-                if not is_prem:
-                    upgrade_text = """
-🎁 **ارتقا به نسخه پرو (ویژه):**
-با تهیه اشتراک ویژه، محدودیت‌های شما به شکل زیر افزایش می‌یابد:
-- ۱۰۰ پیام چت، ۲۰ تحلیل فایل، ۵۰ پادکست، ۱۰۰ تبدیل متن به صدا، ۱۰ ویرایش تصویر در روز!
-
-💳 **هزینه اشتراک ماهانه:** ۱۹۹,۰۰۰ تومان
-شماره کارت: `6104338667771394`
-
-پس از واریز، رسید خود را به همراه **شناسه یکتا** (که در بالا نوشته شده) به پشتیبانی زیر ارسال کنید تا اشتراکتان فعال شود:
-🆔 پشتیبانی: @H_a_m_e_d100
-"""
+                if is_prem:
+                    expire_txt = "نامشخص"
+                    if creds.get("expire_date"):
+                        exp_d = datetime.datetime.fromisoformat(creds["expire_date"])
+                        expire_txt = exp_d.strftime("%Y-%m-%d %H:%M")
+                    status_text = f"🌟 نسخه پرو (معتبر تا {expire_txt})"
+                else:
+                    status_text = "🥉 نسخه رایگان (آزمایشی)"
 
                 account_profile = f"""👤 **اطلاعات حساب کاربری شما**
 
 🔹 **شناسه یکتا:** `{chat_id}`
 🔹 **وضعیت اشتراک:** {status_text}
 
-📊 **سهمیه باقی‌مانده شما در امروز:**
+📊 **سهمیه فعلی شما:**
 - 💬 چت هوشمند: {creds['chat']} پیام
 - 🎨 ساخت عکس: {creds['image']} تصویر
 - 🪄 ویرایش عکس پیشرفته: {creds['edit_image']} تصویر
 - 🎙 ساخت پادکست: {creds['podcast']} برنامه
 - 🗣 تبدیل متن به صدا: {creds['tts']} فایل
 - 📁 تحلیل فایل: {creds['file']} فایل
-- 📝 تبدیل صدا به متن: {creds['stt']} فایل
-{upgrade_text}"""
+- 📝 تبدیل صدا به متن: {creds['stt']} فایل"""
                 await send_with_keyboard(client, chat_id, account_profile, True)
                 return
 
+            # --- دکمه جدید: خرید اشتراک ---
+            if user_text_str in ["/buy", "خرید اشتراک 💎"]:
+                buy_text = f"""💎 **خرید اشتراک ویژه آلفا پرو (یک ماهه)**
+
+با تهیه اشتراک ویژه، محدودیت‌ها را کنار بزنید و از نهایت قدرت هوش مصنوعی لذت ببرید! 🚀
+
+🎁 **بسته طلایی یک‌ماهه شامل:**
+🗣 ۵۰۰ فایل تبدیل متن به صدا (با کیفیت طبیعی)
+🎙 ۱۰۰ برنامه ساخت پادکست کامل
+🪄 ۵۰ عکس ویرایش حرفه‌ای (Flux.2)
+🎨 ۱۰۰ تصویرسازی فوق‌پیشرفته
+📁 ۵۰ تحلیل فایل و سند
+💬 چت هوشمند (۱۰۰۰ پیام)
+📝 ۱۰۰ فایل تبدیل صدا به متن
+
+💳 **هزینه اشتراک یک ماهه:** ۱۹۹,۰۰۰ تومان
+
+💳 **شماره کارت جهت واریز:**
+`6104338667771394`
+
+✅ **نحوه فعال‌سازی:**
+پس از واریز مبلغ، لطفاً رسید پرداختی را به همراه **شناسه یکتای خود** (`{chat_id}`) به آیدی پشتیبانی زیر ارسال کنید تا اشتراک شما فورا فعال گردد:
+
+👨‍💻 **ارتباط با پشتیبانی:**
+🆔 @H_a_m_e_d100"""
+                await send_with_keyboard(client, chat_id, buy_text, True)
+                return
+
             if user_text_str in ["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
                 user_states[chat_id]["mode"] = "chat"
                 user_states[chat_id]["history"] = []