feat: when generate_mesh & generate_video, show planes too

app.py

@@ -232,11 +232,11 @@ def step_1_generate_planes(image):
 
 def step_2_generate_obj(image):
     _, mesh_path = generate_mesh(image, export_mesh=True)
-    return mesh_path, mesh_path
+    return planes_path.shape, mesh_path, mesh_path
 
 def step_3_generate_video(image):
     _, video_path = generate_mesh(image, export_video=True)
-    return video_path
+    return planes_path.shape, video_path
 
 # 从 assets 文件夹中设置示例文件，并限制最多读取 10 个文件
 example_folder = "assets"
@@ -296,8 +296,8 @@ with gr.Blocks() as demo:
 
     # 然后生成模型和视频
     generate_planes_button.click(step_1_generate_planes, inputs=img_input, outputs=planes_output)
-    generate_mesh_button.click(step_2_generate_obj, inputs=img_input, outputs=[obj_file_output, model_output])
-    generate_video_button.click(step_3_generate_video, inputs=img_input, outputs=video_file_output)
+    generate_mesh_button.click(step_2_generate_obj, inputs=img_input, outputs=[planes_output, obj_file_output, model_output])
+    generate_video_button.click(step_3_generate_video, inputs=img_input, outputs=[planes_output, video_file_output])
 
 demo.launch(
     auth=(os.environ.get('AUTH_USERNAME'), os.environ.get('AUTH_PASSWORD'))