more attempting

.ipynb_checkpoints/app-checkpoint.py

@@ -461,9 +461,11 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation):
     plt.xlim((0,250))
     start = (N-2)*60000
     end = (N-2)*60000+50000
-    line = ax.plot(volumes[start:(start+1)], pressures[start:(start+1)], lw=1, color='b')
-    point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
-
+    if animation:
+        line = ax.plot(volumes[start:(start+1)], pressures[start:(start+1)], lw=1, color='b')
+        point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
+    else:
+        line = ax.plot(volumes[start:end], pressures[start:end], lw=1, color='b')
     
     plt.title('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
     #plt.rcParams['fig.suptitle'] = -2.0
@@ -481,7 +483,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation):
         y = pressures[start:end]
         ax.plot(x, y, lw=1, c='b')
         
-    if(animation):
+    if animation:
         anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
         anim.save("prediction.mp4")
         

app.py

@@ -461,9 +461,11 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation):
     plt.xlim((0,250))
     start = (N-2)*60000
     end = (N-2)*60000+50000
-    line = ax.plot(volumes[start:(start+1)], pressures[start:(start+1)], lw=1, color='b')
-    point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
-
+    if animation:
+        line = ax.plot(volumes[start:(start+1)], pressures[start:(start+1)], lw=1, color='b')
+        point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
+    else:
+        line = ax.plot(volumes[start:end], pressures[start:end], lw=1, color='b')
     
     plt.title('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
     #plt.rcParams['fig.suptitle'] = -2.0
@@ -481,7 +483,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation):
         y = pressures[start:end]
         ax.plot(x, y, lw=1, c='b')
         
-    if(animation):
+    if animation:
         anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
         anim.save("prediction.mp4")