try to animate lvad

app.py

@@ -1,6 +1,6 @@
 import os
-# os.system('pip uninstall -y gradio')
-# os.system('pip install gradio==5.0.1')
+os.system('pip uninstall -y gradio')
+os.system('pip install gradio==5.0.1')
 
 
 import gradio as gr
@@ -800,7 +800,7 @@ def generate_example():
         return video, animated, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
 
-def lvad_plots(Rm, Ra, Emax, Emin, Vd, Tc, start_v, beta):
+def lvad_plots(Rm, Ra, Emax, Emin, Vd, Tc, start_v, beta, loop_simulated=True):
 
     ncycle = 10000
     
@@ -829,18 +829,50 @@ def lvad_plots(Rm, Ra, Emax, Emin, Vd, Tc, start_v, beta):
     
     #compute new pv loops and ef with lvad added:
     new_ef, pao_ed, pao_es, CO, MAP, Vlvs, Plvs = f_lvad(Rs, Rm, Ra, Rc, Ca, Cs, Cr, Ls, Emin, Vd, Tc, start_v, Emax, c, slope0, w0, x60, y00, y01, y02, y03, y04)
-      
+
+    # Create the figure and the loop that we will manipulate
     fig, ax = plt.subplots(figsize=(6, 4))
-    ax.plot(Vlv0, Plv0, color='blue', label='No LVAD') #blue
-    ax.plot(Vlvs, Plvs, color=(78/255, 192/255, 44/255), label=f"LVAD, ω(0)= {round(w0,2)}r/min") #green
-    plt.xlabel('LV volume (ml)')
-    plt.ylabel('LV pressure (mmHg)')
+    plt.ylim((0,220))
+    plt.xlim((0,250))
+    start = (N-2)*60000
+    end = (N)*60000
+    if loop_simulated:
+        line1 = ax.plot(Vlv0[start:(start+1)], Plv0[start:(start+1)], lw=1, color='b',label='No LVAD')
+        point1 = ax.scatter(Vlv0[start:(start+1)], Plv0[start:(start+1)], c="b", s=5)#, label='End Diastole')
+        line2 = ax.plot(Vlvs[start:(start+1)], Plvs[start:(start+1)], color=(78/255, 192/255, 44/255), label=f"LVAD, ω(0)= {round(w0,2)}r/min")
+        point2 = ax.scatter(Vlvs[start:(start+1)], Plvs[start:(start+1)], c="b", s=5)#, label='End Diastole')
+        #point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5, label='End Systole')
+    else:
+        line = ax.plot(volumes[start:end], pressures[start:end], lw=1, color='b')
+        line1 = ax.plot(Vlv0, Plv0, color='blue', label='No LVAD') #blue
+        line2 = ax.plot(Vlvs, Plvs, color=(78/255, 192/255, 44/255), label=f"LVAD, ω(0)= {round(w0,2)}r/min") #green
+
+    ax.set_xlabel('LV Volume (ml)')
+    ax.set_ylabel('LV Pressure (mmHg)')
+
+    # adjust the main plot to make room for the sliders
+    # fig.subplots_adjust(left=0.25, bottom=0.25)
+
+    def update(frame):
+        # update to add more of the loop
+        end = (N-2)*60000+1000 * frame
+        x = volumes[start:end]
+        y = pressures[start:end]
+        ax.plot(x, y, lw=1, c='b')
+
+
     plt.legend(loc='upper left', framealpha=1)
     plt.ylim((0,220))
     plt.xlim((0,250))
-    #plt.title('Simulated LV Pressure Volume Loop', fontsize=16)
-    
-    return plt, round(ef_nolvad,2), round(new_ef,2), round(co_nolvad,2), round(CO, 2)
+
+    if loop_simulated:
+        # plt.title('', fontsize=16)
+        anim = animation.FuncAnimation(fig, partial(update), frames=100, interval=30)
+        anim.save("simulated_lvad.mp4")
+        anim_plot = "simulated_lvad.mp4"
+        return anim_plot, round(ef_nolvad,2), round(new_ef,2), round(co_nolvad,2), round(CO, 2)
+    else:
+        return plt, round(ef_nolvad,2), round(new_ef,2), round(co_nolvad,2), round(CO, 2)
     
 title = "<h1 style='text-align: center; margin-bottom: 1rem'> Med-Real2Sim: Non-Invasive Medical Digital Twins using Physics-Informed Self-Supervised Learning </h1>"