Create app.py

app.py

@@ -0,0 +1,121 @@
+import asyncio
+import json
+import httpx
+from fastapi import FastAPI, BackgroundTasks
+from pydantic import BaseModel
+
+app = FastAPI()
+
+# آدرس اصلی API مدل Wan2.2
+WAN_API_BASE = "https://wan-ai-wan2-2-s2v.ms.show/gradio_api"
+
+class TaskPayload(BaseModel):
+    job_id: str
+    image_url: str
+    audio_url: str
+    resolution: str
+    callback_url: str
+
+async def upload_file_from_url(client: httpx.AsyncClient, file_url: str, file_type: str):
+    """دانلود فایل از مدیر و آپلود به Wan2.2"""
+    print(f"Downloading {file_type} from {file_url}...")
+    # 1. دانلود فایل از اسپیس مدیر
+    file_resp = await client.get(file_url)
+    if file_resp.status_code != 200:
+        raise Exception(f"Failed to download {file_type}: {file_resp.status_code}")
+    
+    file_bytes = file_resp.content
+    filename = file_url.split("/")[-1]
+    
+    # 2. آپلود به Wan2.2
+    print(f"Uploading {filename} to Wan2.2...")
+    files = {'files': (filename, file_bytes)}
+    upload_resp = await client.post(f"{WAN_API_BASE}/upload", files=files)
+    
+    if upload_resp.status_code != 200:
+        raise Exception(f"Failed to upload {file_type} to Wan2.2")
+    
+    remote_path = upload_resp.json()[0]
+    print(f"Uploaded: {remote_path}")
+    return remote_path
+
+async def process_task(payload: TaskPayload):
+    print(f"Processing Job {payload.job_id}")
+    
+    async with httpx.AsyncClient(timeout=120.0) as client: # تایم‌اوت بالا
+        try:
+            # الف) آپلود فایل‌ها
+            img_remote = await upload_file_from_url(client, payload.image_url, "image")
+            aud_remote = await upload_file_from_url(client, payload.audio_url, "audio")
+            
+            # ب) ارسال درخواست Predict
+            req_data = {
+                "data": [
+                    {"path": img_remote, "meta": {"_type": "gradio.FileData"}},
+                    {"path": aud_remote, "meta": {"_type": "gradio.FileData"}},
+                    payload.resolution
+                ]
+            }
+            
+            predict_resp = await client.post(f"{WAN_API_BASE}/call/predict", json=req_data)
+            if predict_resp.status_code != 200:
+                raise Exception(f"Predict call failed: {predict_resp.text}")
+            
+            event_id = predict_resp.json().get("event_id")
+            print(f"Event ID: {event_id}")
+            
+            # ج) گوش دادن به استریم (Polling the stream)
+            # در پایتون، به جای EventSource، می‌توانیم با یک لوپ ساده GET بزنیم یا stream کنیم
+            final_video_url = None
+            
+            async with client.stream("GET", f"{WAN_API_BASE}/call/predict/{event_id}", headers={"Accept": "text/event-stream"}) as response:
+                async for line in response.aiter_lines():
+                    if line.startswith("data: "):
+                        try:
+                            data = json.loads(line[6:])
+                            # بررسی ساختار خروجی
+                            if isinstance(data, list) and len(data) > 0:
+                                result_item = data[0]
+                                # پیدا کردن URL در ساختارهای مختلف
+                                if isinstance(result_item, dict):
+                                    if "video" in result_item and "url" in result_item["video"]:
+                                        final_video_url = result_item["video"]["url"]
+                                    elif "url" in result_item:
+                                        final_video_url = result_item["url"]
+                                    elif "name" in result_item:
+                                        final_video_url = f"https://wan-ai-wan2-2-s2v.ms.show/file={result_item['name']}"
+                                elif isinstance(result_item, str) and (result_item.endswith(".mp4") or "/file=" in result_item):
+                                     final_video_url = result_item
+                                
+                                if final_video_url:
+                                    # اصلاح لینک نسبی
+                                    if final_video_url.startswith("/"):
+                                        final_video_url = f"https://wan-ai-wan2-2-s2v.ms.show{final_video_url}"
+                                    break
+                        except:
+                            pass
+
+            if final_video_url:
+                print(f"Success! Video: {final_video_url}")
+                # د) بازگرداندن نتیجه به مدیر
+                await client.post(payload.callback_url, json={
+                    "job_id": payload.job_id,
+                    "status": "COMPLETED",
+                    "video_url": final_video_url
+                })
+            else:
+                raise Exception("Stream ended without video URL")
+
+        except Exception as e:
+            print(f"Task Failed: {e}")
+            await client.post(payload.callback_url, json={
+                "job_id": payload.job_id,
+                "status": "FAILED",
+                "message": str(e)
+            })
+
+@app.post("/process")
+async def accept_task(payload: TaskPayload, background_tasks: BackgroundTasks):
+    """تسک را می‌پذیرد و بلافاصله OK برمی‌گرداند تا تایم‌اوت نشود"""
+    background_tasks.add_task(process_task, payload)
+    return {"status": "accepted"}