Update main.py

main.py

@@ -9,7 +9,7 @@ from flask import Flask, request, Response
 from rubpy.bot import BotClient, filters
 
 # ====================================================================
-# 🌐 بخش اول: تنظیمات وب‌سرور FLASK و تونل (پروکسی) برای دور زدن فیلترینگ
+# 🌐 بخش اول: وب‌سرور تونل (Proxy) برای دور زدن قطعی اینترنت سایت
 # ====================================================================
 app = Flask(__name__)
 
@@ -18,12 +18,13 @@ HF_SADA_URL = "https://ezmarynoori-sada.hf.space"
 
 @app.route('/')
 def home():
-    return "ربات روبیکا و سرور پراکسی Ai Sada روشن و فعال است! 🚀"
+    return "ربات روبیکا و سرور تونل Ai Sada فعال است! 🚀"
 
-# این روت تمام درخواست‌های سایت شما را می‌گیرد و به هاگینگ فیس می‌فرستد
-@app.route('/proxy/<target>/<path:subpath>', methods=['GET', 'POST'])
+@app.route('/proxy/<target>/<path:subpath>', methods=['GET', 'POST', 'OPTIONS'])
 def proxy_hub(target, subpath):
-    # تشخیص سرور هدف
+    if request.method == 'OPTIONS':
+        return Response("", status=200)
+
     if target == 'podcast':
         base_url = HF_PODCAST_URL
     elif target == 'sada':
@@ -33,52 +34,56 @@ def proxy_hub(target, subpath):
 
     url = f"{base_url}/{subpath}"
     
-    # انتقال هدرهای امنیتی (فقط هدرهایی که خودمان برای امنیت تعریف کردیم)
+    # هدرهای دور زدن فیلترینگ و کلودفلر
     headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+        'Accept': '*/*',
         'X-App-Auth': request.headers.get('X-App-Auth', ''),
-        'X-CSRF-Token': request.headers.get('X-CSRF-Token', ''),
+        'X-CSRF-Token': request.headers.get('X-CSRF-Token', '')
     }
     
-    # اگر درخواست JSON بود، هدر Content-Type را تنظیم می‌کنیم
     if request.is_json:
         headers['Content-Type'] = 'application/json'
+    elif request.content_type and not request.files:
+        headers['Content-Type'] = request.content_type
 
     # حذف هدرهای خالی
     headers = {k: v for k, v in headers.items() if v}
 
     try:
         if request.method == 'POST':
-            # بررسی اینکه آیا فایل صوتی در درخواست آپلود شده است (مثل بخش کلون صدا)
             if request.files:
-                files = {k: (v.filename, v.stream, v.mimetype) for k, v in request.files.items()}
-                res = requests.post(url, data=request.form, files=files, headers=headers)
+                # تبدیل امن فایل‌های صوتی برای آپلود بدون قطعی استریم
+                files_dict = {}
+                for k, v in request.files.items():
+                    files_dict[k] = (v.filename, v.read(), v.content_type)
+                
+                form_data = dict(request.form)
+                res = requests.post(url, data=form_data, files=files_dict, headers=headers, timeout=120)
             else:
-                # درخواست های معمولی (JSON)
-                res = requests.post(url, data=request.get_data(), headers=headers)
+                data = request.get_data()
+                res = requests.post(url, data=data, headers=headers, timeout=120)
         else:
-            # درخواست های GET (مثل دانلود فایل)
-            res = requests.get(url, params=request.args, headers=headers)
+            res = requests.get(url, params=request.args, headers=headers, timeout=120)
         
-        # برگرداندن پاسخ هاگینگ فیس مستقیما به سایت (بدون تغییر)
         return Response(
             res.content, 
             status=res.status_code, 
             content_type=res.headers.get('Content-Type', 'application/json')
         )
+    except requests.exceptions.Timeout:
+        return Response('{"message": "ارتباط با سرور هوش مصنوعی قطع شد (تایم‌اوت)"}', status=504, mimetype='application/json')
     except Exception as e:
-        print(f"Proxy Error: {e}")
-        return str(e), 500
+        print(f"Proxy Error: {str(e)}")
+        return Response(f'{{"message": "ارور تونل پایتون: {str(e)}"}}', status=500, mimetype='application/json')
 
 def run_flask():
-    # اجرای وب سرور در پس‌زمینه
     app.run(host="0.0.0.0", port=7860, threaded=True)
 
 
 # ====================================================================
-# 🤖 بخش دوم: تنظیمات ربات روبیکا (تولید صدای مستقیم برای کاربران داخلی)
+# 🤖 بخش دوم: ربات روبیکا (تولید صدای مستقیم)
 # ====================================================================
