Update app.py

app.py

@@ -34,54 +34,55 @@ class TaskPayload(BaseModel):
 async def upload_to_wan(client: httpx.AsyncClient, file_url: str, job_id: str):
     """
     دانلود فایل از سرور مدیر و آپلود آن به سرور هوش مصنوعی
-    با سیستم تلاش مجدد (Retry) قدرتمند برای جلوگیری از خطا
+    با تایم‌اوت نامحدود و سیستم تلاش مجدد
     """
     filename = f"{job_id}_{uuid.uuid4().hex[:6]}_{file_url.split('/')[-1]}"
     file_bytes = None
 
     # 1. مرحله دانلود از سرور خودمان (Manager)
-    # سه بار تلاش می‌کند اگر دانلود نشد
+    # تایم‌اوت نامحدود (timeout=None)
     for attempt in range(3):
         try:
-            resp = await client.get(file_url, timeout=60.0)
+            resp = await client.get(file_url, timeout=None)
             if resp.status_code == 200:
                 file_bytes = resp.content
-                break # دانلود موفق بود، خروج از حلقه
+                break 
             else:
                 logger.warning(f"⚠️ [{job_id}] Download attempt {attempt+1} failed: {resp.status_code}")
         except Exception as e:
             logger.warning(f"⚠️ [{job_id}] Download error {attempt+1}: {e}")
-        await asyncio.sleep(2) # مکث کوتاه
+        await asyncio.sleep(2)
     
     if not file_bytes:
         raise Exception("Fatal Error: Could not download file from Manager.")
     
     # 2. مرحله آپلود به سرور هوش مصنوعی (Wan)
-    # پنج بار تلاش می‌کند چون سرور چینی نوسان دارد
+    # تایم‌اوت نامحدود (timeout=None)
     for attempt in range(5):
         try:
             files = {'files': (filename, file_bytes)}
-            # تایم‌اوت بالا برای آپلود (120 ثانیه)
-            wan_resp = await client.post(f"{WAN_API_BASE}/upload", files=files, timeout=120.0)
+            
+            # این خط مهمترین تغییر است: timeout=None برای آپلود فایل‌های سنگین
+            wan_resp = await client.post(f"{WAN_API_BASE}/upload", files=files, timeout=None)
             
             if wan_resp.status_code == 200:
-                return wan_resp.json()[0] # موفقیت! مسیر سرور را برگردان
+                return wan_resp.json()[0] 
             
             logger.warning(f"⚠️ [{job_id}] Upload attempt {attempt+1} failed. Status: {wan_resp.status_code}")
         except Exception as e:
             logger.warning(f"⚠️ [{job_id}] Upload connection error {attempt+1}: {e}")
         
-        await asyncio.sleep(3) # 3 ثانیه استراحت قبل از تلاش بعدی
+        await asyncio.sleep(3)
 
     raise Exception("Fatal Error: Failed to upload file to AI Server after 5 attempts.")
 
 async def process_single_job(payload: TaskPayload):
     """
-    تابع اصلی پردازش که به صورت کاملاً مستقل و ایزوله اجرا می‌شود.
+    تابع اصلی پردازش با زمان انتظار ۲۴ ساعته
     """
     logger.info(f"🟢 [START] Job: {payload.job_id}")
     
-    # استفاده از کلاینت اختصاصی برای هر درخواست (بدون تداخل)
+    # تنظیم تایم‌اوت کلی کلاینت روی None (نامحدود)
     async with httpx.AsyncClient(timeout=None, limits=HighPerformanceLimits) as client:
         try:
             # 1. انتقال فایل‌ها
@@ -99,7 +100,9 @@ async def process_single_job(payload: TaskPayload):
             }
             
             logger.info(f"🚀 [{payload.job_id}] Sending prediction request...")
-            predict_resp = await client.post(f"{WAN_API_BASE}/call/predict", json=req_data, timeout=60.0)
+            
+            # درخواست ساخت با تایم‌اوت نامحدود
+            predict_resp = await client.post(f"{WAN_API_BASE}/call/predict", json=req_data, timeout=None)
             
             if predict_resp.status_code != 200:
                 raise Exception(f"Prediction Request Failed: {predict_resp.text}")
@@ -108,16 +111,18 @@ async def process_single_job(payload: TaskPayload):
             logger.info(f"⏳ [{payload.job_id}] Queued. Event ID: {event_id}")
             
             # 3. حلقه انتظار هوشمند (Polling/Streaming Loop)
-            # تا 600 ثانیه (10 دقیقه) منتظر می‌ماند
             start_time = time.time()
             final_video_url = None
             
