Update main.py

main.py

@@ -3,87 +3,92 @@ import threading
 import random
 import aiohttp
 import requests
+import json
+import time
 import traceback
 import asyncio
-from flask import Flask, request, Response
+from flask import Flask
 from rubpy.bot import BotClient, filters
 
-# ====================================================================
-# 🌐 بخش اول: وب‌سرور تونل (Proxy) برای دور زدن قطعی اینترنت سایت
-# ====================================================================
-app = Flask(__name__)
+# 🔴🔴🔴 مسیر فایل PHP سایت خود را اینجا وارد کنید 🔴🔴🔴
+SITE_URL = "https://aisada.ir/tts/proxy.php"  
 
 HF_PODCAST_URL = "https://ezmarynoori-podgen.hf.space"
-HF_SADA_URL = "https://ezmarynoori-sada.hf.space"
+
+# =======================================================
+# 🌐 ۱. موتور مکنده (Reverse Tunnel) برای زنده کردن سایت
+# =======================================================
+def reverse_tunnel_worker():
+    while True:
+        try:
+            # ۱. گرفتن درخواست‌ها از سایت ایران
+            pull_res = requests.get(f"{SITE_URL}?endpoint=hf-pull", timeout=10)
+            if pull_res.status_code == 200:
+                tasks = pull_res.json()
+                for task in tasks:
+                    task_id = task.get('id')
+                    endpoint = task.get('endpoint')
+                    payload = task.get('payload', {})
+
+                    # تنظیم لینک مقصد (مدل‌های هوش مصنوعی)
+                    if endpoint == 'generate':
+                        target_url = f"{HF_PODCAST_URL}/api/generate"
+                    elif endpoint == 'check-tts-status':
+                        target_url = f"{HF_PODCAST_URL}/api/check_status"
+                    elif endpoint == 'create-full-podcast':
+                        target_url = f"{HF_PODCAST_URL}/api/create-full-podcast"
+                    elif endpoint == 'podcast-status':
+                        task_id_param = task.get('get_params', {}).get('task_id', '')
+                        target_url = f"{HF_PODCAST_URL}/api/podcast-status/{task_id_param}"
+                    else:
+                        continue
+
+                    # ۲. ارسال به مدل هوش مصنوعی (داخل هاگینگ فیس)
+                    headers = {'Content-Type': 'application/json'}
+                    if endpoint == 'podcast-status':
+                        model_res = requests.get(target_url, timeout=90)
+                    else:
+                        model_res = requests.post(target_url, json=payload, headers=headers, timeout=90)
+
+                    # ۳. ارسال فایل/جواب آماده شده به سایت ایران
+                    push_url = f"{SITE_URL}?endpoint=hf-push"
+                    if model_res.status_code == 200:
+                        content_type = model_res.headers.get('Content-Type', '')
+                        if 'audio' in content_type or model_res.content[:4] == b'RIFF':
+                            # ارسال فایل صدا
+                            files = {'file': (f"{task_id}.wav", model_res.content, 'audio/wav')}
+                            requests.post(push_url, data={'id': task_id}, files=files, timeout=30)
+                        else:
+                            # ارسال پاسخ‌های JSON
+                            requests.post(push_url, data={'id': task_id, 'json_data': model_res.content}, timeout=30)
+                    else:
+                        err = {'error': f"خطای هوش مصنوعی ({model_res.status_code})"}
+                        requests.post(push_url, data={'id': task_id, 'json_data': json.dumps(err)}, timeout=30)
+
+        except Exception as e:
+            # بی‌صدا بودن خطا برای جلوگیری از توقف حلقه
+            pass
+            
+        time.sleep(1) # بررسی سایت هر 1 ثانیه
+
+# =======================================================
+# 🌐 ۲. سرور زنده نگه‌دارنده (Flask)
+# =======================================================
+app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات روبیکا و سرور تونل Ai Sada فعال است! 🚀"
-
-@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':
-        base_url = HF_SADA_URL
-    else:
-        return "Invalid target", 400
-
-    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', '')
-    }
-    
-    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_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:
-                data = request.get_data()
-                res = requests.post(url, data=data, headers=headers, timeout=120)
-        else:
-            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: {str(e)}")
-        return Response(f'{{"message": "ارور تونل پایتون: {str(e)}"}}', status=500, mimetype='application/json')
+    return "✅ سرور نجات سایت و ربات روبیکا با موفقیت فعال است!"
 
 def run_flask():
