Update app.py

app.py

@@ -3,7 +3,7 @@ import spaces
 from gradio_litmodel3d import LitModel3D
 import os
 import shutil
-import igl
+
 os.environ['SPCONV_ALGO'] = 'native'
 from typing import *
 import torch
@@ -134,69 +134,6 @@ def extract_glb(
     torch.cuda.empty_cache()
     return glb_path, glb_path
 
-@spaces.GPU(duration=90)
-def extract_glb_with_remesh(
-    state: dict,
-    mesh_simplify: float,
-    texture_size: int,
-    remesh_type: str,
-    target_count: int,
-    req: gr.Request,
-) -> Tuple[str, str]:
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
-    gs, old_mesh = unpack_state(state)
-    
-    # Datos originales
-    V_old = old_mesh.vertices.cpu().numpy()
-    F_old = old_mesh.faces.cpu().numpy()
-    uv_old = old_mesh.uv.cpu().numpy() if hasattr(old_mesh, 'uv') else None
-    face_uv_old = old_mesh.face_uv.cpu().numpy() if hasattr(old_mesh, 'face_uv') else None
-    
-    if remesh_type != "none" and (TRIANGLE_REMESH_AVAILABLE or QUAD_REMESH_AVAILABLE):
-        # Remeshing con libigl
-        if remesh_type == "triangle":
-            V_new, F_new = igl.triangulate(V_old, F_old)
-            V_new, F_new = igl.decimate(V_new, F_new, target_count)
-        elif remesh_type == "quad":
-            V_new, F_new = igl.quad_remesh(V_old, F_old, target_count)
-        
-        # Transferencia de UV usando interpolación baricéntrica
-        if uv_old is not None:
-            barycoord, closest_faces = igl.barycentric_coordinates_tri(
-                V_new, 
-                V_old[F_old[closest_faces,0]],
-                V_old[F_old[closest_faces,1]],
-                V_old[F_old[closest_faces,2]]
-            )
-            
-            uv_new = np.zeros((V_new.shape[0], 2))
-            for i in range(3):
-                uv_new += barycoord[:,i].reshape(-1,1) * uv_old[face_uv_old[closest_faces,i]]
-        else:
-            uv_new = None
-            face_uv_new = None
-    else:
-        # Sin remeshing
-        V_new, F_new = V_old, F_old
-        uv_new = uv_old
-        face_uv_new = face_uv_old
-    
-    # Crear nueva malla
-    new_mesh = edict(
-        vertices=torch.tensor(V_new, device='cuda'),
-        faces=torch.tensor(F_new, device='cuda'),
-        uv=torch.tensor(uv_new, device='cuda') if uv_new is not None else None,
-        face_uv=torch.tensor(face_uv_new, device='cuda') if face_uv_new is not None else None,
-    )
-    
-    # Exportar a GLB
-    glb = postprocessing_utils.to_glb(gs, new_mesh, simplify=mesh_simplify, 
-                                      texture_size=texture_size, verbose=False)
-    glb_path = os.path.join(user_dir, 'sample.glb')
-    glb.export(glb_path)
-    torch.cuda.empty_cache()
-    return glb_path, glb_path
-
 @spaces.GPU
 def extract_gaussian(state: dict, req: gr.Request) -> Tuple[str, str]:
 
@@ -248,10 +185,6 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                 with gr.Row():
                     slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=3.0, step=0.1)
                     slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
-
-            with gr.Accordion(label="Remeshing Settings", open=False):
-                remesh_type = gr.Radio(choices=["none", "triangle", "quad"], label="Remesh Type", value="none")
-                target_count = gr.Slider(1000, 100000, label="Target Vertex/Face Count", value=20000, step=1000)
                 
             generate_btn = gr.Button("Generate")
             
@@ -305,8 +238,8 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     )
 
     extract_glb_btn.click(
-        extract_glb_with_remesh,
-        inputs=[output_buf, mesh_simplify, texture_size, remesh_type, target_count],
+        extract_glb,
+        inputs=[output_buf, mesh_simplify, texture_size],
         outputs=[model_output, download_glb],
     ).then(
         lambda: gr.Button(interactive=True),