Update main.py

main.py

@@ -151,7 +151,7 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو + مدیریت اشتراک حرفه‌ای + دیتاست) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (نسخه پرو + مدیریت اشتراک نامحدود + دیتاست) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -447,9 +447,10 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         await send_with_keyboard(client, chat_id, "❌ متأسفانه پاسخی دریافت نشد. (شاید سایز فایل بیش از حد مجاز بوده است)", False)
         return
 
-    # کسر اعتبار در دیتاست
-    user_credits_db[str_chat_id]["chat"] -= 1
-    save_db(user_credits_db)
+    # کسر اعتبار در دیتاست (اگر نامحدود نیست)
+    if not creds.get("is_premium"):
+        user_credits_db[str_chat_id]["chat"] -= 1
+        save_db(user_credits_db)
 
     history.append({"role": "model", "parts": [{"text": final_answer}]})
     user_states[chat_id]["history"] = history
@@ -537,7 +538,7 @@ 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)
 
@@ -593,7 +594,7 @@ 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)
 
@@ -648,9 +649,10 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
         except Exception: pass
 
         if audio_bytes:
-            # کسر اعتبار در دیتاست
-            user_credits_db[str_chat_id]["tts"] -= 1
-            save_db(user_credits_db)
+            # کسر اعتبار در دیتاست (اگر نامحدود نیست)
+            if not creds.get("is_premium"):
+                user_credits_db[str_chat_id]["tts"] -= 1
+                save_db(user_credits_db)
             
             file_name = f"audio_{random.randint(1000, 999999)}.mp3"
             with open(file_name, "wb") as f: f.write(audio_bytes)
@@ -732,9 +734,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)
 
-        # کسر اعتبار در دیتاست
-        user_credits_db[str_chat_id]["podcast"] -= 1
-        save_db(user_credits_db)
+        # کسر اعتبار در دیتاست (اگر نامحدود نیست)
+        if not creds.get("is_premium"):
+            user_credits_db[str_chat_id]["podcast"] -= 1
+            save_db(user_credits_db)
 
         file_name = f"final_podcast_{random.randint(1000, 999999)}.mp3"
         combined_audio.export(file_name, format="mp3")
@@ -791,9 +794,10 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     except Exception: pass
 
     if transcribed_text:
-        # کسر اعتبار در دیتاست
-        user_credits_db[str_chat_id]["stt"] -= 1
-        save_db(user_credits_db)
+        # کسر اعتبار در دیتاست (اگر نامحدود نیست)
+        if not creds.get("is_premium"):
+            user_credits_db[str_chat_id]["stt"] -= 1
+            save_db(user_credits_db)
         await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
     else: 
         await send_with_keyboard(client, chat_id, f"❌ تبدیل فایل به متن ناموفق بود.", True)
@@ -837,9 +841,10 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     except Exception: pass
 
     if final_answer: 
-        # کسر اعتبار در دیتاست
-        user_credits_db[str_chat_id]["file"] -= 1
-        save_db(user_credits_db)
+        # کسر اعتبار در دیتاست (اگر نامحدود نیست)
+        if not creds.get("is_premium"):
+            user_credits_db[str_chat_id]["file"] -= 1
+            save_db(user_credits_db)
         await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
     else: 
         await send_with_keyboard(client, chat_id, "❌ پاسخی دریافت نشد.", True)
@@ -934,21 +939,21 @@ else:
                         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]["chat"] = 999999
                         user_credits_db[target_id]["image"] = 20
                         user_credits_db[target_id]["edit_image"] = 10
-                        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
+                        user_credits_db[target_id]["podcast"] = 999999
+                        user_credits_db[target_id]["tts"] = 999999
+                        user_credits_db[target_id]["file"] = 999999
+                        user_credits_db[target_id]["stt"] = 999999
                         
                         save_db(user_credits_db)
                         await send_with_keyboard(client, chat_id, f"✅ حساب کاربر `{target_id}` به مدت ۳۰ روز شارژ شد و به پرو ارتقا یافت.", False)
                         
                         # ارسال پیام اطلاع‌رسانی به خود کاربر هدف
                         try:
