attempt saving gif

.ipynb_checkpoints/app-checkpoint.py

@@ -482,11 +482,12 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
         ax.plot(x, y, lw=1, c='b')
 
     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)
     return plot
 
 ## Demo 
@@ -509,7 +510,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")
-        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 = "Physics-informed self-supervised learning for predicting cardiac digital twins with echocardiography"
 
@@ -535,7 +537,7 @@ with gr.Blocks() as demo:
             generate_button = gr.Button("Load sample echocardiogram and generate result")
             with gr.Row():
                 video = gr.PlayableVideo() #format="avi"
-                plot = gr.Plot()
+                plot = gr.PlayableVideo()
             
             with gr.Row():
                 Rm = gr.Number(label="Mitral valve circuit resistance (Rm) mmHg*s/ml:")

app.py

@@ -482,11 +482,12 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
         ax.plot(x, y, lw=1, c='b')
 
     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)
     return plot
 
 ## Demo 
@@ -509,7 +510,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")
-        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 = "Physics-informed self-supervised learning for predicting cardiac digital twins with echocardiography"
 
@@ -535,7 +537,7 @@ with gr.Blocks() as demo:
             generate_button = gr.Button("Load sample echocardiogram and generate result")
             with gr.Row():
                 video = gr.PlayableVideo() #format="avi"
-                plot = gr.Plot()
+                plot = gr.PlayableVideo()
             
             with gr.Row():
                 Rm = gr.Number(label="Mitral valve circuit resistance (Rm) mmHg*s/ml:")