animation changed to animate and video save

.ipynb_checkpoints/app-checkpoint.py

@@ -460,7 +460,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
     plt.ylim((0,220))
     plt.xlim((0,250))
     start = (N-2)*60000
-    end = (N-2)*60000+50000
+    end = (N)*60000
     if animate:
         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)
@@ -484,7 +484,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
         ax.plot(x, y, lw=1, c='b')
         
     if animate:
-        anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
+        anim = animation.FuncAnimation(fig, partial(update), frames=60, interval=1)
         anim.save("prediction.mp4")
         
     return plt, Rm, Ra, Emax, Emin, Vd, Tc, start_v
@@ -514,8 +514,8 @@ def generate_example():
 
         plot, Rm, Ra, Emax, Emin, Vd,Tc, start_v = pvloop_simulator(Rm=round(results[4].item(),2), Ra=round(results[5].item(),2), Emax=results[2].item(), Emin=round(results[3].item(),2), Vd=round(results[6].item(),2), Tc=round(results[0].item(),2), start_v=round(results[1].item(),2))
         video = video.replace("avi", "mp4")
-        # animated = "prediction.mp4"
-        return video, plot, Rm, Ra, Emax, Emin, Vd, Tc, start_v
+        animated = "prediction.mp4"
+        return video, animated, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 title = "Non-Invasive Medical Digital Twins using Physics-Informed Self-Supervised Learning"
 

app.py

@@ -460,7 +460,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
     plt.ylim((0,220))
     plt.xlim((0,250))
     start = (N-2)*60000
-    end = (N-2)*60000+50000
+    end = (N)*60000
     if animate:
         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)
@@ -484,7 +484,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
         ax.plot(x, y, lw=1, c='b')
         
     if animate:
-        anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
+        anim = animation.FuncAnimation(fig, partial(update), frames=60, interval=1)
         anim.save("prediction.mp4")
         
     return plt, Rm, Ra, Emax, Emin, Vd, Tc, start_v
@@ -514,8 +514,8 @@ def generate_example():
 
         plot, Rm, Ra, Emax, Emin, Vd,Tc, start_v = pvloop_simulator(Rm=round(results[4].item(),2), Ra=round(results[5].item(),2), Emax=results[2].item(), Emin=round(results[3].item(),2), Vd=round(results[6].item(),2), Tc=round(results[0].item(),2), start_v=round(results[1].item(),2))
         video = video.replace("avi", "mp4")
-        # animated = "prediction.mp4"
-        return video, plot, Rm, Ra, Emax, Emin, Vd, Tc, start_v
+        animated = "prediction.mp4"
+        return video, animated, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 title = "Non-Invasive Medical Digital Twins using Physics-Informed Self-Supervised Learning"