Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (Runflare Combined Server Core)
+# app.py (Hugging Face Combined Server Core)
 import os
 import asyncio
 import requests
@@ -9,8 +9,6 @@ import time
 from datetime import date
 from fastapi import FastAPI, Request, BackgroundTasks
 from fastapi.responses import HTMLResponse, FileResponse, JSONResponse, RedirectResponse, StreamingResponse
-
-# 🟢 ایمپورت رسمی WSGITransport و WSGIMiddleware جهت یکپارچه‌سازی وب‌سرور داخلی فلاسک
 from fastapi.middleware.wsgi import WSGIMiddleware
 
 # ایمپورت و فراخوانی ماژول مستقل روبیکا بات و صوتی
@@ -34,7 +32,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):
@@ -80,7 +78,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)
@@ -89,7 +86,6 @@ 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)
@@ -119,7 +115,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)
@@ -128,7 +123,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)
@@ -137,18 +131,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)
@@ -157,9 +146,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,
@@ -175,7 +162,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
@@ -186,7 +172,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:
@@ -204,7 +189,7 @@ async def run_tts_background(job_id: str, payload: dict, headers: dict):
             "temperature": payload.get("temperature", 1.5)
         }
         
-        async with httpx.AsyncClient(timeout=180.0, verify=False) as client:
+        async with httpx.AsyncClient(timeout=180.0) as client:
             resp = await client.post(
                 "https://sada8888-tts2.hf.space/api/generate",
                 json=target_payload,
@@ -241,7 +226,8 @@ async def serve_ttspro():
 # =================================================================
 # 2. آینه چت بات (پروکسی خام و بدون بافر برای سرعت حداکثری)
 # =================================================================
-CHAT_SPACE_URL = "https://sada8888-ttslive-chat.hf.space"
+# 🟢 بروزرسانی آدرس سرور چت‌بات به اسپیس فعال و جدید شما
+CHAT_SPACE_URL = "https://opera10-ttslive-chat.hf.space"
 
 @app.get("/chat", response_class=HTMLResponse)
 @app.get("/chat/", response_class=HTMLResponse)
@@ -252,7 +238,7 @@ async def serve_chat():
     return HTMLResponse("<h1>خطا: فایل chat.html پیدا نشد!</h1>", status_code=404)
 
 @app.get("/audio/{filename}")
-async def serve_chat_audio(request: Request, filename: str):
+async def serve_chat_audio(filename: str):
     if filename.startswith("standard_"):
         job_id = filename.replace("standard_", "").replace(".mp3", "").replace(".wav", "")
         audio_data = get_job_audio(job_id)
@@ -263,18 +249,7 @@ async def serve_chat_audio(request: Request, filename: str):
             )
         return HTMLResponse("<h1>فایل صوتی یافت نشد</h1>", status_code=404)
 
-    async def iterfile():
-        try:
-            # 🟢 افزودن verify=False برای پایداری دریافت صوت
-            async with httpx.AsyncClient(timeout=15.0, verify=False) as client:
-                async with client.stream("GET", f"{CHAT_SPACE_URL}/audio/{filename}", headers=get_hf_headers(request)) as r:
-                    r.raise_for_status()
-                    async for chunk in r.aiter_bytes(chunk_size=8192):
-                        if chunk: yield chunk
-        except Exception:
-            yield b""
-            
-    # 🟢 تعریف هوشمند پسوندها و Mime-Type جهت دانلود درست مقالات (PDF و Word) بدون تبدیل شدن به wav
+    # 🟢 تشخیص هوشمند نوع فایل برای پشتیبانی از دانلود فایل‌های متنی ساخته شده (PDF و Word)
     media_type = "audio/wav"
     if filename.endswith(".pdf"):
         media_type = "application/pdf"
@@ -282,7 +257,19 @@ async def serve_chat_audio(request: Request, filename: str):
         media_type = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
     elif filename.endswith(".mp3"):
         media_type = "audio/mpeg"
-        
+    elif filename.endswith(".mp4"):
+        media_type = "video/mp4"
+
+    async def iterfile():
+        try:
+            # افزایش تایم‌اوت به ۱۲۰ ثانیه برای دانلود فایل‌های بزرگتر بدون قطعی
+            async with httpx.AsyncClient(timeout=120.0) as client:
+                async with client.stream("GET", f"{CHAT_SPACE_URL}/audio/{filename}") as r:
+                    r.raise_for_status()
+                    async for chunk in r.aiter_bytes(chunk_size=65536):
+                        if chunk: yield chunk
+        except Exception:
+            yield b""
     return StreamingResponse(iterfile(), media_type=media_type)
 
 @app.post("/api/chat_proxy")