-            while time.time() - start_time < 600:
+            # انتظار تا ۲۴ ساعت (86400 ثانیه)
+            while time.time() - start_time < 86400:
                 try:
                     stream_url = f"{WAN_API_BASE}/call/predict/{event_id}"
                     
-                    # اتصال به استریم برای دریافت وضعیت زنده
-                    async with client.stream("GET", stream_url, headers={"Accept": "text/event-stream"}, timeout=40.0) as response:
+                    # اتصال به استریم
+                    # اینجا timeout=60 تنظیم شده تا اگر اتصال قطع شد، برنامه متوجه شود و دوباره وصل شود
+                    # این تایم‌اوت باعث کنسل شدن کار نمی‌شود، فقط اتصال را رفرش می‌کند
+                    async with client.stream("GET", stream_url, headers={"Accept": "text/event-stream"}, timeout=60.0) as response:
                         async for line in response.aiter_lines():
                             if not line.strip(): continue
                             
@@ -125,60 +130,53 @@ async def process_single_job(payload: TaskPayload):
                                 try:
                                     data = json.loads(line[6:])
                                     
-                                    # بررسی اینکه آیا ویدیو در پاسخ وجود دارد یا خیر
                                     if isinstance(data, list) and len(data) > 0:
                                         res = data[0]
                                         found_url = None
                                         
-                                        # استراتژی‌های مختلف یافتن لینک ویدیو در جیسون
+                                        # جستجوی لینک ویدیو در فرمت‌های مختلف
                                         if isinstance(res, dict):
-                                            # حالت ۱: داخل آبجکت video
                                             found_url = res.get("video", {}).get("url")
-                                            # حالت ۲: مستقیم در url
                                             if not found_url: found_url = res.get("url")
-                                            # حالت ۳: فقط نام فایل
                                             if not found_url and "name" in res: found_url = f"/file={res['name']}"
                                         
-                                        # حالت ۴: رشته مستقیم
                                         elif isinstance(res, str) and ("/file=" in res or ".mp4" in res):
                                             found_url = res
                                         
                                         if found_url:
                                             final_video_url = found_url
-                                            break # یافت شد! خروج از خواندن خطوط
+                                            break 
                                             
                                 except Exception:
-                                    pass # خطاهای ریز جیسون را نادیده بگیر
+                                    pass 
                         
                     if final_video_url:
-                        break # خروج از حلقه زمانی اصلی
+                        break 
                     
-                    # اگر استریم قطع شد ولی ویدیو نیامد، ۵ ثانیه صبر کن و دوباره وصل شو
                     await asyncio.sleep(5)
 
                 except Exception as e:
-                    # قطع شدن اینترنت کارگر مشکلی نیست، دوباره تلاش می‌کنیم
+                    # اگر اینترنت قطع شد یا خطایی رخ داد، ۵ ثانیه صبر کن و دوباره تلاش کن
+                    # چون حلقه اصلی ۸۶۴۰۰ ثانیه است، این خطاها باعث توقف نمی‌شوند
                     await asyncio.sleep(5)
 
             # 4. پایان و ارسال نتیجه
             if final_video_url:
-                # اصلاح لینک‌های نسبی به مطلق
                 if final_video_url.startswith("/"):
                     final_video_url = f"https://wan-ai-wan2-2-s2v.ms.show{final_video_url}"
                 
-                # اصلاح باگ دو اسلش در آدرس
                 final_video_url = final_video_url.replace("//file=", "/file=")
 
                 logger.info(f"✅ [DONE] Job: {payload.job_id} -> Video Ready")
                 
-                # ارسال به مدیر
+                # ارسال نتیجه به سرور مدیر
                 await client.post(payload.callback_url, json={
                     "job_id": payload.job_id,
                     "status": "COMPLETED",
                     "video_url": final_video_url
                 })
             else:
-                raise Exception("Timeout: Video processing took too long (>10 mins).")
+                raise Exception("Timeout: Video processing took too long (>24 hours).")
 
         except Exception as e:
             logger.error(f"❌ [FAIL] Job: {payload.job_id} -> {e}")
@@ -195,12 +193,11 @@ async def process_single_job(payload: TaskPayload):
 @app.post("/process")
 async def accept_task(payload: TaskPayload):
     """
-    این اندپوینت درخواست را می‌پذیرد و بلافاصله پاسخ OK می‌دهد.
-    پردازش اصلی در یک Task جداگانه و موازی شروع می‌شود.
+    دریافت درخواست و شروع پردازش در پس‌زمینه
     """
     asyncio.create_task(process_single_job(payload))
     return {"status": "accepted", "message": "Task started in background"}
 
 @app.get("/")
 async def root():
-    return {"status": "Worker Pro Ready", "mode": "High Concurrency"}
+    return {"status": "Worker Pro Ready", "mode": "Unlimited Timeout (24h)"}