Update app.py

Filled Mesh: High-quality with hole filling

app.py

@@ -182,11 +182,37 @@ def extract_high_quality_mesh(
     
     # Create mesh and save
     simple_mesh = trimesh.Trimesh(vertices=rotated_vertices, faces=faces)
-    glb_path = os.path.join(user_dir, f"{trial_id}_full.glb")
+    glb_path = os.path.join(user_dir, f"{trial_id}_raw.glb")
     simple_mesh.export(glb_path)
     
     return glb_path, glb_path
+
+@spaces.GPU
+def extract_high_quality_mesh_with_holes_filled(
+    state: dict,
+    req: gr.Request,
+) -> Tuple[str, str]:
+    """
+    Save mesh data with hole filling but no other processing.
+    """
+    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    gs, mesh, trial_id = unpack_state(state)
+    
+    # Use to_glb only for hole filling
+    raw_mesh = postprocessing_utils.to_glb(
+        gs,
+        mesh,
+        simplify=0.0,  # No simplification
+        fill_holes=True,  # Enable hole filling
+        fill_holes_max_size=0.04,
+        verbose=True
+    )
     
+    glb_path = os.path.join(user_dir, f"{trial_id}_filled.glb")
+    raw_mesh.export(glb_path)
+    
+    return glb_path, glb_path
+
 @spaces.GPU
 def extract_reduced_glb(
     state: dict,
@@ -221,9 +247,10 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
     ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
     * Upload an image and click "Generate" to create a 3D asset
-    * After generation:
-        * Click "Extract Full GLB" for maximum detail mesh (untextured)
-        * Or use GLB Extraction Settings for a reduced textured version
+    * After generation, choose from three export options:
+        * Raw Mesh: Maximum detail, untextured, may have holes
+        * Filled Mesh: Maximum detail, untextured, holes filled
+        * Reduced GLB: Reduced size with textures
     """)
     
     with gr.Row():
@@ -243,7 +270,8 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                     slat_sampling_steps = gr.Slider(1, 500, label="Sampling Steps", value=12, step=1)
 
             generate_btn = gr.Button("Generate")
-            extract_full_btn = gr.Button("Extract Full GLB", interactive=False)
+            extract_raw_btn = gr.Button("Extract Raw Mesh", interactive=False)
+            extract_filled_btn = gr.Button("Extract Filled Mesh", interactive=False)
             
             with gr.Accordion(label="GLB Extraction Settings", open=False):
                 mesh_simplify = gr.Slider(0.0, 0.98, label="Simplify", value=0.95, step=0.01)
@@ -256,7 +284,8 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             model_output = LitModel3D(label="3D Model Preview", exposure=20.0, height=300)
             gr.Markdown("### Download Options")
             with gr.Row():
-                download_full = gr.DownloadButton(label="Download Full-Quality GLB", interactive=False)
+                download_raw = gr.DownloadButton(label="Download Raw Mesh", interactive=False)
+                download_filled = gr.DownloadButton(label="Download Filled Mesh", interactive=False)
                 download_reduced = gr.DownloadButton(label="Download Reduced GLB", interactive=False)
             
     output_buf = gr.State()
@@ -294,17 +323,28 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         inputs=[image_prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
         outputs=[output_buf, video_output],
     ).then(
-        lambda: [gr.Button(interactive=True), gr.Button(interactive=True), gr.Button(interactive=False), gr.Button(interactive=False)],
-        outputs=[extract_full_btn, extract_reduced_btn, download_full, download_reduced],
+        lambda: [gr.Button(interactive=True), gr.Button(interactive=True), gr.Button(interactive=True),
+                 gr.Button(interactive=False), gr.Button(interactive=False), gr.Button(interactive=False)],
+        outputs=[extract_raw_btn, extract_filled_btn, extract_reduced_btn,
+                 download_raw, download_filled, download_reduced],
     )
 
-    extract_full_btn.click(
+    extract_raw_btn.click(
         extract_high_quality_mesh,
         inputs=[output_buf],
-        outputs=[model_output, download_full],
+        outputs=[model_output, download_raw],
+    ).then(
+        lambda: gr.Button(interactive=True),
+        outputs=[download_raw],
+    )
+
+    extract_filled_btn.click(
+        extract_high_quality_mesh_with_holes_filled,
+        inputs=[output_buf],
+        outputs=[model_output, download_filled],
     ).then(
         lambda: gr.Button(interactive=True),
-        outputs=[download_full],
+        outputs=[download_filled],
     )
 
     extract_reduced_btn.click(