feat: new generate plates method

app.py

@@ -224,15 +224,19 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
 
     return None, None
 
-def step_1_generate_obj(image):
+def step_1_generate_planes(image):
+    planes_path = generate_mesh(image)
+    return planes_path
+
+def step_2_generate_obj(image):
     mesh_path, _ = generate_mesh(image, export_mesh=True)
     return mesh_path, mesh_path
 
-def step_2_generate_video(image):
+def step_3_generate_video(image):
     _, video_path = generate_mesh(image, export_video=True)
     return video_path
 
-def step_3_display_3d_model(mesh_file):
+def step_4_display_3d_model(mesh_file):
     return mesh_file
 
 # 从 assets 文件夹中设置示例文件，并限制最多读取 10 个文件
@@ -252,21 +256,23 @@ 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)
+            examples_component = gr.Examples(examples=examples, inputs=img_input, outputs=None, examples_per_page=5)
+            generate_planes_button = gr.Button("生成平面图")
             generate_mesh_button = gr.Button("生成模型")
             generate_video_button = gr.Button("生成视频")
         with gr.Column():
+            planes_output = gr.Image(label="平面图")
             model_output = LitModel3D(
                 clear_color=[0, 0, 0, 0],  # 可调整背景颜色，以获得更好的对比度
                 label="模型可视化",
                 scale=1.0,
-                tonemapping="aces",  # 可使用 aces 色调映射，使灯光更逼真
-                exposure=1.1,        # 可调节曝光以控制亮度
-                contrast=1.1,        # 可略微增加对比度，以获得更好的深度
-                camera_position=(0, 0, 2),  # 将设置初始摄像机位置，使模型居中
-                zoom_speed=0.5,      # 将调整变焦速度，以便更好地控制
-                pan_speed=0.5,       # 将调整摇摄速度，以便更好地控制
-                interactive=False     # 这样用户就可以与模型进行交互
+                tonemapping="aces",        # 可使用 aces 色调映射，使灯光更逼真
+                exposure=1.1,              # 可调节曝光以控制亮度
+                contrast=1.1,              # 可略微增加对比度，以获得更好的深度
+                camera_position=(0, 0, 2), # 将设置初始摄像机位置，使模型居中
+                zoom_speed=0.5,            # 将调整变焦速度，以便更好地控制
+                pan_speed=0.5,             # 将调整摇摄速度，以便更好地控制
+                interactive=False          # 这样用户就可以与模型进行交互
             )
 
     with gr.Row():
@@ -278,19 +284,16 @@ with gr.Blocks() as demo:
 
     # 清除输出
     def clear_model_viewer():
-        """在加载新模型前重置 Model3D 组件。"""
+        """在加载新模型前重置 Gradio。"""
         return gr.update(value=None)
 
-    def generate_and_visualize(image):
-        mesh_path = step_1_generate_obj(image)
-        return mesh_path, mesh_path
-
     # 首先清除输出的 3D 模型
     img_input.change(clear_model_viewer, inputs=None, outputs=model_output)
 
     # 然后生成模型和视频
-    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_planes_button.click(step_1_generate_planes, input=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)
 
 demo.launch(
     auth=(os.environ.get('AUTH_USERNAME'), os.environ.get('AUTH_PASSWORD'))