Update app.py

app.py

@@ -44,6 +44,13 @@ def t2v_generation_async(prompt, size, watermark_wan, seed = -1):
                                         size=size,
                                         seed=seed,
                                         watermark_wanx=watermark_wan)
+        print("API Response:", rsp)  # 添加详细日志
+        
+        # 检查响应是否有效
+        if rsp is None or not hasattr(rsp, 'output') or rsp.output is None:
+            gr.Warning(f"Warning: Invalid response from API: {rsp}")
+            return None, False, gr.Button(visible=True)
+            
         task_id = rsp.output.task_id
         status = False
         return task_id, status, gr.Button(visible=False)
@@ -78,7 +85,13 @@ def i2v_generation_async(prompt, image, watermark_wan, seed = -1):
             
         rsp = VideoSynthesis.async_call(model="wanx2.1-i2v-plus", prompt=prompt, seed=seed,
                                         img_url=img_url, watermark_wanx=watermark_wan)
-        print(rsp)
+        print("API Response:", rsp)  # 添加详细日志
+        
+        # 检查响应是否有效
+        if rsp is None or not hasattr(rsp, 'output') or rsp.output is None:
+            gr.Warning(f"Warning: Invalid response from API: {rsp}")
+            return "", None, gr.Button(visible=True)
+            
         task_id = rsp.output.task_id
         status = False
         return task_id, status, gr.Button(visible=False)