Update app.py

app.py

@@ -1053,9 +1053,7 @@ class torus:
                 coin[2, iC, iF] = self.Z[self.P[iF, iC] - 1]
         return coin
 
-
-
-def make_voxels(shape, length, height, width, diameter):
+def make_voxels_without_figure(shape, length, height, width, diameter):
     pos = [0, 0, 0]
     if shape == "box":
         mesh = box(length, width, height, pos)
@@ -1081,7 +1079,11 @@ def make_voxels(shape, length, height, width, diameter):
     hull = scipy.spatial.Delaunay(hull_points)
     within = hull.find_simplex(test_points) >= 0
 
-    return plotly_fig(within*1.0)
+    return within*1.0
+    
+    
+def make_voxels(shape, length, height, width, diameter):
+    return plotly_fig(make_voxels_without_figure(shape, length, height, width, diameter))
 
 # This function loads a fuckton of data
 def load_data():
@@ -1267,7 +1269,7 @@ def geometry(index):
 def simple_analysis(index, choice, shape, length, width, height, diameter): 
     forward_net = Network("16forward_structure.json", "16forward_weights.h5")
     if choice == "Construct Shape from Parameters":  
-        return forward_net.analysis_from_geometry(make_voxels(shape, length, height, width, diameter))
+        return forward_net.analysis_from_geometry(make_voxels_without_figure(shape, length, height, width, diameter))
     elif choice == "Pick Shape from Dataset":
         return forward_net.analysis(index)