feat: add video generate method

app.py

@@ -175,7 +175,7 @@ def _default_render_cameras(batch_size: int = 1):
     render_camera_intrinsics = _default_intrinsics().unsqueeze(0).repeat(extrinsics.shape[0], 1, 1)
     render_cameras = _build_camera_standard(extrinsics, render_camera_intrinsics)
     return render_cameras.unsqueeze(0).repeat(batch_size, 1, 1)
-    
+
 @spaces.GPU
 def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export_mesh=False, export_video=True, fps=30):
     image = preprocess_image(image, source_size).to(model_wrapper.device)
@@ -239,6 +239,7 @@ def step_3_display_3d_model(mesh_file):
 example_folder = "assets"
 examples = [os.path.join(example_folder, f) for f in os.listdir(example_folder) if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))][:10]
 
+
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("""
@@ -249,15 +250,11 @@ with gr.Blocks() as demo:
         """)
 
     with gr.Row():
-
         with gr.Column():
             img_input = gr.Image(type="pil", label="输入图像")
             examples_component = gr.Examples(examples=examples, inputs=img_input, outputs=None, examples_per_page=3)
             generate_mesh_button = gr.Button("生成模型")
-            # generate_video_button = gr.Button("生成视频")
-            obj_file_output = gr.File(label="下载 .obj 文件")
-            # video_file_output = gr.File(label="下载视频")
-
+            generate_video_button = gr.Button("生成视频")
         with gr.Column():
             model_output = LitModel3D(
                 clear_color=[0, 0, 0, 0],  # 可调整背景颜色，以获得更好的对比度
@@ -269,24 +266,32 @@ with gr.Blocks() as demo:
                 camera_position=(0, 0, 2),  # 将设置初始摄像机位置，使模型居中
                 zoom_speed=0.5,      # 将调整变焦速度，以便更好地控制
                 pan_speed=0.5,       # 将调整摇摄速度，以便更好地控制
-                interactive=True     # 这样用户就可以与模型进行交互
+                interactive=False     # 这样用户就可以与模型进行交互
             )
-            
-        
-    # clear outputs
+
+    with gr.Row():
+        with gr.Column():
+            video_file_output = gr.File(label="下载视频")
+        with gr.Column():
+            obj_file_output = gr.File(label="下载 .obj 文件")
+
+
+    # 清除输出
     def clear_model_viewer():
         """在加载新模型前重置 Model3D 组件。"""
         return gr.update(value=None)
-    
+
     def generate_and_visualize(image):
         mesh_path = step_1_generate_obj(image)
         return mesh_path, mesh_path
 
-    # first we clear the existing 3D model
+    # 首先清除输出的 3D 模型
     img_input.change(clear_model_viewer, inputs=None, outputs=model_output)
 
-    # then, generate the mesh and video
+    # 然后生成模型和视频
     generate_mesh_button.click(step_1_generate_obj, inputs=img_input, outputs=[obj_file_output, model_output])
-    # generate_video_button.click(step_2_generate_video, inputs=img_input, outputs=video_file_output)
+    generate_video_button.click(step_2_generate_video, inputs=img_input, outputs=video_file_output)
 
-demo.launch(auth=(os.environ.get('AUTH_USERNAME'), os.environ.get('AUTH_PASSWORD')))
+demo.launch(
+    auth=(os.environ.get('AUTH_USERNAME'), os.environ.get('AUTH_PASSWORD'))
+)