exception handling

app.py

@@ -8,6 +8,7 @@ import numpy as np
 import numpy.typing as npt
 from dcgan import DCGAN3D_G
 import pathlib
+import time
 pv.start_xvfb()
 
 
@@ -130,32 +131,35 @@ def main():
     slices, mesh, dist = create_uniform_mesh_marching_cubes(img)
 
     pv.set_plot_theme("document")
-    pl = pv.Plotter(shape=(1, 1),
+    pl1 = pv.Plotter(shape=(1, 1),
                          window_size=(view_width, view_height))
-    _ = pl.add_mesh(slices, cmap="gray")
-
+    _ = pl1.add_mesh(slices, cmap="gray")
     slices_html = DummyWriteable()
-    pl.export_html(slices_html)
+    try:
+        pl1.export_html(slices_html)
+    except RuntimeError as e:
+        print(e)
 
-    pl = pv.Plotter(shape=(1, 1),
+    pl2 = pv.Plotter(shape=(1, 1),
                          window_size=(view_width, view_height))
-    _ = pl.add_mesh(mesh, scalars=dist)
+    _ = pl2.add_mesh(mesh, scalars=dist)
     mesh_html = DummyWriteable()
-    pl.export_html(mesh_html)
+    try:
+        pl2.export_html(mesh_html)
+    except RuntimeError as e:
+        print(e)
 
     st.header("2D Cross-Section of Generated Volume")
     fig = create_matplotlib_figure(img, img.shape[0]//2)
     st.pyplot(fig=fig)
 
-    source_code = slices_html.html
     st.header("3D Intersections")
-    components.html(source_code, width=view_width, height=view_height)
+    components.html(slices_html.html, width=view_width, height=view_height)
     st.markdown("_Click and drag to spin, right click to shift._")
 
 
-    source_code = mesh_html.html
     st.header("3D Pore Space Mesh")
-    components.html(source_code, width=view_width, height=view_height)
+    components.html(mesh_html.html, width=view_width, height=view_height)
     st.markdown("_Click and drag to spin, right click to shift._")
 
     st.markdown("""