Update main.py

main.py

@@ -21,46 +21,46 @@ 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":[
+    "rows": [
         # ردیف اول
         {
-            "buttons":[
-                {"id": "account_btn", "type": "Simple", "button_text": "حساب کاربری 👤"},
+            "buttons": [
                 {"id": "chat_btn", "type": "Simple", "button_text": "چت با هوش مصنوعی 🤖"}
             ]
         },
         # ردیف دوم
         {
-            "buttons":[
-                {"id": "edit_img_btn", "type": "Simple", "button_text": "ویرایش تصاویر 🪄"},
-                {"id": "img_btn", "type": "Simple", "button_text": "ساخت تصاویر🎨"}
+            "buttons": [
+                {"id": "img_btn", "type": "Simple", "button_text": "ساخت تصاویر🎨"},
+                {"id": "edit_img_btn", "type": "Simple", "button_text": "ویرایش تصاویر 🪄"}
             ]
         },
         # ردیف سوم
         {
-            "buttons":[
+            "buttons": [
                 {"id": "podcast_btn", "type": "Simple", "button_text": "ساخت پادکست 🎙️"},
                 {"id": "tts_btn", "type": "Simple", "button_text": "تبدیل متن به صدا🗣️"}
             ]
         },
         # ردیف چهارم
         {
-            "buttons":[
+            "buttons": [
                 {"id": "stt_btn", "type": "Simple", "button_text": "فایل صوتی به متن 📝"},
                 {"id": "file_btn", "type": "Simple", "button_text": "تحلیل فایل 📁"}
             ]
         },
         # ردیف پنجم
         {
-            "buttons":[
+            "buttons": [
+                {"id": "account_btn", "type": "Simple", "button_text": "حساب کاربری 👤"},
                 {"id": "cancel_btn", "type": "Simple", "button_text": "برگشت♻️"}
             ]
         }
@@ -247,7 +247,7 @@ processed_message_ids = set()
 user_last_request_time = {}
 
 
-# --- بخش جدید: پردازش ویرایش عکس با FLUX.2-dev ---
+# --- بخش پردازش ویرایش عکس با FLUX.2-dev ---
 async def process_image_edit(client, chat_id, image_bytes, prompt):
     if not HF_TOKENS:
         await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
@@ -299,7 +299,7 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
         await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره عکس ویرایش شده:\n{str(e)[:150]}", True)
 
 
-# --- پردازش‌های قبلی (چت، عکس، صدا، پادکست، استخراج متن و تحلیل) بدون تغییر ---
+# --- پردازش‌های قبلی ---
 async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=None):
     if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
     proc_msg = await send_with_keyboard(client, chat_id, "🧠 در حال پردازش...", False)
@@ -638,26 +638,42 @@ else:
     async def main_handler(client, update):
         try:
             current_time = time.time()
+            
+            # --- 🛡 استخراج دقیق شناسه چت 🛡 ---
+            chat_id = None
+            if hasattr(update, 'object_guid') and update.object_guid:
+                chat_id = update.object_guid
+            elif hasattr(update, 'author_guid') and update.author_guid:
+                chat_id = update.author_guid
+            else:
+                chat_id = getattr(update, "chat_id", None)
+                
+            if not chat_id: return
+            
             msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
+            
+            # استخراج آیدی پیام
             msg_id = getattr(update, "message_id", None)
             if not msg_id and msg_obj:
                 msg_id = msg_obj.get("message_id") if isinstance(msg_obj, dict) else getattr(msg_obj, "message_id", None)
             
-            if msg_id:
-                if msg_id in processed_message_ids: return
-                processed_message_ids.add(msg_id)
+            # 🛡 کلید ترکیبی ضد تداخل: آیدی چت + آیدی پیام 🛡
+            unique_msg_key = f"{chat_id}_{msg_id}" if msg_id else None
+            if unique_msg_key:
+                if unique_msg_key in processed_message_ids: return
+                processed_message_ids.add(unique_msg_key)
                 if len(processed_message_ids) > 5000: processed_message_ids.clear()
             
-            chat_id = getattr(update, "chat_id", None) or getattr(update, "author_guid", None) or getattr(update, "object_guid", None)
-            if not chat_id: return
-
-            last_req_time = user_last_request_time.get(chat_id, 0)
-            if current_time - last_req_time < 1.0: return
-            user_last_request_time[chat_id] = current_time
-
+            # سیستم محدودکننده سرعت کاربر (در صورتی که فرمان start/ نباشد)
             user_text = getattr(update, "text", "") or getattr(msg_obj, "text", "")
             user_text_str = str(user_text).strip() if user_text else ""
             
+            if user_text_str not in ["/start", "سلام"]:
+                last_req_time = user_last_request_time.get(chat_id, 0)
+                if current_time - last_req_time < 1.0: return
+            user_last_request_time[chat_id] = current_time
+
+            # تشخیص فایل
             is_file = False
             file_name = "unknown_file.jpg"
             
@@ -683,15 +699,13 @@ else:
 
             if user_text_str in ["/start", "سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
                 user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
-                await send_with_keyboard(client, chat_id, "سلام! به ربات هوشمند آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، یکی از بخش‌های زیر را انتخاب کنید:", True)
+                await send_with_keyboard(client, chat_id, "سلام کاربر عزیز! به ربات هوشمند آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، یکی از بخش‌های زیر را انتخاب کنید:", True)
                 return
 
-            # --- بخش جدید: حساب کاربری ---
+            # --- شخصی‌سازی حساب کاربری با شناسه چت (داینامیک) ---
             if user_text_str in ["/account", "حساب کاربری 👤"]:
-                # متن پیشنهادی برای محدودیت‌های نسخه رایگان
-                account_profile = f"""👤 **اطلاعات حساب کاربری**
+                account_profile = f"""👤 **اطلاعات حساب کاربری شما**
 
-🔹 **نام:** حامد نوری
 🔹 **شناسه یکتا:** `{chat_id}`
 🔹 **وضعیت اشتراک:** 🥉 نسخه رایگان (آزمایشی)
 
@@ -716,7 +730,6 @@ else:
                 await send_with_keyboard(client, chat_id, "🎨 وارد بخش **ساخت عکس** شدید.\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}
                 await send_with_keyboard(client, chat_id, "🪄 به بخش **ویرایش عکس (Flux.2)** خوش آمدید.\n\nلطفاً ابتدا عکسی که می‌خواهید ویرایش کنید را بفرستید:", True)
@@ -763,7 +776,6 @@ else:
                 if user_text_str: asyncio.create_task(process_image(client, chat_id, user_text_str))
                 return
 
-            # --- مسیریابی بخش ویرایش تصویر ---
             elif current_mode == "image_edit_waiting_for_image":
                 if not is_file: return await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا یک عکس ارسال کنید.", False)
                 await send_with_keyboard(client, chat_id, "📥 در حال دانلود عکس...", False)