@@ -292,8 +279,7 @@ async def chat_proxy_bridge(request: Request):
 
         async def stream_generator():
             try:
-                # 🟢 افزودن verify=False برای دور زدن لایه امنیتی هدرها در داکر
-                async with httpx.AsyncClient(timeout=60.0, verify=False) as client:
+                async with httpx.AsyncClient(timeout=60.0) as client:
                     async with client.stream(
                         "POST", 
                         f"{CHAT_SPACE_URL}/api/chat_proxy",
@@ -317,39 +303,6 @@ async def chat_proxy_bridge(request: Request):
     except Exception as e:
         return JSONResponse({"status": "error", "message": "سرور موقتا در دسترس نیست."}, status_code=500)
 
-# 🟢 اندپوینت جدید: پروکسی استریمینگ و فوق‌العاده پایدار ساخت فایل (مقاله) از سرور چت اصلی با رفع خطای SSL
-@app.post("/api/create_file")
-async def create_file_proxy_bridge(request: Request):
-    try:
-        body_bytes = await request.body()
-
-        async def stream_generator():
-            try:
-                # 🟢 اتصال مستقیم با تنظیم تایم‌اوت ۵ دقیقه‌ای برای مقالات طولانی و دور زدن گواهی SSL
-                async with httpx.AsyncClient(verify=False, timeout=httpx.Timeout(120.0, read=300.0)) as client:
-                    async with client.stream(
-                        "POST", 
-                        f"{CHAT_SPACE_URL}/api/create_file",
-                        content=body_bytes,
-                        headers={"Content-Type": "application/json"}
-                    ) as resp:
-                        async for chunk in resp.aiter_bytes():
-                            if chunk:
-                                yield chunk
-            except Exception as e:
-                err = f"data: {json.dumps({'type': 'error', 'message': f'خطا در ارتباط رانفلر با سرور اصلی: {str(e)}'})}\n\n"
-                yield err.encode('utf-8')
-
-        headers = {
-            "Cache-Control": "no-cache, no-transform",
-            "X-Accel-Buffering": "no",
-            "Connection": "keep-alive",
-            "Transfer-Encoding": "chunked"
-        }
-        return StreamingResponse(stream_generator(), media_type="text/event-stream", headers=headers)
-    except Exception as e:
-        return JSONResponse({"status": "error", "message": "سرور موقتا در دسترس نیست."}, status_code=500)
-
 
 # =================================================================
 # 3. دور زدن فیلترینگ برای API های متن به صدا (TTS Proxy) با کش یکپارچه
@@ -899,6 +852,42 @@ async def proxy_gemma_chat(request: Request):
         return JSONResponse({"status": "error", "message": f"خطا در ارتباط رانفلر: {str(e)}"}, status_code=500)
 
 
+# =================================================================
+# 🟢 پروکسی جدید: استریمینگ و انتقال مستقیم ساخت مقاله (/api/create_file)
+# =================================================================
+@app.post("/api/create_file")
+async def proxy_create_file(request: Request):
+    try:
+        body_bytes = await request.body()
+        headers = get_hf_headers(request)
+        
+        # استریم صریح داده‌ها به صورت کاملاً غیرهمزمان و بدون بافر کردن در رم
+        async def stream_generator():
+            try:
+                async with httpx.AsyncClient(timeout=300.0) as client:
+                    async with client.stream(
+                        "POST",
+                        f"{CHAT_SPACE_URL}/api/create_file",
+                        content=body_bytes,
+                        headers={"Content-Type": "application/json"}
+                    ) as resp:
+                        async for chunk in resp.aiter_bytes():
+                            if chunk:
+                                yield chunk
+            except Exception as e:
+                yield f"data: {json.dumps({'type': 'error', 'message': f'خطای رانفلر: {str(e)}'})}\n\n".encode('utf-8')
+
+        headers_to_return = {
+            "Cache-Control": "no-cache, no-transform",
+            "Connection": "keep-alive",
+            "X-Accel-Buffering": "no",
+            "Content-Type": "text/event-stream"
+        }
+        return StreamingResponse(stream_generator(), media_type="text/event-stream", headers=headers_to_return)
+    except Exception as e:
+        return JSONResponse({"status": "error", "message": str(e)}, status_code=500)
+
+
 # =================================================================
 # 7. یکپارچه‌سازی وب‌سرور داخلی فلاسک رانفلر و اجرای همزمان (WSGI)
 # =================================================================