Update main.py

main.py

@@ -6,14 +6,13 @@ import traceback
 import asyncio
 from flask import Flask
 from rubpy.bot import BotClient, filters
-from rubpy.types import InlineKeyboardMarkup, InlineKeyboardButton
 
 # --- تنظیمات وب سرور ---
 app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات روبیکا (نسخه دکمه‌های شیشه‌ای) روشن است! 🚀"
+    return "ربات روبیکا (منوی هوشمند) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -32,41 +31,23 @@ WORKER_URLS =[
     "https://hamed744-ttspro9.hf.space/generate",
 ]
 
-# --- لیست کامل ۳۰ گوینده ---
-SPEAKERS =[
-    ("شهاب (مرد)", "Charon"), ("آوا (زن)", "Zephyr"), ("نوید (مرد)", "Achird"), 
-    ("آرمان (مرد)", "Zubenelgenubi"), ("مهسا (زن)", "Vindemiatrix"), ("دانا (مرد)", "Rasalgethi"),
-    ("سامان (مرد)", "Sadachbia"), ("آرش (مرد)", "Sadaltager"), ("شبنم (زن)", "Sulafat"),
-    ("سحر (زن)", "Laomedeia"), ("مریم (زن)", "Achernar"), ("بهرام (مرد)", "Alnilam"),
-    ("نیکان (مرد)", "Schedar"), ("فرناز (زن)", "Gacrux"), ("سارا (زن)", "Pulcherrima"),
-    ("مانی (مرد)", "Umbriel"), ("آرتین (مرد)", "Algieba"), ("دلنواز (زن)", "Despina"),
-    ("روژان (زن)", "Erinome"), ("امید (مرد)", "Algenib"), ("بردیا (مرد)", "Orus"),
-    ("ترانه (زن)", "Aoede"), ("نیکو (زن)", "Callirrhoe"), ("هستی (زن)", "Autonoe"),
-    ("کامیار (مرد)", "Enceladus"), ("کیانوش (مرد)", "Iapetus"), ("پویا (مرد)", "Puck"),
-    ("مهتاب (زن)", "Kore"), ("سام (مرد)", "Fenrir"), ("لیدا (زن)", "Leda")
-]
+# --- دیکشنری ۳۰ گوینده ---
+SPEAKERS = {
+    "1": ("شهاب (مرد)", "Charon"), "2": ("آوا (زن)", "Zephyr"), "3": ("نوید (مرد)", "Achird"), 
+    "4": ("آرمان (مرد)", "Zubenelgenubi"), "5": ("مهسا (زن)", "Vindemiatrix"), "6": ("دانا (مرد)", "Rasalgethi"),
+    "7": ("سامان (مرد)", "Sadachbia"), "8": ("آرش (مرد)", "Sadaltager"), "9": ("شبنم (زن)", "Sulafat"),
+    "10": ("سحر (زن)", "Laomedeia"), "11": ("مریم (زن)", "Achernar"), "12": ("بهرام (مرد)", "Alnilam"),
+    "13": ("نیکان (مرد)", "Schedar"), "14": ("فرناز (زن)", "Gacrux"), "15": ("سارا (زن)", "Pulcherrima"),
+    "16": ("مانی (مرد)", "Umbriel"), "17": ("آرتین (مرد)", "Algieba"), "18": ("دلنواز (زن)", "Despina"),
+    "19": ("روژان (زن)", "Erinome"), "20": ("امید (مرد)", "Algenib"), "21": ("بردیا (مرد)", "Orus"),
+    "22": ("ترانه (زن)", "Aoede"), "23": ("نیکو (زن)", "Callirrhoe"), "24": ("هستی (زن)", "Autonoe"),
+    "25": ("کامیار (مرد)", "Enceladus"), "26": ("کیانوش (مرد)", "Iapetus"), "27": ("پویا (مرد)", "Puck"),
+    "28": ("مهتاب (زن)", "Kore"), "29": ("سام (مرد)", "Fenrir"), "30": ("لیدا (زن)", "Leda")
+}
 
 # دیکشنری برای ذخیره موقت متن کاربر
 user_states = {}
 
