Update app.py

app.py

@@ -431,7 +431,7 @@ def solve_ODE_for_volume(Rm, Ra, Emax, Emin, Vd, Tc, start_v, t):
     
     return volumes
 
-def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
+def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True, loop_simulated=False):
     
 
     # Define initial parameters
@@ -491,13 +491,18 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True):
     if animate:
         anim = animation.FuncAnimation(fig, partial(update), frames=100, interval=30)
         anim.save("prediction.mp4")
+
+    if loop_simulated:
+        plt.title('Simulated LV Pressure Volume Loop', fontsize=16)
+        anim = animation.FuncAnimation(fig, partial(update), frames=100, interval=30)
+        anim.save("simulated.mp4")
         
     return plt, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 def pvloop_simulator_plot_only(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
-    plot,_,_,_,_,_,_,_ =pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=True)
-    plt.title('Simulated LV Pressure Volume Loop', fontsize=16)
-    return plot
+    plot,_,_,_,_,_,_,_ =pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animate=False, loop_simulated=True)
+    animated_sim = "simulated.mp4"
+    return animated_sim
 
 #########################################
 # LVAD functions