Update app.py

app.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-import os, requests, base64, uuid, time, threading
+import os, requests, base64, uuid, threading
 
 app = FastAPI()
 
@@ -33,8 +33,11 @@ def call_model(url, payload, timeout=60):
         raise Exception(resp.text)
     data = resp.json()
 
+    # TEXT-TO-VIDEO zwraca listę
     if isinstance(data, list):
         return data[0].get("video")
+
+    # IMAGE / AVATAR zwracają dict
     return data.get("video")
 
 
@@ -101,4 +104,3 @@ def status(task_id: str):
     if task_id not in TASKS:
         raise HTTPException(status_code=404, detail="Task not found")
     return TASKS[task_id]
-