Update main.py

main.py

@@ -2,23 +2,83 @@ import os
 import threading
 import random
 import aiohttp
+import requests
 import traceback
 import asyncio
-from flask import Flask
+from flask import Flask, request, Response
 from rubpy.bot import BotClient, filters
 
-# --- تنظیمات وب سرور ---
+# ====================================================================
+# 🌐 بخش اول: تنظیمات وب‌سرور FLASK و تونل (پروکسی) برای دور زدن فیلترینگ
+# ====================================================================
 app = Flask(__name__)
 
+HF_PODCAST_URL = "https://ezmarynoori-podgen.hf.space"
+HF_SADA_URL = "https://ezmarynoori-sada.hf.space"
+
 @app.route('/')
 def home():
-    return "ربات روبیکا (منوی هوشمند) روشن است! 🚀"
+    return "ربات روبیکا و سرور پراکسی Ai Sada روشن و فعال است! 🚀"
+
+# این روت تمام درخواست‌های سایت شما را می‌گیرد و به هاگینگ فیس می‌فرستد
+@app.route('/proxy/<target>/<path:subpath>', methods=['GET', 'POST'])
+def proxy_hub(target, subpath):
+    # تشخیص سرور هدف
+    if target == 'podcast':
+        base_url = HF_PODCAST_URL
+    elif target == 'sada':
+        base_url = HF_SADA_URL
+    else:
+        return "Invalid target", 400
+
+    url = f"{base_url}/{subpath}"
+    
+    # انتقال هدرهای امنیتی (فقط هدرهایی که خودمان برای امنیت تعریف کردیم)
+    headers = {
+        'X-App-Auth': request.headers.get('X-App-Auth', ''),
+        'X-CSRF-Token': request.headers.get('X-CSRF-Token', ''),
+    }
+    
+    # اگر درخواست JSON بود، هدر Content-Type را تنظیم می‌کنیم
+    if request.is_json:
+        headers['Content-Type'] = 'application/json'
+
+    # حذف هدرهای خالی
+    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)
+            else:
+                # درخواست های معمولی (JSON)
+                res = requests.post(url, data=request.get_data(), headers=headers)
+        else:
+            # درخواست های GET (مثل دانلود فایل)
+            res = requests.get(url, params=request.args, headers=headers)
+        
+        # برگرداندن پاسخ هاگینگ فیس مستقیما به سایت (بدون تغییر)
+        return Response(
+            res.content, 
+            status=res.status_code, 
+            content_type=res.headers.get('Content-Type', 'application/json')
+        )
+    except Exception as e:
+        print(f"Proxy Error: {e}")
+        return str(e), 500
 
 def run_flask():
-    app.run(host="0.0.0.0", port=7860)
+    # اجرای وب سرور در پس‌زمینه
+    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",
@@ -31,7 +91,7 @@ WORKER_URLS =[
     "https://hamed744-ttspro9.hf.space/generate",
 ]
 
-# --- دیکشنری ۳۰ گوینده ---
+# لیست گویندگان
 SPEAKERS = {
     "1": ("شهاب (مرد)", "Charon"), "2": ("آوا (زن)", "Zephyr"), "3": ("نوید (مرد)", "Achird"), 
     "4": ("آرمان (مرد)", "Zubenelgenubi"), "5": ("مهسا (زن)", "Vindemiatrix"), "6": ("دانا (مرد)", "Rasalgethi"),
@@ -48,7 +108,7 @@ SPEAKERS = {
 # دیکشنری برای ذخیره موقت متن کاربر
 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(لطفاً چند ثانیه صبور باشید)")
@@ -109,12 +169,10 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
         traceback.print_exc()
 
 
-# --- تنظیمات ربات روبیکا ---
+# راه‌اندازی ربات روبیکا
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
-if not bot_token:
-    print("خطا: توکن وارد نشده است!")
-else:
+if bot_token:
     bot = BotClient(bot_token)
 
     @bot.on_update(filters.private)
@@ -139,7 +197,7 @@ else:
             user_text_str = str(user_text).strip()
 
             # پیام خوشامدگویی
-            if user_text_str in ["/start", "سلام"]:
+            if user_text_str in["/start", "سلام"]:
                 welcome_text = "سلام! 🎙️\nمن ربات هوش مصنوعی تبدیل متن به صدا هستم.\n\nهر متنی که دوست داری رو برام بفرست تا با ده‌ها گوینده مختلف برات بخونمش!"
                 await update.reply(welcome_text)
                 return
@@ -186,12 +244,20 @@ else:
             await update.reply(menu_text)
 
         except Exception as e:
-            print(f"خطای کلی سیستم: {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("ربات تبدیل متن به صدا روشن شد...")
+        print("سیستم کامل شد! تونل Flask و ربات روبیکا همزمان روشن شدند...")
         bot.run()