-# تابع ساخت دکمه‌های شیشه‌ای (دو ستونه)
-def get_speakers_keyboard():
-    keys = []
-    row =[]
-    for name, sp_id in SPEAKERS:
-        row.append(InlineKeyboardButton(text=name, callback_data=f"sp_{sp_id}"))
-        if len(row) == 2: # هر ردیف ۲ دکمه
-            keys.append(row)
-            row =[]
-    if row: # اضافه کردن دکمه‌های باقی‌مانده
-        keys.append(row)
-        
-    try:
-        return InlineKeyboardMarkup(inline_keyboard=keys)
-    except TypeError:
-        return InlineKeyboardMarkup(keys)
-
-
 # --- پردازش و ساخت صدا در پس‌زمینه ---
 async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
     try:
@@ -136,46 +117,10 @@ if not bot_token:
 else:
     bot = BotClient(bot_token)
 
-    # دریافت تمام آپدیت‌ها (هم پیام متنی و هم کلیک روی دکمه شیشه‌ای)
-    @bot.on_update()
+    @bot.on_update(filters.private)
     async def main_handler(client, update):
         try:
-            # ۱. بررسی کلیک روی دکمه شیشه‌ای (Callback Query)
-            call_data = getattr(update, "data", None)
-            if not call_data and hasattr(update, "callback_query"):
-                call_data = getattr(update.callback_query, "data", None)
-
-            if call_data and isinstance(call_data, str) and call_data.startswith("sp_"):
-                speaker_id = call_data.split("sp_")[1]
-                
-                # پیدا کردن اسم گوینده انتخاب شده
-                speaker_name = "ناشناس"
-                for name, sp in SPEAKERS:
-                    if sp == speaker_id:
-                        speaker_name = name
-                        break
-                        
-                # پیدا کردن آیدی کاربر
-                chat_id = getattr(update, "chat_id", None) or getattr(update, "author_guid", None) or getattr(update, "object_guid", None)
-                if not chat_id and hasattr(update, "callback_query"):
-                    chat_id = getattr(update.callback_query, "chat_id", None) or getattr(update.callback_query, "author_guid", None)
-                if not chat_id and hasattr(update, "message"):
-                    chat_id = getattr(update.message, "chat_id", None) or getattr(update.message, "author_guid", None)
-                    
-                if not chat_id:
-                    return
-                
-                # دریافت متنی که کاربر قبلاً فرستاده بود
-                user_text = user_states.get(chat_id)
-                if not user_text:
-                    await client.send_message(chat_id, "❌ متن شما پیدا نشد یا زمان زیادی از آن گذشته است. لطفاً متن خود را دوباره ارسال کنید.")
-                    return
-                
-                # ارسال پروسه ساخت صدا به پس‌زمینه (تا ربات هنگ نکند)
-                asyncio.create_task(process_tts(client, chat_id, user_text, speaker_id, speaker_name))
-                return
-
-            # ۲. بررسی دریافت پیام متنی
+            # استخراج متن پیام
             user_text = ""
             if hasattr(update, "text") and update.text:
                 user_text = update.text
@@ -184,34 +129,62 @@ else:
             elif hasattr(update, "new_message") and hasattr(update.new_message, "text") and update.new_message.text:
                 user_text = update.new_message.text
 
-            if user_text:
-                chat_id = getattr(update, "chat_id", None) or getattr(update, "author_guid", None) or getattr(update, "object_guid", None)
-                if not chat_id:
-                    return
+            if not user_text:
+                return
 
