Add projection to plot

app.py

@@ -198,14 +198,20 @@ def plot_expression(arousal, dominance, valence):
         & (y == round(dominance * voxels))
         & (z == round(valence * voxels))
     )
-    colors = np.empty(voxel.shape, dtype=object)
+    projection = (
+        (x == round(arousal * voxels))
+        & (y == round(dominance * voxels))
+        & (z < round(valence * voxels))
+    )
+    colors = np.empty((voxel | projection).shape, dtype=object)
     colors[voxel] = "#fcb06c"
+    colors[projection] = "#fed7a9"
     ax = plt.figure().add_subplot(projection='3d')
-    ax.voxels(voxel, facecolors=colors, edgecolor='k')
-    ax.set_aspect("equal")
+    ax.voxels(voxel | projection, facecolors=colors, edgecolor='k')
     ax.set_xlim([0, voxels])
     ax.set_ylim([0, voxels])
     ax.set_zlim([0, voxels])
+    ax.set_aspect("equal")
     ax.set_xlabel("arousal", fontsize="large", labelpad=0)
     ax.set_ylabel("dominance", fontsize="large", labelpad=0)
     ax.set_zlabel("valence", fontsize="large", labelpad=0)