Update app.py

app.py

@@ -8,13 +8,19 @@ import time
 from datetime import date
 from fastapi import FastAPI, Request, BackgroundTasks
 from fastapi.responses import HTMLResponse, FileResponse, JSONResponse, RedirectResponse, StreamingResponse
+
+# 🟢 اضافه کردن این خط به هدر فایل جهت لود شدن درست وب‌سرور فلاسک
 from fastapi.middleware.wsgi import WSGIMiddleware
 
-# 🟢 ایمپورت و فراخوانی ماژول مستقل روبیکا بات و صوتی
+# ایمپورت و فراخوانی ماژول مستقل روبیکا بات و صوتی
+from rubikabot import router as rubikabot_router
+from soti import router as soti_router
+
+# ایمپورت و فراخوانی ماژول مستقل روبیکا بات و صوتی
 from rubikabot import router as rubikabot_router
 from soti import router as soti_router
 
-# 🟢 ایمپورت فایل جدید اکشن (فلاسک) که از روی اسپیس tts3 کپی کرده‌اید
+# ایمپورت فایل فلاسک که به اسم action.py ذخیره کرده‌اید
 import action
 
 app = FastAPI()
@@ -31,19 +37,7 @@ os.makedirs(USAGE_DIR, exist_ok=True)
 CLONE_USAGE_LIMIT = 1
 EDIT_USAGE_LIMIT = 5
 
-# 🟢 میان‌افزار هوشمند جهت تنظیم خودکار آدرس دامنه عمومی رانفلر شما در ماژول اکشن
-@app.middleware("http")
-async def set_space_url_middleware(request: Request, call_next):
-    if not getattr(action, "SPACE_HOST", None):
-        host = request.headers.get("host", "")
-        if host:
-            action.SPACE_HOST = host
-            action.SPACE_URL = f"https://{host}"
-            print(f"[Runflare Auto-Config] Set SPACE_URL dynamically to: {action.SPACE_URL}", flush=True)
-    return await call_next(request)
-
 def get_client_ip(request: Request) -> str:
-    """استخراج دقیق آی‌پی واقعی کاربر از پشت پروکسی‌های Cloudflare و هاست رانفلر"""
     cf_ip = request.headers.get("CF-Connecting-IP")
     if cf_ip:
         return cf_ip.strip()
@@ -54,7 +48,7 @@ def get_client_ip(request: Request) -> str:
         return request.client.host
     return "0.0.0.0"
 
-# --- بخش مدیریت دائمی اعتبار کلون صدا (ذخیره روی دیسک رانفلر) ---
+# --- بخش مدیریت دائمی اعتبار کلون صدا ---
 def get_local_clone_usage(key: str) -> dict:
     today_str = date.today().isoformat()
     safe_key = "".join([c for c in key if c.isalnum() or c in ["-", "_", "."]])
@@ -77,7 +71,6 @@ def save_local_clone_usage(key: str, data: dict):
     except: pass
 
 def check_local_clone_limit(fingerprint: str, ip: str) -> bool:
-    """بررسی محدودیت کلون صدا: اگر اثر انگشت یا آی‌پی سهمیه‌اش پر باشد، بلاک می‌کند"""
     for key in [fingerprint, ip]:
         if not key: continue
         data = get_local_clone_usage(key)
@@ -86,14 +79,13 @@ def check_local_clone_limit(fingerprint: str, ip: str) -> bool:
     return False
 
 def use_local_clone(fingerprint: str, ip: str):
-    """ثبت مصرف سهمیه کلون صدا روی هارد دیسک رانفلر"""
     for key in [fingerprint, ip]:
         if not key: continue
         data = get_local_clone_usage(key)
         data["count"] += 1
         save_local_clone_usage(key, data)
 