-                            await send_with_keyboard(client, target_id, "🎉 **کاربر گرامی، تبریک!**\n\nحساب شما با موفقیت توسط پشتیبانی به **🌟 نسخه پرو (ویژه)** ارتقا یافت.\nهم‌اکنون بسته‌های یک‌ماهه شما فعال گردید.\n\nجهت مشاهده جزئیات روی دکمه «حساب کاربری 👤» کلیک کنید.", True)
+                            await send_with_keyboard(client, target_id, "🎉 **کاربر گرامی، تبریک!**\n\nحساب شما با موفقیت توسط پشتیبانی به **🌟 نسخه پرو (ویژه)** ارتقا یافت.\nهم‌اکنون بسته‌های نامحدود و طلایی یک‌ماهه شما فعال گردید.\n\nجهت مشاهده جزئیات روی دکمه «حساب کاربری 👤» کلیک کنید.", True)
                         except Exception as e:
                             print(f"Failed to notify target user: {e}")
                             
@@ -1004,7 +1009,7 @@ 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)
@@ -1020,47 +1025,59 @@ else:
                     status_text = "🥉 نسخه رایگان (آزمایشی)"
                     daily_note = "*سهمیه شما هر روز ساعت ۰۰:۰۰ بامداد به صورت خودکار مجدداً شارژ می‌گردد.*"
 
+                # استفاده از عبارت نامحدود برای کاربران پرو
+                chat_rem = "نامحدود ∞" if is_prem else creds['chat']
+                podcast_rem = "نامحدود ∞" if is_prem else creds['podcast']
+                tts_rem = "نامحدود ∞" if is_prem else creds['tts']
+                file_rem = "نامحدود ∞" if is_prem else creds['file']
+                stt_rem = "نامحدود ∞" if is_prem else creds['stt']
+                
+                # برای عکس‌ها همان تعداد را نشان می‌دهیم
+                image_rem = creds['image']
+                edit_image_rem = creds['edit_image']
+
                 account_profile = f"""👤 **اطلاعات حساب کاربری شما**
 
 🔹 **شناسه یکتا:** `{chat_id}`
 🔹 **وضعیت اشتراک:** {status_text}
 
 📊 **سهمیه باقی‌مانده شما:**
-- 💬 چت هوشمند: {creds['chat']} پیام
-- 🎨 ساخت عکس: {creds['image']} تصویر
-- 🪄 ویرایش عکس پیشرفته: {creds['edit_image']} تصویر
-- 🎙 ساخت پادکست: {creds['podcast']} برنامه
-- 🗣 تبدیل متن به صدا: {creds['tts']} فایل
-- 📁 تحلیل فایل: {creds['file']} فایل
-- 📝 تبدیل صدا به متن: {creds['stt']} فایل
+- 💬 چت هوشمند: {chat_rem}
+- 🎨 ساخت عکس: {image_rem} تصویر
+- 🪄 ویرایش عکس پیشرفته: {edit_image_rem} تصویر
+- 🎙 ساخت پادکست: {podcast_rem}
+- 🗣 تبدیل متن به صدا: {tts_rem}
+- 📁 تحلیل فایل: {file_rem}
+- 📝 تبدیل صدا به متن: {stt_rem}
 
 {daily_note}"""
                 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}`
 
@@ -1069,17 +1086,17 @@ else:
                 await send_with_keyboard(client, chat_id, buy_text, True)
                 return
 
-            # --- انتقال اکانت ---
+            # --- دکمه انتقال اکانت ---
             if user_text_str in ["/transfer", "انتقال اکانت از برنامه به ربات"]:
                 transfer_text = f"""🔄 **انتقال اکانت از برنامه به ربات**
 
 کاربر گرامی، در صورتی که داخل برنامه «هوش مصنوعی آلفا» پیش‌تر اشتراک تهیه کرده‌اید، نیازی به خرید مجدد اشتراک داخل ربات نیست! 🎉
 
-کافیست **شناسه یکتای** ربات روبیکای خود را کپی کرده و برای پشتیبانی ما ارسال کنید تا اکانت اشتراکی شما به سرعت از برنامه به ربات روبیکا انتقال داده شود.
+کافیست **شناسه یکتای** ربات روبیکای خود را کپی کرده و برای پشتیبانی ما در برنامه هوش مصنوعی آلفا ارسال کنید تا اکانت اشتراکی شما به سرعت از برنامه به ربات روبیکا انتقال داده شود.
 
 🔑 **شناسه یکتای ربات شما:** `{chat_id}`
 
-👨‍💻 **جهت انتقال، آیدی فوق را به پشتیبانی زیر ارسال کنید:**
+👨‍💻 **دقت کنید شناسه ربات رو به پشتیبانی داخل خود برنامه هوش مصنوعی آلفا ارسال کنید:**
 🆔 @H_a_m_e_d100"""
                 await send_with_keyboard(client, chat_id, transfer_text, True)
                 return