Update main.py

main.py

@@ -8,6 +8,8 @@ import asyncio
 import base64
 import mimetypes
 import io
+import json
+import datetime
 from flask import Flask
 from rubpy.bot import BotClient, filters
 
@@ -19,6 +21,52 @@ from pydub import AudioSegment
 # --- ثبت زمان روشن شدن ربات برای جلوگیری از پرتاب پیام‌های قدیمی ---
 BOT_STARTUP_TIME = time.time()
 
+# --- سیستم دیتابیس حساب کاربری و مدیریت اعتبار ---
+DB_FILE = "users_db.json"
+db_lock = threading.Lock()
+
+def load_db():
+    if os.path.exists(DB_FILE):
+        try:
+            with open(DB_FILE, "r", encoding="utf-8") as f:
+                return json.load(f)
+        except Exception:
+            return {}
+    return {}
+
+def save_db(db_data):
+    with db_lock:
+        try:
+            with open(DB_FILE, "w", encoding="utf-8") as f:
+                json.dump(db_data, f, ensure_ascii=False, indent=4)
+        except Exception as e:
+            print("خطا در ذخیره دیتابیس:", e)
+
+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] = {}
+        
+    user_data = user_credits_db[str_chat_id]
+    
+    # بازنشانی خودکار سهمیه روزانه
+    if user_data.get("last_reset") != today_str:
+        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
+        save_db(user_credits_db)
+        
+    return user_data
+
 # --- تابع تبدیل اعداد فارسی/عربی به انگلیسی ---
 def to_english_digits(text):
     if not text:
@@ -34,43 +82,38 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو + چت چندرسانه‌ای + پرامپت‌نویس + پادکست + ویرایش عکس) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (نسخه پرو + دیتابیس اعتبار + پادکست + ویرایش عکس) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
 
 
-# --- ساختار کیبورد آپدیت شده (دقیقاً طبق چیدمان درخواستی شما در ۵ ردیف) ---
+# --- ساختار کیبورد آپدیت شده (۵ ردیف) ---
 MAIN_KEYPAD_DICT = {
     "rows": [
-        # ردیف اول
         {
             "buttons": [
                 {"id": "chat_btn", "type": "Simple", "button_text": "چت با هوش مصنوعی 🤖"}
             ]
         },
-        # ردیف دوم
         {
             "buttons": [
                 {"id": "img_btn", "type": "Simple", "button_text": "ساخت تصاویر🎨"},
                 {"id": "edit_img_btn", "type": "Simple", "button_text": "ویرایش تصاویر 🪄"}
             ]
         },
-        # ردیف سوم
         {
             "buttons": [
                 {"id": "podcast_btn", "type": "Simple", "button_text": "ساخت پادکست 🎙️"},
                 {"id": "tts_btn", "type": "Simple", "button_text": "تبدیل متن به صدا🗣️"}
             ]
         },
-        # ردیف چهارم
         {
             "buttons": [
                 {"id": "stt_btn", "type": "Simple", "button_text": "فایل صوتی به متن 📝"},
                 {"id": "file_btn", "type": "Simple", "button_text": "تحلیل فایل 📁"}
             ]
         },
-        # ردیف پنجم
         {
             "buttons": [
                 {"id": "account_btn", "type": "Simple", "button_text": "حساب کاربری 👤"},
@@ -81,7 +124,6 @@ MAIN_KEYPAD_DICT = {
     "resize_keyboard": True
 }
 
-
 # --- تابع ارسال منحصراً برای کیبورد پایین صفحه ---
 async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
     try:
@@ -105,7 +147,6 @@ async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
 # --- 🚨 تابع هوشمند دانلود فایل 🚨 ---
 async def helper_download_file(client, msg_obj):
     errors = []
-    
     file_id = None
     file_obj = None
     for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
@@ -262,7 +303,13 @@ user_last_request_time = {}
 
 # --- ۱. پردازش چت متنی و چندرسانه‌ای ---
 async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=None):
+    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)
+
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
+    
     proc_msg = await send_with_keyboard(client, chat_id, "🧠 در حال پردازش...", False)
     history = user_states[chat_id].get("history", [])
     new_parts = []
@@ -321,6 +368,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)
+
     history.append({"role": "model", "parts": [{"text": final_answer}]})
     user_states[chat_id]["history"] = history
 
@@ -346,9 +397,16 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
     except Exception:
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد.", False)
 
+
 # --- ۲. پردازش ساخت عکس ---
 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)
+
     if not HF_TOKENS: return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
+    
     proc_msg = await send_with_keyboard(client, chat_id, "✨ در حال ترجمه و بهینه‌سازی پرامپت شما توسط جیمینای...\n(تبدیل به پرامپت حرفه‌ای)", False)
     enhanced_prompt = prompt 
     if GEMINI_KEYS:
