Upload 2 files

requirements.txt

@@ -1,4 +1,4 @@
-gradio==5.13.1
+gradio>=5.13.1
 numpy>=1.24.0
 plotly>=5.18.0
-py-vox-io
+git+https://github.com/MiguelPeFr/py-vox-io.git

viewer.py

@@ -47,12 +47,12 @@ def create_3d_scatter(voxels, palette):
         x=x, y=y, z=z,
         mode='markers',
         marker=dict(
-            size=6,  # Reduced size for better visualization
+            size=6, 
             color=rgb_colors,
             opacity=1.0,
             symbol='square',  # Using square symbol (supported by Plotly)
             line=dict(width=0),  # Remove border lines completely
-            sizemode='diameter',  # Fixed size regardless of zoom
+            sizemode='diameter', 
             sizeref=1.0  # Reference scale for consistent size
         )
     )])
@@ -63,14 +63,14 @@ def create_3d_scatter(voxels, palette):
     center_z = (z.max() + z.min()) / 2
     max_range = max(x.max() - x.min(), y.max() - y.min(), z.max() - z.min())
     
-    # Update layout for better visualization
+    # better visualization
     fig.update_layout(
         scene=dict(
             aspectmode='cube',  # Force cubic aspect ratio
             camera=dict(
                 up=dict(x=0, y=1, z=0),
                 center=dict(x=0, y=0, z=0),
-                eye=dict(x=1.5, y=0.9, z=0.9)  # Adjusted camera position for a closer view while maintaining angle
+                eye=dict(x=1.5, y=0.9, z=0.9) 
             ),
             xaxis=dict(range=[center_x - max_range/1.5, center_x + max_range/1.5], showgrid=True, gridwidth=1, gridcolor='rgba(128, 128, 128, 0.2)'),
             yaxis=dict(range=[center_y - max_range/1.5, center_y + max_range/1.5], showgrid=True, gridwidth=1, gridcolor='rgba(128, 128, 128, 0.2)'),
@@ -90,19 +90,70 @@ def display_vox_model(vox_file):
     try:
         # Load the vox model
         if not vox_file:
-            return "Please upload a .vox file"
+            fig = go.Figure()
+            fig.add_annotation(
+                text="Please upload a .vox file",
+                xref="paper", yref="paper",
+                x=0.5, y=0.5,
+                showarrow=False,
+                font=dict(size=16, color="white")
+            )
+            fig.update_layout(
+                paper_bgcolor='rgba(0,0,0,1)',
+                plot_bgcolor='rgba(0,0,0,1)'
+            )
+            return fig
         
         if not vox_file.name.endswith('.vox'):
-            return "Please upload a valid .vox file"
+            fig = go.Figure()
+            fig.add_annotation(
+                text="Please upload a valid .vox file",
+                xref="paper", yref="paper",
+                x=0.5, y=0.5,
+                showarrow=False,
+                font=dict(size=16, color="white")
+            )
+            fig.update_layout(
+                paper_bgcolor='rgba(0,0,0,1)',
+                plot_bgcolor='rgba(0,0,0,1)'
+            )
+            return fig
         
         print(f"Loading vox file: {vox_file.name}")
-        voxels, palette = load_vox_model(vox_file.name)
+        voxels, palette = load_vox_model(vox_file.temp_path if hasattr(vox_file, 'temp_path') else vox_file.name)
         
         if voxels is None or palette is None:
-            return "Error: Could not load voxel data from file"
+            fig = go.Figure()
+            error_message = "Error: Could not load voxel data from file\n"
+            error_message += "This might be due to version incompatibility.\n"
+            error_message += "The viewer currently supports .vox files up to version 200."
+            fig.add_annotation(
+                text=error_message,
+                xref="paper", yref="paper",
+                x=0.5, y=0.5,
+                showarrow=False,
+                font=dict(size=16, color="white")
+            )
+            fig.update_layout(
+                paper_bgcolor='rgba(0,0,0,1)',
+                plot_bgcolor='rgba(0,0,0,1)'
+            )
+            return fig
         
         if voxels.size == 0:
-            return "Error: No voxels found in the model"
+            fig = go.Figure()
+            fig.add_annotation(
+                text="Error: No voxels found in the model",
+                xref="paper", yref="paper",
+                x=0.5, y=0.5,
+                showarrow=False,
+                font=dict(size=16, color="white")
+            )
+            fig.update_layout(
+                paper_bgcolor='rgba(0,0,0,1)',
+                plot_bgcolor='rgba(0,0,0,1)'
+            )
+            return fig
         
         print(f"Model loaded successfully. Shape: {voxels.shape}, Palette size: {len(palette)}")
         
@@ -112,17 +163,30 @@ def display_vox_model(vox_file):
         return fig
     except Exception as e:
         print(f"Error details: {str(e)}")
-        return f"Error loading model: {str(e)}"
+        # Create an empty figure with error message
+        fig = go.Figure()
+        fig.add_annotation(
+            text=f"Error loading model: {str(e)}",
+            xref="paper", yref="paper",
+            x=0.5, y=0.5,
+            showarrow=False,
+            font=dict(size=16, color="white")
+        )
+        fig.update_layout(
+            paper_bgcolor='rgba(0,0,0,1)',
+            plot_bgcolor='rgba(0,0,0,1)'
+        )
+        return fig
 
 # Create Gradio interface
 interface = gr.Interface(
     fn=display_vox_model,
     inputs=gr.File(label="Upload .vox file"),
-    outputs=gr.Plot(label="3D Voxel Model"),
+    outputs=gr.Plot(label="3D Voxel Model"),  # Remove the type parameter
     title="Voxel Model Viewer",
     description="Upload a .vox file to view the 3D voxelized model.",
-    examples=[["examples/modelo_optimizado.vox"]]  # Updated path
+    examples=[["examples/modelo_optimizado.vox"]]
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()