-# --- بخش مدیریت دائمی اعتبار ویرایش تصاویر (ذخیره روی دیسک رانفلر) ---
+# --- بخش مدیریت دائمی اعتبار ویرایش تصاویر ---
 def get_local_edit_usage(key: str) -> dict:
     current_week = date.today().isocalendar()[1]
     safe_key = "".join([c for c in key if c.isalnum() or c in ["-", "_", "."]])
@@ -116,7 +108,6 @@ def save_local_edit_usage(key: str, data: dict):
     except: pass
 
 def check_local_edit_limit(fingerprint: str, ip: str) -> bool:
-    """بررسی محدودیت ادیت عکس: اگر اثر انگشت یا آی‌پی سهمیه هفتگی‌اش پر باشد، بلاک می‌کند"""
     for key in [fingerprint, ip]:
         if not key: continue
         data = get_local_edit_usage(key)
@@ -125,7 +116,6 @@ def check_local_edit_limit(fingerprint: str, ip: str) -> bool:
     return False
 
 def use_local_edit(fingerprint: str, ip: str):
-    """ثبت مصرف سهمیه ویرایش عکس روی هارد دیسک رانفلر"""
     for key in [fingerprint, ip]:
         if not key: continue
         data = get_local_edit_usage(key)
@@ -134,18 +124,13 @@ def use_local_edit(fingerprint: str, ip: str):
 
 
 def clean_old_files():
-    """پاکسازی خودکار فایل‌های قدیمی برای جلوگیری از پر شدن حافظه سرور"""
     now = time.time()
-    
-    # فایل‌های صوتی و جاب‌ها بعد از ۳۰ دقیقه (۱۸۰۰ ثانیه) حذف می‌شوند
     try:
         for f in os.listdir(JOBS_DIR):
             fpath = os.path.join(JOBS_DIR, f)
             if os.path.isfile(fpath) and os.stat(fpath).st_mtime < now - 1800:
                 os.remove(fpath)
     except: pass
-    
-    # لاگ‌های اعتبار روزانه کاربران بعد از ۴۸ ساعت (۱۷۲۸۰۰ ثانیه) حذف می‌شوند
     try:
         for f in os.listdir(USAGE_DIR):
             fpath = os.path.join(USAGE_DIR, f)
@@ -154,9 +139,7 @@ def clean_old_files():
     except: pass
 
 def save_job_state(job_id: str, status: str, audio_data: bytes = None, error_message: str = None):
-    """ذخیره وضعیت کار در فایل متنی مشترک دیسک"""
     clean_old_files()
-
     state = {
         "status": status,
         "error_message": error_message,
@@ -172,7 +155,6 @@ def save_job_state(job_id: str, status: str, audio_data: bytes = None, error_mes
             f.write(audio_data)
 
 def get_job_state(job_id: str):
-    """خواندن وضعیت کار از فایل متنی مشترک دیسک"""
     state_file = os.path.join(JOBS_DIR, f"{job_id}.json")
     if not os.path.exists(state_file):
         return None
@@ -183,7 +165,6 @@ def get_job_state(job_id: str):
         return None
 
 def get_job_audio(job_id: str):
-    """خواندن دیتای باینری صوتی ذخیره شده روی دیسک"""
     audio_file = os.path.join(JOBS_DIR, f"{job_id}.mp3")
     if os.path.exists(audio_file):
         try:
@@ -419,20 +400,19 @@ def get_hf_headers(request: Request):
         headers["X-Forwarded-For"] = request.client.host
     return headers
 
-
-# =================================================================
-# 4. بازنشانی آدرس‌ها و وب‌هوک‌های محلی اکشنز (جایگزین sada8888-tts3)
-# =================================================================
+HF_PODCAST_SPACE_URL = "https://sada8888-tts2.hf.space"
 
 @app.post("/api/generate")
 async def submit_job(request: Request, background_tasks: BackgroundTasks):
     payload = await request.json()
     headers = get_hf_headers(request)
 
-    # 🟢 تشخیص هوشمند درخواست تولید عکس (ارسال مستقیم و محلی به فلاسکِ اکشن)
+    # 🟢 تشخیص هوشمند درخواست تولید عکس (ارسال به فلاسک اکشن به کمک WSGITransport)
     if "action_name" in payload and "prompt" in payload:
         try:
-            async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=120.0) as client:
+            # 🟢 ساخت ترنسپورت رسمی سازگار با آخرین نسخه httpx
+            transport = httpx.WSGITransport(app=action.app)
+            async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=120.0) as client:
                 resp = await client.post("/api/generate", json=payload, headers=headers)
                 return JSONResponse(resp.json(), status_code=resp.status_code)
         except Exception as e:
