Update main.py

main.py

@@ -16,16 +16,36 @@ 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": "💬 چت"},
+                {"id": "img_btn", "type": "Simple", "button_text": "🎨 عکس"}
+            ]
+        },
+        {
+            "buttons": [
+                {"id": "tts_btn", "type": "Simple", "button_text": "🎙️ صدا"},
+                {"id": "cancel_btn", "type": "Simple", "button_text": "❌ لغو"}
+            ]
+        }
+    ]
+}
+
+# حالت کلاسیک (برای نسخه‌های جدید rubpy)
+KEYPAD_OBJ = None
 try:
-    from rubpy.models import Keypad, KeypadRow, Button
-    MAIN_KEYPAD = Keypad(
+    from rubpy.bot.models import Keypad, KeypadRow, Button
+    KEYPAD_OBJ = Keypad(
         rows=[
             KeypadRow(buttons=[
                 Button(id="chat_btn", type="Simple", button_text="💬 چت"),
@@ -38,21 +58,74 @@ try:
         ]
     )
 except ImportError:
-    MAIN_KEYPAD = None
-
-
-# --- تابع ارسال کیبورد ---
+    try:
+        from rubpy.models import Keypad, KeypadRow, Button
+        KEYPAD_OBJ = Keypad(
+            rows=[
+                KeypadRow(buttons=[
+                    Button(id="chat_btn", type="Simple", button_text="💬 چت"),
+                    Button(id="img_btn", type="Simple", button_text="🎨 عکس")
+                ]),
+                KeypadRow(buttons=[
+                    Button(id="tts_btn", type="Simple", button_text="🎙️ صدا"),
+                    Button(id="cancel_btn", type="Simple", button_text="❌ لغو")
+                ])
+            ]
+        )
+    except Exception:
+        pass
+
+
+# --- تابع ارسال کیبورد هوشمند (امتحان تمام متدها) ---
 async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
-    if not use_keyboard or not MAIN_KEYPAD:
+    if not use_keyboard:
         await client.send_message(chat_id, text)
         return
 
+    # لیست تمام قالب‌های ممکن برای ارسال
+    payloads = []
+    if KEYPAD_OBJ:
+        payloads.append(KEYPAD_OBJ)
+    payloads.append(MAIN_KEYPAD_DICT)
+
+    for payload in payloads:
+        # تست ১: کیبورد شیشه‌ای (قوی‌ترین روش روبیکا)
+        try:
+            await client.send_message(chat_id, text, inline_keypad=payload)
+            return
+        except Exception:
+            pass
+            
+        # تست ۲: کیبورد پایین صفحه
+        try:
+            await client.send_message(chat_id, text, chat_keypad=payload, chat_keypad_type="New")
+            return
+        except Exception:
+            pass
+            
+        # تست ۳: ریپلای مارکاپ عمومی
+        try:
+            await client.send_message(chat_id, text, reply_markup=payload)
+            return
+        except Exception:
+            pass
+
+    # اگر به هیچ وجه نسخه کتابخانه شما از کیبورد پشتیبانی نکرد!
+    await client.send_message(chat_id, text)
+
+
+# --- تابع کمکی برای استخراج آیدی کلیک ---
+def extract_button_id(update):
     try:
-        # استانداردترین پارامتر کتابخانه برای دکمه‌ها
-        await client.send_message(chat_id, text, reply_markup=MAIN_KEYPAD)
-    except Exception as e:
-        print(f"خطای نمایش کیبورد: {e}")
-        await client.send_message(chat_id, text)
+        if hasattr(update, 'aux_data') and update.aux_data:
+            return getattr(update.aux_data, 'button_id', None)
+        if hasattr(update, 'new_message') and update.new_message and hasattr(update.new_message, 'aux_data') and update.new_message.aux_data:
+            return getattr(update.new_message.aux_data, 'button_id', None)
+        if hasattr(update, 'message') and update.message and hasattr(update.message, 'aux_data') and update.message.aux_data:
+            return getattr(update.message.aux_data, 'button_id', None)
+    except Exception:
+        pass
+    return None
 
 
 # --- لیست تمام سرورهای تولید صدای شما ---
@@ -68,7 +141,6 @@ WORKER_URLS =[
     "https://hamed744-ttspro9.hf.space/generate",
 ]
 
-# --- دیکشنری ۳۰ گوینده ---
 SPEAKERS = {
     "1": ("شهاب (مرد)", "Charon"), "2": ("آوا (زن)", "Zephyr"), "3": ("نوید (مرد)", "Achird"), 
     "4": ("آرمان (مرد)", "Zubenelgenubi"), "5": ("مهسا (زن)", "Vindemiatrix"), "6": ("دانا (مرد)", "Rasalgethi"),
@@ -90,7 +162,6 @@ HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS = [k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 user_states = {}
-
 FOOTER_MENU = "\n\n➖➖➖➖➖➖\n👇 از دکمه‌های پایین برای تغییر بخش استفاده کنید."
 
 
@@ -158,11 +229,9 @@ async def process_gemini(client, chat_id, prompt):
                 await send_with_keyboard(client, chat_id, chunk, use_kb)
                 await asyncio.sleep(2.5) 
             except Exception as send_err:
-                print(f"خطا در ارسال قطعه {idx}: {send_err}")
                 await asyncio.sleep(2.5)
                 
     except Exception as e:
-        print(f"خطای سیستم پردازش متن جیمینای: {e}")
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد." + FOOTER_MENU, True)
 
 
@@ -197,8 +266,8 @@ async def process_image(client, chat_id, prompt):
     try:
         file_name = f"image_{random.randint(1000, 999999)}.jpg"
         rgb_im = generated_image.convert('RGB')
-        
         rgb_im.save(file_name, format="JPEG", quality=100)
+        
         await asyncio.sleep(1)
         caption_text = "🎨 تصویر شما آماده شد!\n(مدل: Z-Image-Turbo)" + FOOTER_MENU
         
@@ -243,7 +312,6 @@ async def process_image(client, chat_id, prompt):
             os.remove(file_name)
             
     except Exception as e:
-        print(f"خطای پردازش تصویر: {e}")
         await send_with_keyboard(client, chat_id, "❌ خطایی در فرآیند ذخیره عکس رخ داد." + FOOTER_MENU, True)
 
 
@@ -337,23 +405,41 @@ else:
     @bot.on_update(filters.private)
     async def main_handler(client, update):
         try:
+            # ۱. استخراج متن پیام کاربر
             user_text = getattr(update, "text", "") or getattr(getattr(update, "message", None), "text", "") or getattr(getattr(update, "new_message", None), "text", "")
-            if not user_text: return
+            
+            # ۲. استخراج آیدی دکمه در صورتی که کاربر روی کیبورد شیشه‌ای کلیک کرده باشد
+            button_id = extract_button_id(update)
+            
+            user_text_str = str(user_text).strip() if user_text else ""
+
+            # ۳. اگر دکمه فشرده شد، متن به طور خودکار تنظیم شود!
+            if button_id == "chat_btn":
+                user_text_str = "/chat"
+            elif button_id == "img_btn":
+                user_text_str = "/image"
+            elif button_id == "tts_btn":
+                user_text_str = "/tts"
+            elif button_id == "cancel_btn":
+                user_text_str = "/cancel"
+
+            # جلوگیری از پردازش رویدادهای خالی
+            if not user_text_str and not button_id: 
+                return
 
             chat_id = getattr(update, "chat_id", None) or getattr(update, "author_guid", None) or getattr(update, "object_guid", None)
-            if not chat_id: return
-
-            user_text_str = str(user_text).strip()
+            if not chat_id: 
+                return
 
             if chat_id not in user_states:
                 user_states[chat_id] = {"mode": None, "text": ""}
 
-            # مدیریت کیبورد و کامندهای لغو/شروع
+            # مدیریت کامندهای لغو/شروع
             if user_text_str in ["/start", "سلام", "لغو", "/cancel", "❌ لغو"]:
                 user_states[chat_id] = {"mode": None, "text": ""}
                 menu = (
                     "سلام! به ربات هوشمند آلفا خوش آمدید 🤖\n\n"
-                    "لطفاً برای شروع، از دکمه‌های پایین یکی از بخش‌ها را انتخاب کنید:\n\n"
+                    "لطفاً برای شروع، از دکمه‌های شیشه‌ای زیر یکی از بخش‌ها را انتخاب کنید:\n\n"
                     "💬 چت با هوش مصنوعی\n"
                     "🎨 طراحی عکس\n"
                     "🎙️ تبدیل متن به صدا"
@@ -428,7 +514,6 @@ else:
                 return
 
         except Exception as e:
-            print(f"خطای کلی سیستم: {e}")
             traceback.print_exc()
 
 if __name__ == "__main__":