Update app.py

app.py

@@ -208,15 +208,34 @@ def generate_video(image):
     print("video_done")
     return video_data
     
+import traceback
+
+def safe_generate_video(image):
+    try:
+        # 尝试生成视频
+        return generate_video(image), None
+    except Exception as e:
+        # 捕获任何异常并返回错误消息
+        error_msg = f"An error occurred: {str(e)}"
+        # 可选：打印堆栈跟踪信息以便于调试
+        print(traceback.format_exc())
+        return None, error_msg
+
 # 定义 Gradio 接口
 interface = gr.Interface(
-    fn=lambda img: generate_video(img),
+    fn=lambda img: safe_generate_video(img.convert('RGB')),  # 确保输入是RGB格式的图片
     inputs=gr.Image(type="pil"),
-    outputs=gr.Video(),
+    outputs=[
+        gr.Video(label="Generated Video"),
+        gr.Textbox(label="Error Messages", placeholder="No errors", lines=5)
+    ],
     title="InspiroV Video Generation",
-    description="Upload an image to generate a video",
+    description="Upload an image to generate a video. Any errors will be displayed below.",
     theme="soft"
 )
 
 # 启动 Gradio 应用
 interface.launch()
+
+# 启动 Gradio 应用
+interface.launch()