Update app.py

app.py

@@ -1,12 +1,13 @@
 """
 Gradio app: .vox → .glb conversion + built-in Model3D viewer.
 
-Now we no longer use HTML + WebGL. This is simpler, more reliable, and fully compatible with Gradio 5+.
+Now using safer voxel-to-mesh conversion via occupancy grid + marching cubes.
 """
 
 import gradio as gr
 import tempfile
 import trimesh
+import numpy as np
 from pathlib import Path
 from pyvox.parser import VoxParser
 
@@ -14,15 +15,21 @@ from pyvox.parser import VoxParser
 def vox_to_glb(file_path):
     vox = VoxParser(file_path).parse()
     model = vox.models[0]
+    size_x, size_y, size_z = model.size.x, model.size.y, model.size.z
     voxels = model.voxels
 
-    cubes = []
+    if not voxels:
+        raise ValueError("No voxels found in file")
+
+    # Create occupancy grid
+    grid = np.zeros((size_x, size_y, size_z), dtype=bool)
     for v in voxels:
-        cube = trimesh.creation.box(extents=(1,1,1))
-        cube.apply_translation([v.x, v.y, v.z])
-        cubes.append(cube)
+        grid[v.x, v.y, v.z] = True
+
+    # Convert occupancy grid to mesh
+    mesh = trimesh.voxel.ops.matrix_to_marching_cubes(grid)
 
-    mesh = trimesh.util.concatenate(cubes)
+    # Export to temporary glb
     tmp = tempfile.NamedTemporaryFile(suffix='.glb', delete=False)
     mesh.export(tmp.name)
     return tmp.name
@@ -33,7 +40,11 @@ def handle_upload(file):
         return None
     ext = Path(file.name).suffix.lower()
     if ext == '.vox':
-        return vox_to_glb(file.name)
+        try:
+            return vox_to_glb(file.name)
+        except Exception as e:
+            print(f"Failed to convert vox: {e}")
+            return None
     elif ext in ['.glb', '.gltf']:
         return file.name
     else:
@@ -49,5 +60,4 @@ with gr.Blocks() as demo:
 
 
 if __name__ == '__main__':
-    demo.launch()
-
+    demo.launch()