Update app.py

app.py

@@ -47,12 +47,10 @@ def generate_video(image, prompt=""):
     }
     
     payload = {
-        "inputs": [{"name": "input", "inputType": "URL", "url": image_base64}],
-        "arguments": [
-            {"name": "prompt", "value": prompt},
-            {"name": "duration", "value": "5"},
-            {"name": "aspect_ratio", "value": "16:9"}
-        ]
+        "image": image_base64,
+        "prompt": prompt,
+        "duration": 5,
+        "aspect_ratio": "16:9"
     }
     
     try:
@@ -60,7 +58,7 @@ def generate_video(image, prompt=""):
         response.raise_for_status()
         
         data = response.json()
-        task_id = data.get("taskId") or data.get("task_id")
+        task_id = data.get("task_id") or data.get("taskId")
         if not task_id:
             return "Error: No task ID returned by the API."
         
@@ -70,8 +68,8 @@ def generate_video(image, prompt=""):
             status_response = requests.get(status_url, headers=headers, timeout=30)
             status_response.raise_for_status()
             status_data = status_response.json()
-            if status_data.get("status") == "succeed":
-                video_url = status_data.get("task_result", {}).get("videos", [{}])[0].get("url")
+            if status_data.get("status") == "succeeded":
+                video_url = status_data.get("video_url") or status_data.get("result", {}).get("video_url")
                 if not video_url:
                     return "Error: No video URL in API response."
                 # Download the video
@@ -87,12 +85,14 @@ def generate_video(image, prompt=""):
         
         return "Error: Video generation timed out."
     
-    except requests.exceptions.RequestException as e:
+    except requests.exceptions.HTTPError as e:
         return f"Error: API request failed. Details: {str(e)}"
+    except requests.exceptions.RequestException as e:
+        return f"Error: Network issue. Details: {str(e)}"
 
 def chatbot_interface(image, prompt):
     """
-    Gradio interface for image-to-image video generation.
+    Gradio interface for image-to-video generation.
     
     Args:
         image: Uploaded image file