@@ -491,8 +471,9 @@ async def check_status(request: Request):
             else:
                 return JSONResponse({"status": "processing"})
                 
-        # در غیر این صورت، هدایت درخواست بررسی وضعیت به فلاسک داخلی (کلون صدا یا تصویر)
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=30.0) as client:
+        # هدایت درخواست بررسی وضعیت به فلاسک داخلی با WSGITransport
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=30.0) as client:
             resp = await client.post("/api/check_status", json=payload, headers=get_hf_headers(request))
             return JSONResponse(resp.json(), status_code=resp.status_code)
     except Exception as e: 
@@ -519,7 +500,6 @@ async def podcast_status_proxy(request: Request):
 # =================================================================
 # 5. سرویس استودیوی پادکست و مسیریابی هوشمند دیتابیس‌ها
 # =================================================================
-HF_PODCAST_SPACE_URL = "https://sada8888-tts2.hf.space"
 
 @app.get("/podcast", response_class=HTMLResponse)
 @app.get("/podcast/", response_class=HTMLResponse)
@@ -541,8 +521,9 @@ async def smart_check_credit(request: Request):
                 resp = await client.post(url, json=payload, headers=headers)
                 return JSONResponse(resp.json(), status_code=resp.status_code)
         else:
-            # 🟢 فراخوانی مستقیم و داخلی فلاسک رانفلر
-            async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=30.0) as client:
+            # 🟢 فراخوانی مستقیم و داخلی فلاسک رانفلر با WSGITransport
+            transport = httpx.WSGITransport(app=action.app)
+            async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=30.0) as client:
                 resp = await client.post("/api/check-credit", json=payload, headers=headers)
                 return JSONResponse(resp.json(), status_code=resp.status_code)
     except Exception as e: 
@@ -561,8 +542,9 @@ async def smart_use_credit(request: Request):
                 resp = await client.post(url, json=payload, headers=headers)
                 return JSONResponse(resp.json(), status_code=resp.status_code)
         else:
-            # 🟢 فراخوانی مستقیم و داخلی فلاسک رانفلر
-            async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=30.0) as client:
+            # 🟢 فراخوانی مستقیم و داخلی فلاسک رانفلر با WSGITransport
+            transport = httpx.WSGITransport(app=action.app)
+            async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=30.0) as client:
                 resp = await client.post("/api/use-credit", json=payload, headers=headers)
                 return JSONResponse(resp.json(), status_code=resp.status_code)
     except Exception as e:
@@ -656,7 +638,8 @@ async def proxy_check_image_credit(request: Request):
     try:
         body = await request.body()
         headers = get_hf_headers(request)
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=30.0) as client:
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=30.0) as client:
             resp = await client.post("/api/check-image-credit", content=body, headers=headers)
             return JSONResponse(resp.json(), status_code=resp.status_code)
     except Exception as e:
@@ -667,7 +650,8 @@ async def proxy_use_image_credit(request: Request):
     try:
         body = await request.body()
         headers = get_hf_headers(request)
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=30.0) as client:
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=30.0) as client:
             resp = await client.post("/api/use-image-credit", content=body, headers=headers)
             return JSONResponse(resp.json(), status_code=resp.status_code)
     except Exception as e:
@@ -685,12 +669,9 @@ async def proxy_check_edit_credit(request: Request):
         if subscription_status == 'paid':
             return JSONResponse({"credits_remaining": "unlimited", "limit_reached": False})
 
