clean scene 9

app.py

@@ -80,15 +80,20 @@ def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25):
             faces=int(face_number),
         )
 
-        if mesh is None:
-            raise ValueError("Mesh generation returned None")
+        if mesh is None or mesh.vertices.shape[0] == 0 or mesh.faces.shape[0] == 0:
+            raise ValueError("Mesh generation returned an empty mesh")
+
+        # Безопасная очистка визуала
+        if hasattr(mesh, "visual") and mesh.visual is not None:
+            try:
+                mesh.visual = None
+            except Exception:
+                print("[WARN] Failed to clear visual, skipping")
 
-        # Очистка визуала, метаданных и имени
-        mesh.visual = None
         mesh.metadata.clear()
         mesh.name = "geometry_0"
 
-        # Экспорт в GLB без scene/world
+        # Экспорт .glb
         glb_data = export_glb(mesh)
         with open(output_path, "wb") as f:
             f.write(glb_data)

inference_triposg.py

@@ -89,14 +89,25 @@ def pymesh_to_trimesh(mesh):
     faces = mesh.face_matrix()#.tolist()
     return trimesh.Trimesh(vertices=verts, faces=faces)  #, vID, fID
 
+# def simplify_mesh(mesh: trimesh.Trimesh, n_faces):
+#     if mesh.faces.shape[0] > n_faces:
+#         ms = mesh_to_pymesh(mesh.vertices, mesh.faces)
+#         ms.meshing_merge_close_vertices()
+#         ms.meshing_decimation_quadric_edge_collapse(targetfacenum = n_faces)
+#         return pymesh_to_trimesh(ms.current_mesh())
+#     else:
+#         return mesh
+
 def simplify_mesh(mesh: trimesh.Trimesh, n_faces):
     if mesh.faces.shape[0] > n_faces:
         ms = mesh_to_pymesh(mesh.vertices, mesh.faces)
         ms.meshing_merge_close_vertices()
-        ms.meshing_decimation_quadric_edge_collapse(targetfacenum = n_faces)
-        return pymesh_to_trimesh(ms.current_mesh())
-    else:
-        return mesh
+        ms.meshing_decimation_quadric_edge_collapse(targetfacenum=n_faces)
+        simplified = ms.current_mesh()
+        if simplified is None or simplified.face_number() == 0:
+            return None
+        return pymesh_to_trimesh(simplified)
+    return mesh
 
 if __name__ == "__main__":
     device = "cuda"