Update app.py

app.py

@@ -322,8 +322,8 @@ 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(
-            generate_video_basic, 
-            inputs=[image_input,description_input],
+            lambda img, desc: safe_generate_video(lambda: generate_video_basic(img, desc)), 
+            inputs=[image_input, description_input],
             outputs=[video_output, error_output]
         )
 
@@ -336,10 +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(
-            generate_video_pro, 
+            lambda img, desc: safe_generate_video(lambda: generate_video_pro(img, desc)),  
             inputs=[image_input_pro, description_input_pro],
             outputs=[video_output_pro, error_output_pro]
         )
 
 demo.launch()
-