-                # پیام خوشامدگویی
-                if user_text == "/start" or user_text == "سلام":
-                    welcome_text = "سلام! 🎙️\nمن ربات هوش مصنوعی تبدیل متن به صدا هستم.\n\nهر متنی که دوست داری رو برام بفرست تا با گوینده‌های حرفه‌ای برات بخونمش!"
-                    await client.send_message(chat_id, welcome_text)
+            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 user_text_str in ["/start", "سلام"]:
+                welcome_text = "سلام! 🎙️\nمن ربات هوش مصنوعی تبدیل متن به صدا هستم.\n\nهر متنی که دوست داری رو برام بفرست تا با ده‌ها گوینده مختلف برات بخونمش!"
+                await update.reply(welcome_text)
+                return
+
+            # بررسی اینکه آیا کاربر در حال انتخاب گوینده است (ارسال عدد)
+            if chat_id in user_states:
+                if user_text_str.isdigit() and user_text_str in SPEAKERS:
+                    speaker_name, speaker_id = SPEAKERS[user_text_str]
+                    saved_text = user_states[chat_id]
+                    
+                    # پاک کردن متن از حافظه پس از انتخاب گوینده
+                    del user_states[chat_id]
+                    
+                    # ارسال پروسه ساخت صدا به پس‌زمینه
+                    asyncio.create_task(process_tts(client, chat_id, saved_text, speaker_id, speaker_name))
                     return
-                
-                # محدودیت کاراکتر
-                if len(user_text) > 500:
-                    await client.send_message(chat_id, "⚠️ کاربر گرامی، لطفاً متنی کوتاه‌تر از ۵۰۰ کاراکتر بفرستید.")
+                elif user_text_str.isdigit():
+                    await update.reply("❌ شماره وارد شده نامعتبر است! لطفاً یک عدد بین ۱ تا ۳۰ بفرستید.")
                     return
 
-                # ذخیره متن کاربر در حافظه ربات
-                user_states[chat_id] = user_text
-                
-                # ساخت و ارسال دکمه‌های شیشه‌ای گویندگان
-                kb = get_speakers_keyboard()
-                try:
-                    await client.send_message(chat_id, "🎙️ لطفاً از لیست زیر گوینده مورد نظر خود را برای خواندن این متن انتخاب کنید:", reply_markup=kb)
-                except Exception as e:
-                    print(f"خطا در ارسال دکمه: {e}")
-                    await client.send_message(chat_id, "خطا در نمایش دکمه‌ها. ربات ر��بیکا در حال حاضر از دکمه شیشه‌ای پشتیبانی نمی‌کند.")
+            # اگر پیام عدد نبود، یعنی کاربر یک متن جدید برای تبدیل فرستاده است
+            if len(user_text_str) > 500:
+                await update.reply("⚠️ کاربر گرامی، لطفاً متنی کوتاه‌تر از ۵۰۰ کاراکتر بفرستید.")
                 return
 
+            # ذخیره متن در حافظه
+            user_states[chat_id] = user_text_str
+            
+            # ارسال منوی عددی گویندگان
+            menu_text = """📝 متن شما با موفقیت ذخیره شد!
+لطفاً فقط **شماره** گوینده مورد نظر خود را از لیست زیر بفرستید (مثلا بفرستید: 1):
+
+1. شهاب (مرد) | 2. آوا (زن) | 3. نوید (مرد)
+4. آرمان (مرد) | 5. مهسا (زن) | 6. دانا (مرد)
+7. سامان (مرد) | 8. آرش (مرد) | 9. شبنم (زن)
+10. سحر (زن) | 11. مریم (زن) | 12. بهرام (مرد)
+13. نیکان (مرد)| 14. فرناز (زن) | 15. سارا (زن)
+16. مانی (مرد) | 17. آرتین (مرد) | 18. دلنواز (زن)
+19. روژان (زن) | 20. امید (مرد) | 21. بردیا (مرد)
+22. ترانه (زن) | 23. نیکو (زن) | 24. هستی (زن)
+25. کامیار (مرد)| 26. کیانوش (مرد)| 27. پویا (مرد)
+28. مهتاب (زن) | 29. سام (مرد) | 30. لیدا (زن)"""
+
+            await update.reply(menu_text)
+
         except Exception as e:
             print(f"خطای کلی سیستم: {e}")
             traceback.print_exc()