Update app.py

app.py

@@ -4,14 +4,8 @@ import trimesh
 import tempfile
 import os
 import struct
-import logging
 from pathlib import Path
 from typing import Tuple
-import shutil
-
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
 
 class VoxParser:
     """MagicaVoxel .vox file parser"""
@@ -33,8 +27,7 @@ class VoxParser:
             if header != 'VOX ':
                 raise ValueError("Invalid VOX file header")
             
-            version = struct.unpack('<I', data[offset:offset+4])[0]
-            offset += 4
+            offset += 4  # Skip version
             
             voxels = []
             palette = []
@@ -42,6 +35,9 @@ class VoxParser:
             
             # Parse chunks
             while offset < len(data):
+                if offset + 12 > len(data):
+                    break
+                    
                 chunk_id = data[offset:offset+4].decode('ascii', errors='ignore')
                 offset += 4
                 chunk_size = struct.unpack('<I', data[offset:offset+4])[0]
@@ -50,26 +46,28 @@ class VoxParser:
                 offset += 4
                 
                 if chunk_id == 'SIZE':
-                    size = {
-                        'x': struct.unpack('<I', data[offset:offset+4])[0],
-                        'y': struct.unpack('<I', data[offset+4:offset+8])[0],
-                        'z': struct.unpack('<I', data[offset+8:offset+12])[0]
-                    }
+                    if offset + 12 <= len(data):
+                        size = {
+                            'x': struct.unpack('<I', data[offset:offset+4])[0],
+                            'y': struct.unpack('<I', data[offset+4:offset+8])[0],
+                            'z': struct.unpack('<I', data[offset+8:offset+12])[0]
+                        }
                     offset += chunk_size
                 elif chunk_id == 'XYZI':
-                    num_voxels = struct.unpack('<I', data[offset:offset+4])[0]
-                    offset += 4
-                    
-                    for i in range(num_voxels):
-                        if offset + 4 <= len(data):
-                            x, y, z, color_index = struct.unpack('BBBB', data[offset:offset+4])
-                            voxels.append({
-                                'x': x,
-                                'y': y,
-                                'z': z,
-                                'color_index': color_index
-                            })
-                            offset += 4
+                    if offset + 4 <= len(data):
+                        num_voxels = struct.unpack('<I', data[offset:offset+4])[0]
+                        offset += 4
+                        
+                        for i in range(num_voxels):
+                            if offset + 4 <= len(data):
+                                x, y, z, color_index = struct.unpack('BBBB', data[offset:offset+4])
+                                voxels.append({
+                                    'x': x,
+                                    'y': y,
+                                    'z': z,
+                                    'color_index': color_index
+                                })
+                                offset += 4
                 elif chunk_id == 'RGBA':
                     for i in range(256):
                         if offset + 4 <= len(data):
@@ -88,7 +86,7 @@ class VoxParser:
                 'size': size
             }
         except Exception as e:
-            raise ValueError(f"Error parsing VOX file: {e}")
+            return {'voxels': [], 'palette': self._default_palette(), 'size': {'x': 0, 'y': 0, 'z': 0}}
     
     def _default_palette(self) -> list:
         colors = []
@@ -126,7 +124,6 @@ class VoxToGlbConverter:
             return output_path, f"Converted {voxel_count} voxels to GLB format"
                 
         except Exception as e:
-            logger.error(f"Conversion error: {e}")
             return "", f"Error converting file: {str(e)}"
     
     def create_mesh_from_voxels(self, voxel_data: dict) -> trimesh.Trimesh:
@@ -214,8 +211,8 @@ def create_gradio_interface():
         # Use Gradio's native Model3D component for GLB files
         model_3d = gr.Model3D(
             label="GLB Preview",
-            height=400,
-            camera_radius=10.0,
+            height=300,
+            camera_radius=8.0,
             camera_center=(0, 0, 0),
             zoom_speed=1.0,
             pan_speed=2.0,
@@ -233,7 +230,7 @@ def create_gradio_interface():
         
         convert_btn.click(
             fn=convert_with_preview,
-            inputs=[vox_input],
+            inputs=[vox_file],
             outputs=[glb_output, status_output, model_3d]
         )