www

app.py

@@ -54,7 +54,7 @@ class WanAnimateApp:
         
         # Make the initial API request
         url = self.url
-        response = requests.post(url, json=payload, headers=headers, timeout=60)
+        response = requests.post(url, json=payload, headers=headers)
         
         # Check if request was successful
         if response.status_code != 200:
@@ -74,7 +74,7 @@ class WanAnimateApp:
         }
         
         while True:
-            response = requests.get(get_url, headers=headers, timeout=60)
+            response = requests.get(get_url, headers=headers)
             if response.status_code != 200:
                 raise Exception(f"Failed to get task status: {response.status_code}: {response.text}")
             
@@ -86,16 +86,16 @@ class WanAnimateApp:
                 # Task completed successfully, return video URL
                 video_url = result["output"]["results"]["video_url"]
                 return video_url, "SUCCEEDED"
-            elif task_status == "PENDING" or task_status == "RUNNING":
-                # Task is still running, wait and retry
-                time.sleep(10)  # Wait 10 seconds before polling again
-            else:
-                # Task failed or unknown, raise an exception with error message
+            elif task_status == "FAILED":
+                # Task failed, raise an exception with error message
                 error_msg = result.get("output", {}).get("message", "Unknown error")
                 code_msg = result.get("output", {}).get("code", "Unknown code")
                 print(f"\n\nTask failed: {error_msg} Code: {code_msg} TaskId: {task_id}\n\n")
                 return None, f"Task failed: {error_msg} Code: {code_msg} TaskId: {task_id}"
                 # raise Exception(f"Task failed: {error_msg} TaskId: {task_id}")
+            else:
+                # Task is still running, wait and retry
+                time.sleep(5)  # Wait 5 seconds before polling again
 
 def start_app():
     import argparse