Update app.py

app.py

@@ -1,6 +1,8 @@
 import pyvista as pv
 import json
 from stpyvista import stpyvista
+import matplotlib.pyplot as plt
+import matplotlib.colors as mcolors
 
     # Load the .obj file
 mesh = pv.read('file.obj')
@@ -12,6 +14,7 @@ with open('dental-labels4.json', 'r') as file:
     # Assuming labels_data['labels'] is a list of labels
 labels = labels_data['labels']
 
+
     # Make sure the number of labels matches the number of vertices or faces
 assert len(labels) == mesh.n_points or len(labels) == mesh.n_cells
 
@@ -25,8 +28,16 @@ elif len(labels) == mesh.n_cells:
 # Create a pyvista plotter
 plotter = pv.Plotter()
 
+cmap = plt.cm.get_cmap('tab20', n_clusters)  # Using a colormap with sufficient distinct colors
+
+colors = cmap(np.linspace(0, 1, n_clusters))  # Generate colors
+
+# Convert colors to a format acceptable by PyVista
+colormap = mcolors.ListedColormap(colors)
+
 # Add the mesh to the plotter with labels as a scalar field
-plotter.add_mesh(mesh, scalars='Labels', show_scalar_bar=True, cmap='jet')
+#plotter.add_mesh(mesh, scalars='Labels', show_scalar_bar=True, cmap='jet')
+plotter.add_mesh(mesh, scalars='Labels', show_scalar_bar=True, cmap=colormap, clim=[0, n_clusters - 1])
 
 # Show the plot
 #plotter.show()