@@ -400,6 +458,10 @@ 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)
+
     try:
         file_name = f"image_{random.randint(1000, 999999)}.jpg"
         rgb_im = generated_image.convert('RGB')
@@ -411,11 +473,16 @@ async def process_image(client, chat_id, prompt):
         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)
 
+
 # --- ۲.۵. پردازش ویرایش عکس با FLUX.2-dev ---
 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)
+
     if not HF_TOKENS:
-        await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
-        return
+        return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
 
     proc_msg = await send_with_keyboard(client, chat_id, "🪄 در حال پردازش و اعمال جادوی FLUX.2 روی عکس شما...\n(این فرآیند ممکن است کمی طول بکشد)", False)
     
@@ -427,7 +494,6 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     for token in keys_to_try:
         try:
             hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
-            # استفاده از متد image_to_image با ارسال دیتای بایت عکس و پرامپت متنی کاربر
             generated_image = await hf_client.image_to_image(
                 image_bytes,
                 prompt=prompt,
@@ -446,8 +512,11 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     except Exception: pass
 
     if not generated_image:
-        await send_with_keyboard(client, chat_id, f"❌ متأسفانه ویرایش عکس انجام نشد.\n\n⚠️ علت:\n{last_error_log[:200]}", True)
-        return
+        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)
 
     try:
         file_name = f"edited_flux_{random.randint(1000, 999999)}.jpg"
@@ -465,6 +534,11 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
 
 # --- ۳. پردازش ساخت صدا ---
 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)
+
     try:
         proc_msg = await send_with_keyboard(client, chat_id, f"⏳ در حال ساخت صدا با «{speaker_name}»...\n(لطفاً صبور باشید)", False)
         payload = {"text": user_text, "speaker": speaker_id, "temperature": 1.5, "prompt": "", "use_live_model": True}
@@ -495,6 +569,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)
+            
             file_name = f"audio_{random.randint(1000, 999999)}.mp3"
             with open(file_name, "wb") as f: f.write(audio_bytes)
             await asyncio.sleep(1)
@@ -504,8 +582,14 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
         else: await send_with_keyboard(client, chat_id, f"❌ سرورها درگیر هستند.\nدلیل: {last_error}", True)
     except Exception: traceback.print_exc()
 
+
 # --- ۳.۵. پردازش پادکست ---
 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)
+
     proc_msg = await send_with_keyboard(client, chat_id, "📻 در حال بررسی موضوع و نگارش سناریوی پادکست توسط هوش مصنوعی...\n(لطفاً صبور باشید)", False)
     available_speakers = []
     for num_key, (spk_name, spk_id) in SPEAKERS.items():
@@ -569,6 +653,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)
+
         file_name = f"final_podcast_{random.randint(1000, 999999)}.mp3"
         combined_audio.export(file_name, format="mp3")
         
@@ -581,12 +669,19 @@ async def process_podcast(client, chat_id, prompt):
 
         caption = f"🎧 پادکست شما با موفقیت میکس و ساخته شد!\n\n💡 موضوع شما: {prompt}"
         upload_result = await helper_upload_file(client, chat_id, file_name, "Voice", caption)
-        if upload_result is not True: await send_with_keyboard(client, chat_id, f"❌ پادکست با موفقیت ساخته شد اما روبیکا خطای آپلود داد.\n\n`{str(upload_result)[:800]}`", True)
+        if upload_result is not True: await send_with_keyboard(client, chat_id, f"❌ پادکست ساخته شد اما روبیکا خطای آپلود داد.\n\n`{str(upload_result)[:800]}`", True)
         if os.path.exists(file_name): os.remove(file_name)
 
+
 # --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن ---
 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)
+
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
+    
     proc_msg = await send_with_keyboard(client, chat_id, "📝 در حال گوش دادن و پیاده‌سازی متن...", False)
     base64_data = base64.b64encode(audio_bytes).decode('utf-8')
     mime_type, _ = mimetypes.guess_type(file_name)
@@ -616,12 +711,24 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
             if msg_id: await client.delete_messages(chat_id, [msg_id])
     except Exception: pass
 
-    if transcribed_text: await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
-    else: await send_with_keyboard(client, chat_id, f"❌ تبدیل فایل به متن ناموفق بود.", True)
+    if transcribed_text:
+        # کسر اعتبار
+        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)
+
 
 # --- ۵. پردازش تحلیل فایل مستقل ---
 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)
+
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلید جیمینای تنظیم نیست.", False)
+    
     proc_msg = await send_with_keyboard(client, chat_id, "👁️ در حال تحلیل فایل...", False)
     base64_data = base64.b64encode(file_bytes).decode('utf-8')
     mime_type, _ = mimetypes.guess_type(file_name)
