obj

app.py

@@ -1,37 +1,10 @@
-import shutil
 import streamlit as st
-import stl
-from mpl_toolkits import mplot3d
-from matplotlib import pyplot
-from mpl_toolkits.mplot3d import axes3d, Axes3D
-
-
-file = st.file_uploader("Please upload an STL Object file", type=["STL"])
-
-if file is not None:
-    # save the uploaded file to disk
-    with open("stlfile.stl", "wb") as buffer:
-        shutil.copyfileobj(file, buffer)
-
-    # Create a new plot
-    figure = pyplot.figure()
-    axes = figure.add_subplot(projection='3d')
-
-    # Load the STL files and add the vectors to the plot
-    
-    mesh = stl.mesh.Mesh.from_file("stlfile.stl")
-
-    axes.add_collection3d(mplot3d.art3d.Poly3DCollection(mesh.vectors, color='lightgrey'))
-    #axes.plot_surface(mesh.x,mesh.y,mesh.z)
-    # Auto scale to the mesh size
-    scale = mesh.points.flatten()
-    axes.auto_scale_xyz(scale, scale, scale)
-
-    #turn off grid and axis from display      
-    pyplot.axis('off')
-
-    #set viewing angle
-    axes.view_init(azim=120)
-
-    # Show the plot to the screen
-    st.pyplot(figure)
+import streamlit.components.v1 as components
+from obj2html import obj2html
+
+# run in the root folder, not inside exaples, to have the relative path working
+obj_path = "obj1.obj"
+html_string = obj2html(obj_path, html_elements_only=True)
+components.html(html_string)
+with open(obj_path) as f:
+    st.download_button('Download model.obj', f, file_name="model.obj")

requirements.txt

@@ -1,5 +1,5 @@
 pyvista
 stpyvista
 matplotlib
-stl
+obj2html
 numpy-stl