fix: generate_mesh

app.py

@@ -177,7 +177,7 @@ def _default_render_cameras(batch_size: int = 1):
     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):
+def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export_mesh=False, export_video=False, fps=30):
     image = preprocess_image(image, source_size).to(model_wrapper.device)
     source_camera = _default_source_camera(batch_size=1).to(model_wrapper.device)
 
@@ -196,7 +196,7 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
             mesh_path = "xiaoxis_mesh.obj"
             mesh.export(mesh_path, 'obj')
 
-            return mesh_path, mesh_path
+            return planes, mesh_path
 
         if export_video:
             render_cameras = _default_render_cameras(batch_size=1).to(model_wrapper.device)
@@ -220,25 +220,24 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
             video_path = "xiaoxis_video.mp4"
             imageio.mimwrite(video_path, frames, fps=fps)
 
-            return None, video_path
+            return planes, video_path
+
+        return planes, None
 
     return None, None
 
 def step_1_generate_planes(image):
-    planes_path = generate_mesh(image)
+    planes_path, _ = generate_mesh(image)
     return planes_path
 
 def step_2_generate_obj(image):
-    mesh_path, _ = generate_mesh(image, export_mesh=True)
+    _, mesh_path = generate_mesh(image, export_mesh=True)
     return mesh_path, mesh_path
 
 def step_3_generate_video(image):
     _, video_path = generate_mesh(image, export_video=True)
     return video_path
 
-def step_4_display_3d_model(mesh_file):
-    return mesh_file
-
 # 从 assets 文件夹中设置示例文件，并限制最多读取 10 个文件
 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]