more attempting

.ipynb_checkpoints/app-checkpoint.py

@@ -427,7 +427,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):
+def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation):
     
 
     # Define initial parameters
@@ -465,7 +465,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
     point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
 
     
-    fig.suptitle('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
+    plt.title('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
     #plt.rcParams['fig.suptitle'] = -2.0
     #ax.set_title(f'Mitral valve circuit resistance (Rm): {Rm} mmHg*s/ml \n Aortic valve circuit resistance (Ra): {Ra} mmHg*s/ml', fontsize=6)
     ax.set_xlabel('LV Volume (ml)')
@@ -480,14 +480,16 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
         x = volumes[start:end]
         y = pressures[start:end]
         ax.plot(x, y, lw=1, c='b')
-
-    anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
-    anim.save("prediction.mp4")
+        
+    if(animation):
+        anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
+        anim.save("prediction.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)
-    fig.suptitle('Simulated PI-SSL LV Pressure Volume Loop', fontsize=16)
+    plot,_,_,_,_,_,_,_ =pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation=False)
+    plt.title('Simulated PI-SSL LV Pressure Volume Loop', fontsize=16)
     return plot
 
 ## Demo 
@@ -510,8 +512,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, animated, Rm, Ra, Emax, Emin, Vd, Tc, start_v
+        # animated = "prediction.mp4"
+        return video, plot, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 title = "Physics-informed self-supervised learning for predicting cardiac digital twins with echocardiography"
 

app.py

@@ -427,7 +427,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):
+def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation):
     
 
     # Define initial parameters
@@ -465,7 +465,7 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
     point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
 
     
-    fig.suptitle('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
+    plt.title('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
     #plt.rcParams['fig.suptitle'] = -2.0
     #ax.set_title(f'Mitral valve circuit resistance (Rm): {Rm} mmHg*s/ml \n Aortic valve circuit resistance (Ra): {Ra} mmHg*s/ml', fontsize=6)
     ax.set_xlabel('LV Volume (ml)')
@@ -480,14 +480,16 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
         x = volumes[start:end]
         y = pressures[start:end]
         ax.plot(x, y, lw=1, c='b')
-
-    anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
-    anim.save("prediction.mp4")
+        
+    if(animation):
+        anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
+        anim.save("prediction.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)
-    fig.suptitle('Simulated PI-SSL LV Pressure Volume Loop', fontsize=16)
+    plot,_,_,_,_,_,_,_ =pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v, animation=False)
+    plt.title('Simulated PI-SSL LV Pressure Volume Loop', fontsize=16)
     return plot
 
 ## Demo 
@@ -510,8 +512,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, animated, Rm, Ra, Emax, Emin, Vd, Tc, start_v
+        # animated = "prediction.mp4"
+        return video, plot, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 title = "Physics-informed self-supervised learning for predicting cardiac digital twins with echocardiography"