Update app.py

app.py

@@ -312,35 +312,33 @@ def safe_generate_video(image):
 # 定义 Gradio 接口
 with gr.Blocks() as demo:
     gr.Markdown("Upload an image and provide a description to generate a video.")
+
     with gr.Tab("Basic Version"):
         with gr.Row():
             image_input = gr.Image(type="pil")
             description_input = gr.Textbox(label="Description", placeholder="Enter description here", lines=2)
         with gr.Row():
-            video_output = gr.Video(label="Generated Video")
-            error_output = gr.Textbox(label="Error Messages", placeholder="No errors", lines=5)
-        gr.Interface(
-            fn=generate_video_basic,
-            inputs=[image_input, description_input],
-            outputs=[video_output, error_output],
-            title="Basic Version Video Generation",
-            description="Upload an image and some descriptions to generate a video in the basic version. Any errors will be displayed below.",
-            theme="soft"
+            submit_button = gr.Button("Generate Basic Video")
+        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],
+            outputs=[video_output, error_output]
         )
+
     with gr.Tab("Pro Version"):
         with gr.Row():
-            image_input = gr.Image(type="pil")
-            description_input = gr.Textbox(label="Description", placeholder="Enter description here", lines=2)
+            image_input_pro = gr.Image(type="pil")
+            description_input_pro = gr.Textbox(label="Description", placeholder="Enter description here", lines=2)
         with gr.Row():
-            video_output = gr.Video(label="Generated Video")
-            error_output = gr.Textbox(label="Error Messages", placeholder="No errors", lines=5)
-        gr.Interface(
-            fn=generate_video_pro,
-            inputs=[image_input, description_input],
-            outputs=[video_output, error_output],
-            title="Pro Version Video Generation",
-            description="Upload an image and some descriptions to generate a video in the pro version. Any errors will be displayed below.",
-            theme="soft"
+            submit_button_pro = gr.Button("Generate Pro Video")
+        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, 
+            inputs=[image_input_pro, description_input_pro],
+            outputs=[video_output_pro, error_output_pro]
         )
 
 demo.launch()