Update app.py

app.py

@@ -7,7 +7,6 @@ from pathlib import Path
 import logging
 from typing import Tuple, Optional
 import struct
-import shutil
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -15,24 +14,22 @@ logger = logging.getLogger(__name__)
 
 class VoxParser:
     """Parse MagicaVoxel .vox files"""
-    def __init__(self, file_path):
-        self.file_path = file_path
+    def __init__(self, file_data):
+        self.file_data = file_data
         self.voxels = []
         self.palette = []
         self.size = {}
+        self.offset = 0
         
     def parse(self) -> dict:
         """Parse the .vox file and extract voxel data"""
         try:
-            # Handle both file paths and file-like objects
-            if hasattr(self.file_path, 'read'):
-                # File-like object from Gradio
-                data = self.file_path.read()
+            # Handle both bytes and file objects
+            if hasattr(self.file_data, 'read'):
+                data = self.file_data.read()
             else:
-                # File path
-                with open(self.file_path, 'rb') as f:
-                    data = f.read()
-            
+                data = self.file_data
+                
             self.buffer = data
             self.view = memoryview(data)
             self.offset = 0
@@ -150,41 +147,27 @@ class VoxToGlbConverter:
         Convert .vox file to .glb file
         
         Args:
-            vox_file: Gradio file object or file path
+            vox_file: Gradio file object
             
         Returns:
             Tuple of (glb_file_path, status_message)
         """
         try:
-            # Create a temporary file to handle the upload
-            with tempfile.NamedTemporaryFile(delete=False, suffix='.vox') as tmp_file:
-                if hasattr(vox_file, 'name'):
-                    # Gradio file object - copy to temp file
-                    shutil.copy(vox_file.name, tmp_file.name)
-                    temp_vox_path = tmp_file.name
-                else:
-                    # Direct file path
-                    temp_vox_path = vox_file
-                
-                # Parse the .vox file
-                parser = VoxParser(temp_vox_path)
-                voxel_data = parser.parse()
-                
-                if not voxel_data['voxels']:
-                    return "", "No voxels found in the file"
-                
-                # Create mesh from voxels
-                mesh = self.create_mesh_from_voxels(voxel_data)
-                
-                # Save as GLB to a permanent location
-                output_path = str(Path(tempfile.gettempdir()) / f"converted_{os.path.basename(temp_vox_path)}.glb")
-                mesh.export(output_path)
-                
-                # Clean up temp file
-                if temp_vox_path != vox_file:
-                    os.unlink(temp_vox_path)
-                
-                return output_path, f"Successfully converted {len(voxel_data['voxels'])} voxels to GLB format"
+            # Parse the .vox file directly from the file object
+            parser = VoxParser(vox_file)
+            voxel_data = parser.parse()
+            
+            if not voxel_data['voxels']:
+                return "", "No voxels found in the file"
+            
+            # Create mesh from voxels
+            mesh = self.create_mesh_from_voxels(voxel_data)
+            
+            # Save as GLB
+            output_path = str(Path(tempfile.gettempdir()) / f"converted_model.glb")
+            mesh.export(output_path)
+            
+            return output_path, f"Successfully converted {len(voxel_data['voxels'])} voxels to GLB format"
                 
         except Exception as e:
             logger.error(f"Conversion error: {e}")
@@ -321,4 +304,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=True,
         show_error=True
-    ) 
+    )