Upload 4 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+file.obj filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,17 +1,34 @@
-import streamlit as st
-import streamlit.components.v1 as components
-import pyvista
-from pyvista import examples
-pyvista.start_xvfb()
-mesh = examples.load_uniform()
-pl = pyvista.Plotter(shape=(1,2))
-_ = pl.add_mesh(mesh, scalars='Spatial Point Data', show_edges=True)
-pl.subplot(0,1)
-_ = pl.add_mesh(mesh, scalars='Spatial Cell Data', show_edges=True)
-pl.export_html('pyvista.html')  # doctest:+SKIP
+import pyvista as pv
+import json
+from stpyvista import stpyvista
 
-option=st.sidebar.radio('Pyvista',('On','Off'))
-if option=='On':
-  HtmlFile = open("pyvista.html", 'r', encoding='utf-8')
-  source_code = HtmlFile.read() 
-  components.html(source_code, height = 500,width=500)
+    # Load the .obj file
+mesh = pv.read('file.obj')
+
+    # Load the JSON file
+with open('output/dental-labels4.json', 'r') as file:
+    labels_data = json.load(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
+
+# If labels correspond to vertices
+if len(labels) == mesh.n_points:
+    mesh.point_data['Labels'] = labels
+# If labels correspond to faces
+elif len(labels) == mesh.n_cells:
+    mesh.cell_data['Labels'] = labels
+    
+# Create a pyvista plotter
+plotter = pv.Plotter()
+
+# Add the mesh to the plotter with labels as a scalar field
+plotter.add_mesh(mesh, scalars='Labels', show_scalar_bar=True, cmap='jet')
+
+# Show the plot
+#plotter.show()
+## Send to streamlit
+stpyvista(plotter)

file.obj

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:581b9a026e2ce734f6335f34aa900e8114dc33e2a83541ebd6bb26536382545e
+size 18769177

requirements.txt

@@ -1,7 +1,4 @@
 streamlit
 pyvista
 pythreejs
-trame-vtk
-trame==3.3.0
-trame-vuetify
-nest_asyncio
+stpyvista==0.0.11