-    app.run(host="0.0.0.0", port=7860, threaded=True)
+    # شروع بکار موتور مکنده در پس‌زمینه
+    threading.Thread(target=reverse_tunnel_worker, daemon=True).start()
+    app.run(host="0.0.0.0", port=7860)
 
 
-# ====================================================================
-# 🤖 بخش دوم: ربات روبیکا (تولید صدای مستقیم)
-# ====================================================================
+# =======================================================
+# 🤖 ۳. ربات روبیکا
+# =======================================================
 WORKER_URLS =[
     "https://hamed744-ttspro.hf.space/generate",
     "https://hamed744-ttspro2.hf.space/generate",
@@ -97,28 +102,20 @@ WORKER_URLS =[
 ]
 
 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")
+    "1": ("شهاب", "Charon"), "2": ("آوا", "Zephyr"), "3": ("نوید", "Achird"), 
+    "4": ("آرمان", "Zubenelgenubi"), "5": ("مهسا", "Vindemiatrix"), "6": ("دانا", "Rasalgethi"),
+    "7": ("سامان", "Sadachbia"), "8": ("آرش", "Sadaltager"), "9": ("شبنم", "Sulafat"),
+    "10": ("سحر", "Laomedeia")
 }
 
 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(لطفاً چند ثانیه صبور باشید)")
-
+        await client.send_message(chat_id, f"⏳ در حال ساخت صدا...\n(گوینده: {speaker_name})")
         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 = "هیچ پاسخی دریافت نشد"
 
         workers = WORKER_URLS.copy()
         random.shuffle(workers)
@@ -128,64 +125,39 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
                 try:
                     async with session.post(worker_url, json=payload, timeout=60) as response:
                         if response.status == 200:
-                            content_type = response.headers.get('Content-Type', '')
-                            if 'audio' in content_type or response.content_length > 1000:
-                                audio_bytes = await response.read()
-                                break
-                            else:
-                                last_error = "سرور به جای صدا، فایل نامعتبر داد."
-                        else:
-                            last_error = f"ارور سرور ({response.status})"
-                except Exception as e:
-                    last_error = str(e)
+                            audio_bytes = await response.read()
+                            break
+                except Exception:
                     continue
 
         if audio_bytes:
-            file_name = f"audio_{random.randint(1000, 999999)}.wav"
-            with open(file_name, "wb") as f:
-                f.write(audio_bytes)
-            
-            try:
-                await client.send_voice(chat_id, file_name)
-            except Exception:
-                try:
-                    await client.send_document(chat_id, file_name)
-                except Exception:
-                    await client.send_message(chat_id, "✅ صدا ساخته شد اما محدودیت ارسال فایل در روبیکا وجود دارد.")
-            
-            if os.path.exists(file_name):
-                os.remove(file_name)
+            file_name = f"voice_{random.randint(1000, 9999)}.wav"
+            with open(file_name, "wb") as f: f.write(audio_bytes)
+            try: await client.send_voice(chat_id, file_name)
+            except: await client.send_document(chat_id, file_name)
+            if os.path.exists(file_name): os.remove(file_name)
         else:
-            await client.send_message(chat_id, f"❌ خطای سرور:\n{last_error}\n\nلطفا چند ثانیه دیگر دوباره تلاش کنید.")
+            await client.send_message(chat_id, "❌ خطای سرور. لطفاً مجدد تلاش کنید.")
 
     except Exception as e:
-        print(f"خطای پردازش صدا: {e}")
         traceback.print_exc()
 
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
 if bot_token:
     bot = BotClient(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
-            elif hasattr(update, "message") and hasattr(update.message, "text") and update.message.text:
-                user_text = update.message.text
-            elif hasattr(update, "new_message") and hasattr(update.new_message, "text") and update.new_message.text:
-                user_text = update.new_message.text
-
+            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
             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", "سلام"]:
-                await update.reply("سلام! 🎙️\nمن ربات هوش مصنوعی تبدیل متن به صدا هستم.\nهر متنی دوست داری بفرست تا با ده‌ها گوینده مختلف برات بخونمش!")
+            if user_text_str in["/start", "سلام"]:
+                await update.reply("سلام! 🎙️\nهر متنی دوست داری بفرست تا با صداهای مختلف برات بخونمش.")
                 return
 
             if chat_id in user_states:
@@ -194,35 +166,13 @@ if bot_token:
                     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. شهاب (مرد) | 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)
+            await update.reply("📝 متن شما ذخیره شد!\nیک شماره بین 1 تا 10 برای انتخاب گوینده بفرستید.")
 
         except Exception as e:
             traceback.print_exc()
 
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
-    if bot_token:
-        print("ربات روبیکا و سرور تونل روشن شد...")
-        bot.run()
+    if bot_token: bot.run()