Trying to make it load faster by pre-loading meshes

bootstrap.py

@@ -3,7 +3,7 @@ import streamlit.web.bootstrap
 from streamlit import config as _config
 
 proj_dir = Path(__file__).parent
-filename = proj_dir / "Disc Golf Simulator.py"
+filename = proj_dir / "disc_golf_simulator.py"
 
 _config.set_option("server.headless", True)
 args = []

disc_golf_simulator.py

@@ -4,7 +4,7 @@ from logging import getLogger
 from pathlib import Path
 
 from shotshaper.projectile import DiscGolfDisc
-from utilities.visualize import get_plot, get_stl, get_subplots, visualize_disc
+from utilities.visualize import get_plot, get_subplots, visualize_disc, stl_meshes
 
 # Define the default values
 default_U = 24.2
@@ -49,15 +49,16 @@ def main():
                                       value=default_pitch,
                                       step=0.1)
 
-        with st.spinner(text="Calculating Flight Path..."):
+        with st.spinner(text="Calculating Disc Orientation..."):
             pos = np.array((0, 0, z0))
             disc_dict = DiscGolfDisc(disc_name)
 
-            stl_mesh = get_stl(proj_dir / 'shotshaper' / 'discs' / (disc_name + '.stl'))
+            stl_mesh = stl_meshes[disc_name]
             fig = visualize_disc(stl_mesh, nose=nose, roll=roll)
 
             st.markdown("""## Disc Orientation""")
             st.plotly_chart(fig)
+        with st.spinner(text="Calculating Flight Path..."):
             st.markdown("""## Flight Path""")
             shot = disc_dict.shoot(speed=U, omega=omega, pitch=pitch,
                                    position=pos, nose_angle=nose, roll_angle=roll)

utilities/visualize.py

@@ -1,26 +1,23 @@
 import math
-
 import numpy as np
 import plotly.graph_objects as go
+from pathlib import Path
 from plotly.colors import sequential
 from stl.mesh import Mesh
 
 from .extrema import find_extrema
 
+proj_dir = Path(__file__).parents[1]
 
-def get_stl(stl_file):
-    """
-    Taken from https://community.plotly.com/t/view-3d-cad-data/16920/9
-    """
-    stl_mesh = Mesh.from_file(stl_file)
-    return stl_mesh
+# Taken from https://community.plotly.com/t/view-3d-cad-data/16920/9
+stl_meshes = {disc.stem: Mesh.from_file(disc) for disc in (proj_dir / 'shotshaper' / 'discs').glob('*.stl')}
 
 
 def visualize_disc(stl_mesh, nose, roll):
     """
     Taken from https://community.plotly.com/t/view-3d-cad-data/16920/9
     """
-    stl_mesh.rotate([1, 0, 0], math.radians(-1*nose))
+    stl_mesh.rotate([1, 0, 0], math.radians(-1 * nose))
     stl_mesh.rotate([0, 1, 0], math.radians(roll))
     # stl_mesh.rotate([0, 0, 1], math.radians(z_angle))