Update main.py

main.py

@@ -85,7 +85,8 @@ def save_db(db_data):
 user_credits_db = load_db()
 
 def get_user_credits(chat_id):
-    str_chat_id = str(chat_id)
+    # اطمینان از پاک بودن استرینگ از علائم مزاحم
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     today_str = datetime.date.today().isoformat()
     
     if str_chat_id not in user_credits_db:
@@ -381,7 +382,7 @@ user_last_request_time = {}
 
 # --- ۱. پردازش چت متنی و چندرسانه‌ای ---
 async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=None):
-    str_chat_id = str(chat_id)
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["chat"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار پیام‌های چت شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
@@ -478,7 +479,7 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
 
 # --- ۲. پردازش ساخت عکس ---
 async def process_image(client, chat_id, prompt):
-    str_chat_id = str(chat_id)
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["image"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت عکس شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
@@ -554,7 +555,7 @@ async def process_image(client, chat_id, prompt):
 
 # --- ۲.۵. پردازش ویرایش عکس با FLUX.2-dev ---
 async def process_image_edit(client, chat_id, image_bytes, prompt):
-    str_chat_id = str(chat_id)
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["edit_image"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار ویرایش عکس شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
@@ -612,7 +613,7 @@ 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)
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["tts"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل متن به صدای شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
@@ -663,7 +664,7 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
 
 # --- ۳.۵. پردازش پادکست ---
 async def process_podcast(client, chat_id, prompt):
-    str_chat_id = str(chat_id)
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["podcast"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار ساخت پادکست شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
@@ -753,7 +754,7 @@ async def process_podcast(client, chat_id, prompt):
 
 # --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن ---
 async def process_stt(client, chat_id, audio_bytes, file_name):
-    str_chat_id = str(chat_id)
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["stt"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار تبدیل صدا به متن شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
@@ -800,7 +801,7 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
 
 # --- ۵. پردازش تحلیل فایل مستقل ---
 async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
-    str_chat_id = str(chat_id)
+    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["file"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار تحلیل فایل شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
@@ -910,10 +911,22 @@ else:
             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 len(parts) >= 2:
+                        # حذف علائم مزاحم از آیدی
+                        target_id = parts[1].replace("`", "").replace("'", "").replace('"', "").strip()
                         if target_id not in user_credits_db:
-                            user_credits_db[target_id] = {"is_premium": False}
+                            user_credits_db[target_id] = {
+                                "is_premium": False,
+                                "expire_date": None,
+                                "last_reset": "",
+                                "chat": 10,
+                                "image": 5,
+                                "edit_image": 1,
+                                "podcast": 2,
+                                "tts": 5,
+                                "file": 1,
+                                "stt": 5
+                            }
                         
                         user_credits_db[target_id]["is_premium"] = True
                         
@@ -932,14 +945,21 @@ else:
                         
                         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)
+                        except Exception as e:
+                            print(f"Failed to notify target user: {e}")
+                            
                     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 len(parts) >= 2:
+                        target_id = parts[1].replace("`", "").replace("'", "").replace('"', "").strip()
                         if target_id in user_credits_db:
                             user_credits_db[target_id]["is_premium"] = False
                             user_credits_db[target_id]["expire_date"] = None
@@ -947,6 +967,14 @@ else:
                             user_credits_db[target_id]["last_reset"] = "" 
                             save_db(user_credits_db)
                         await send_with_keyboard(client, chat_id, f"✅ اشتراک کاربر `{target_id}` لغو شد و به رایگان تبدیل گشت.", False)
+                        
+                        # ارسال پیام اطلاع‌رسانی لغو به کاربر هدف
+                        try:
+                            await send_with_keyboard(client, target_id, "⚠️ کاربر گرامی، اشتراک ویژه شما به پایان رسید و حساب شما به نسخه رایگان (آزمایشی) تغییر یافت.", True)
+                        except Exception:
+                            pass
+                    else:
+                        await send_with_keyboard(client, chat_id, "⚠️ فرمت صحیح: `/downgrade ID`", False)
                     return
 
             # تشخیص فایل