revert: to 5864223c556776d63e24b63ec2cba1ba170a2558

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=False, fps=30):
+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)
     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 planes, mesh_path
+            return mesh_path, mesh_path
 
         if export_video:
             render_cameras = _default_render_cameras(batch_size=1).to(model_wrapper.device)
@@ -220,23 +220,20 @@ 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 planes, video_path
-
-        return planes, None
+            return None, video_path
 
     return None, None
 
-def step_1_generate_planes(image):
-    planes_path, _ = generate_mesh(image)
-    return planes_path.shape
-
-def step_2_generate_obj(image):
-    _, mesh_path = generate_mesh(image, export_mesh=True)
-    return planes_path.shape, mesh_path, mesh_path
+def step_1_generate_obj(image):
+    mesh_path, _ = generate_mesh(image, export_mesh=True)
+    return mesh_path, mesh_path
 
-def step_3_generate_video(image):
+def step_2_generate_video(image):
     _, video_path = generate_mesh(image, export_video=True)
-    return planes_path.shape, video_path
+    return video_path
+
+def step_3_display_3d_model(mesh_file):
+    return mesh_file
 
 # 从 assets 文件夹中设置示例文件，并限制最多读取 10 个文件
 example_folder = "assets"
@@ -255,27 +252,21 @@ 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=5)
-            generate_planes_button = gr.Button("生成平面图")
+            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("生成视频")
         with gr.Column():
-            planes_output = gr.Image(
-                label="平面图",
-                type="pil",
-                interactive=False
-            )
             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():
@@ -287,18 +278,20 @@ with gr.Blocks() as demo:
 
     # 清除输出
     def clear_model_viewer():
-        """在加载新模型前重置 Gradio。"""
-        update_output = gr.update(value=None)
-        return update_output, update_output
+        """在加载新模型前重置 Model3D 组件。"""
+        return gr.update(value=None)
+
+    def generate_and_visualize(image):
+        mesh_path = step_1_generate_obj(image)
+        return mesh_path, mesh_path
 
-    # 首先清除输出的数据
-    img_input.change(clear_model_viewer, inputs=None, outputs=[planes_output, model_output])
+    # 首先清除输出的 3D 模型
+    img_input.change(clear_model_viewer, inputs=None, outputs=model_output)
 
     # 然后生成模型和视频
-    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=[planes_output, obj_file_output, model_output])
-    generate_video_button.click(step_3_generate_video, inputs=img_input, outputs=[planes_output, video_file_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)
 
 demo.launch(
     auth=(os.environ.get('AUTH_USERNAME'), os.environ.get('AUTH_PASSWORD'))
-)
+)