-        # بررسی دائمی قفل دوگانه (آی‌پی + اثر انگشت) از دیسک رانفلر
         limit_reached = check_local_edit_limit(fingerprint, client_ip)
-        
         current_week = date.today().isocalendar()[1]
         
-        # بررسی بدترین سناریوی سهمیه هفتگی دیسکی
         rem_fp = 0
         if fingerprint:
             rec = get_local_edit_usage(fingerprint)
@@ -730,14 +711,15 @@ async def proxy_use_edit_credit(request: Request):
     except Exception as e:
         return JSONResponse({"status": "error", "message": f"error: {str(e)}"}, status_code=500)
 
-# --- پروکسی درخواست‌های تولید و ویرایش به فلاسک داخلی رانفلر ---
+# --- پروکسی درخواست‌های تولید و ویرایش به فلاسک داخلی با WSGITransport ---
 @app.post("/api/generate-video")
 async def proxy_generate_video(request: Request):
     try:
         body = await request.body()
         headers = get_hf_headers(request)
         
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=300.0) as client:
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=300.0) as client:
             resp = await client.post("/api/generate-video", content=body, headers=headers)
             try:
                 return JSONResponse(resp.json(), status_code=resp.status_code)
@@ -752,7 +734,8 @@ async def proxy_merge_videos(request: Request):
         body = await request.body()
         headers = get_hf_headers(request)
         
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=400.0) as client:
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=400.0) as client:
             resp = await client.post("/api/merge-videos", content=body, headers=headers)
         
         if resp.status_code == 200:
@@ -774,7 +757,8 @@ async def proxy_merge_videos(request: Request):
 async def proxy_video_status(request: Request, run_id: str):
     try:
         headers = get_hf_headers(request)
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=30.0) as client:
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=30.0) as client:
             resp = await client.get(f"/api/status/{run_id}", headers=headers)
             return JSONResponse(resp.json(), status_code=resp.status_code)
     except Exception as e:
@@ -782,7 +766,6 @@ async def proxy_video_status(request: Request, run_id: str):
 
 @app.get("/static/images/{filename}")
 async def proxy_static_images(request: Request, filename: str):
-    """خوانش مستقیم فایل‌های ساخته شده گیت‌هاب اکشنز از روی دیسک محلی رانفلر بدون مصرف ترافیک شبکه"""
     local_path = os.path.join("static/images", filename)
     if os.path.exists(local_path):
         media_type = "application/octet-stream"
@@ -804,7 +787,8 @@ async def proxy_edit_image(request: Request):
         body = await request.body()
         headers = get_hf_headers(request)
         
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=120.0) as client:
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=120.0) as client:
             resp = await client.post("/api/edit", content=body, headers=headers)
             try:
                 return JSONResponse(resp.json(), status_code=resp.status_code)
@@ -819,7 +803,8 @@ async def proxy_gemma_chat(request: Request):
         body = await request.body()
         headers = get_hf_headers(request)
         
-        async with httpx.AsyncClient(app=action.app, base_url="http://local", timeout=120.0) as client:
+        transport = httpx.WSGITransport(app=action.app)
+        async with httpx.AsyncClient(transport=transport, base_url="http://local", timeout=120.0) as client:
             resp = await client.post("/api/chat", content=body, headers=headers)
             try:
                 return JSONResponse(resp.json(), status_code=resp.status_code)
@@ -833,5 +818,4 @@ async def proxy_gemma_chat(request: Request):
 # 7. یکپارچه‌سازی وب‌سرور داخلی فلاسک رانفلر و اجرای همزمان (WSGI)
 # =================================================================
 
-# 🟢 اتصال مستقیم هسته وب‌سرور فلاسک به یوویکورن رانفلر به عنوان نرم‌افزار مکمل اصلی
 app.mount("/", WSGIMiddleware(action.app))