Update app.py

app.py

@@ -298,10 +298,10 @@ def generate_video_pro(image,desc):
 
 import traceback
 
-def safe_generate_video(image):
+def safe_generate_video(func, *args):
     try:
-        # 尝试生成视频
-        return generate_video(image), None
+        # 尝试生成视频，调用传入的函数
+        return func(*args), None
     except Exception as e:
         # 捕获任何异常并返回错误消息
         error_msg = f"An error occurred: {str(e)}"
@@ -309,7 +309,7 @@ def safe_generate_video(image):
         print(traceback.format_exc())
         return None, error_msg
 
-# 定义 Gradio 接口
+
 with gr.Blocks() as demo:
     gr.Markdown("Upload an image and provide a description to generate a video.")
 
@@ -322,7 +322,7 @@ with gr.Blocks() as demo:
         video_output = gr.Video(label="Generated Video")
         error_output = gr.Textbox(label="Error Messages", placeholder="No errors", lines=5)
         submit_button.click(
-            lambda img, desc: safe_generate_video(lambda: generate_video_basic(img, desc)), 
+            lambda img, desc: safe_generate_video(generate_video_basic, img, desc),
             inputs=[image_input, description_input],
             outputs=[video_output, error_output]
         )
@@ -336,9 +336,9 @@ with gr.Blocks() as demo:
         video_output_pro = gr.Video(label="Generated Video")
         error_output_pro = gr.Textbox(label="Error Messages", placeholder="No errors", lines=5)
         submit_button_pro.click(
-            lambda img, desc: safe_generate_video(lambda: generate_video_pro(img, desc)),  
+            lambda img, desc: safe_generate_video(generate_video_pro, img, desc),
             inputs=[image_input_pro, description_input_pro],
             outputs=[video_output_pro, error_output_pro]
         )
-
 demo.launch()
+