@@ -650,8 +757,13 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
             if msg_id: await client.delete_messages(chat_id, [msg_id])
     except Exception: pass
 
-    if final_answer: await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
-    else: await send_with_keyboard(client, chat_id, "❌ پاسخی دریافت نشد.", True)
+    if final_answer: 
+        # کسر اعتبار
+        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)
 
 
 # --- تنظیمات ربات روبیکا ---
@@ -666,7 +778,6 @@ else:
             current_time = time.time()
             
             # 🛡 ۱. فیلتر پیام‌های قدیمی (Backlog) در زمان ری‌استارت اسپیس 🛡
-            # اگر از زمان روشن شدن ربات کمتر از 8 ثانیه گذشته، پیام را نادیده بگیر
             if current_time - BOT_STARTUP_TIME < 8.0:
                 return
             
@@ -695,9 +806,9 @@ else:
                 processed_message_ids.add(unique_msg_key)
                 if len(processed_message_ids) > 5000: processed_message_ids.clear()
             
-            # 🛡 سیستم محدودکننده سرعت کاربر (Rate Limit) برای همه پیام‌ها (حتی استارت) 🛡
+            # 🛡 سیستم محدودکننده سرعت کاربر (Rate Limit) 🛡
             last_req_time = user_last_request_time.get(chat_id, 0)
-            if current_time - last_req_time < 1.5:  # فاصله 1.5 ثانیه بین هر پیام الزامی است
+            if current_time - last_req_time < 1.5:
                 return
             user_last_request_time[chat_id] = current_time
 
@@ -734,21 +845,24 @@ else:
                 await send_with_keyboard(client, chat_id, "سلام! به ربات هوشمند آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:", True)
                 return
 
-            # --- حساب کاربری ---
+            # --- حساب کاربری پویا و دقیق ---
             if user_text_str in ["/account", "حساب کاربری 👤"]:
+                creds = get_user_credits(chat_id)
                 account_profile = f"""👤 **اطلاعات حساب کاربری شما**
 
 🔹 **شناسه یکتا:** `{chat_id}`
 🔹 **وضعیت اشتراک:** 🥉 نسخه رایگان (آزمایشی)
 
-📊 **سهمیه روزانه شما در نسخه رایگان:**
-- 💬 چت هوشمند: ۳۰ پیام
-- 🎨 ساخت عکس: ۵ تصویر
-- 🪄 ویرایش عکس پیشرفته: ۳ تصویر
-- 🎙 ساخت پادکست: ۱ برنامه
-- 🗣 تبدیل متن به صدا: ۵ فایل
+📊 **سهمیه باقی‌مانده شما در امروز:**
+- 💬 چت هوشمند: {creds['chat']} پیام
+- 🎨 ساخت عکس: {creds['image']} تصویر
+- 🪄 ویرایش عکس پیشرفته: {creds['edit_image']} تصویر
+- 🎙 ساخت پادکست: {creds['podcast']} برنامه
+- 🗣 تبدیل متن به صدا: {creds['tts']} فایل
+- 📁 تحلیل فایل: {creds['file']} فایل
+- 📝 تبدیل صدا به متن: {creds['stt']} فایل
 
-*نکته: در آینده با ارتقا به نسخه آلفا پرو می‌توانید تمام این محدودیت‌ها را بردارید.* 🚀"""
+*نکته: اعتبار شما هر روز ساعت ۰۰:۰۰ به صورت خودکار شارژ می‌شود. با ارتقا به نسخه آلفا پرو می‌توانید تمام این محدودیت‌ها را بردارید.* 🚀"""
                 await send_with_keyboard(client, chat_id, account_profile, True)
                 return
 
@@ -834,7 +948,6 @@ else:
                     user_states[chat_id]["text"] = user_text_str
                     user_states[chat_id]["mode"] = "tts_waiting_for_speaker"
                     
-                    # بازگرداندن متن کامل لیست گویندگان
                     speakers_menu = """✅ متن شما ذخیره شد.
 لطفاً **شماره** گوینده مورد نظر خود را بفرستید:
 1. شهاب | 2. آوا | 3. نوید
@@ -851,7 +964,6 @@ else:
                 return
 
             elif current_mode == "tts_waiting_for_speaker":
-                # تبدیل اعداد فارسی/عربی احتمالی به انگلیسی
                 normalized_text = to_english_digits(user_text_str)
                 
                 if normalized_text.isdigit() and normalized_text in SPEAKERS:
@@ -903,5 +1015,5 @@ else:
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
     if bot_token:
-        print("ربات آلفا روشن شد...")
+        print("ربات آلفا با موفقیت به همراه دیتابیس روشن شد...")
         bot.run()