-
-# سرورهای تولید صدای ربات
 WORKER_URLS =[
     "https://hamed744-ttspro.hf.space/generate",
     "https://hamed744-ttspro2.hf.space/generate",
@@ -91,7 +96,6 @@ WORKER_URLS =[
     "https://hamed744-ttspro9.hf.space/generate",
 ]
 
-# لیست گویندگان
 SPEAKERS = {
     "1": ("شهاب (مرد)", "Charon"), "2": ("آوا (زن)", "Zephyr"), "3": ("نوید (مرد)", "Achird"), 
     "4": ("آرمان (مرد)", "Zubenelgenubi"), "5": ("مهسا (زن)", "Vindemiatrix"), "6": ("دانا (مرد)", "Rasalgethi"),
@@ -105,22 +109,13 @@ SPEAKERS = {
     "28": ("مهتاب (زن)", "Kore"), "29": ("سام (مرد)", "Fenrir"), "30": ("لیدا (زن)", "Leda")
 }
 
-# دیکشنری برای ذخیره موقت متن کاربر
 user_states = {}
 
-# پردازش و ساخت صدا در پس‌زمینه (برای ربات)
 async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
     try:
         await client.send_message(chat_id, f"⏳ در حال ساخت صدا با گوینده «{speaker_name}»...\n(لطفاً چند ثانیه صبور باشید)")
 
-        payload = {
-            "text": user_text,
-            "speaker": speaker_id,
-            "temperature": 1.5,
-            "prompt": "",
-            "use_live_model": True
-        }
-        
+        payload = {"text": user_text, "speaker": speaker_id, "temperature": 1.5, "prompt": "", "use_live_model": True}
         headers = {"User-Agent": "Mozilla/5.0", "Content-Type": "application/json"}
         audio_bytes = None
         last_error = "هیچ پاسخی دریافت نشد"
@@ -129,7 +124,7 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
         random.shuffle(workers)
         
         async with aiohttp.ClientSession(headers=headers) as session:
-            for worker_url in workers[:3]: # حداکثر ۳ تلاش در سرورهای مختلف
+            for worker_url in workers[:3]:
                 try:
                     async with session.post(worker_url, json=payload, timeout=60) as response:
                         if response.status == 200:
@@ -156,20 +151,17 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
                 try:
                     await client.send_document(chat_id, file_name)
                 except Exception:
-                    await client.send_message(chat_id, "✅ صدا ساخته شد اما ربات محدودیت ارسال فایل دارد.")
+                    await client.send_message(chat_id, "✅ صدا ساخته شد اما محدودیت ارسال فایل در روبیکا وجود دارد.")
             
             if os.path.exists(file_name):
                 os.remove(file_name)
         else:
-            error_msg = f"❌ سرورها در حال حاضر خواب یا درگیر هستند.\n\n🔴 دلیل فنی:\n{last_error}\n\nلطفا چند ثانیه دیگر دوباره تلاش کنید."
-            await client.send_message(chat_id, error_msg)
+            await client.send_message(chat_id, f"❌ خطای سرور:\n{last_error}\n\nلطفا چند ثانیه دیگر دوباره تلاش کنید.")
 
     except Exception as e:
         print(f"خطای پردازش صدا: {e}")
         traceback.print_exc()
 
-
-# راه‌اندازی ربات روبیکا
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
 if bot_token:
@@ -178,7 +170,6 @@ if bot_token:
     @bot.on_update(filters.private)
     async def main_handler(client, update):
         try:
-            # استخراج متن پیام
             user_text = ""
             if hasattr(update, "text") and update.text:
                 user_text = update.text
@@ -187,48 +178,33 @@ if bot_token:
             elif hasattr(update, "new_message") and hasattr(update.new_message, "text") and update.new_message.text:
                 user_text = update.new_message.text
 
-            if not user_text:
-                return
-
+            if not user_text: 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
+            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)
+            if user_text_str in ["/start", "سلام"]:
+                await update.reply("سلام! 🎙️\nمن ربات هوش مصنوعی تبدیل متن به صدا هستم.\nهر متنی دوست داری بفرست تا با ده‌ها گوینده مختلف برات بخونمش!")
                 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]
-                    
-                    # ارسال پروسه ساخت صدا به پس‌زمینه
+                    saved_text = user_states.pop(chat_id)
                     asyncio.create_task(process_tts(client, chat_id, saved_text, speaker_id, speaker_name))
                     return
                 elif user_text_str.isdigit():
                     await update.reply("❌ شماره وارد شده نامعتبر است! لطفاً یک عدد بین ۱ تا ۳۰ بفرستید.")
                     return
 
-            # اگر پیام عدد نبود، یعنی کاربر یک متن جدید برای تبدیل فرستاده است
             if len(user_text_str) > 500:
                 await update.reply("⚠️ کاربر گرامی، لطفاً متنی کوتاه‌تر از ۵۰۰ کاراکتر بفرستید.")
                 return
 
-            # ذخیره متن در حافظه
             user_states[chat_id] = user_text_str
             
-            # ارسال منوی عددی گویندگان
-            menu_text = """📝 متن شما با موفقیت ذخیره شد!
-لطفاً فقط **شماره** گوینده مورد نظر خود را از لیست زیر بفرستید (مثلا بفرستید: 1):
+            menu_text = """📝 متن ذخیره شد! فقط **شماره** گوینده را از لیست زیر بفرستید:
 
 1. شهاب (مرد) | 2. آوا (زن) | 3. نوید (مرد)
 4. آرمان (مرد) | 5. مهسا (زن) | 6. دانا (مرد)
@@ -240,24 +216,13 @@ if bot_token:
 22. ترانه (زن) | 23. نیکو (زن) | 24. هستی (زن)
 25. کامیار (مرد)| 26. کیانوش (مرد)| 27. پویا (مرد)
 28. مهتاب (زن) | 29. سام (مرد) | 30. لیدا (زن)"""
-
             await update.reply(menu_text)
 
         except Exception as e:
-            print(f"خطای کلی ربات: {e}")
             traceback.print_exc()
-else:
-    print("خطا: توکن RUBIKA_AUTH تنظیم نشده است!")
-
 
-# ====================================================================
-# 🚀 اجرای همزمان Flask و Rubika Bot
-# ====================================================================
 if __name__ == "__main__":
-    # اجرای Flask در یک Thread مجزا
     threading.Thread(target=run_flask, daemon=True).start()
-    
-    # اجرای ربات روبیکا در Thread اصلی
     if bot_token:
-        print("سیستم کامل شد! تونل Flask و ربات روبیکا همزمان روشن شدند...")
+        print("ربات روبیکا و سرور تونل روشن شد...")
         bot.run()