add EF, CO

.ipynb_checkpoints/app-checkpoint.py

@@ -805,12 +805,6 @@ def lvad_plots(Rm, Ra, Emax, Emin, Vd, Tc, start_v, gamma):
     #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)
       
-    print("\nParameters: Tc, start_v, Emax:", Tc, start_v, Emax)
-    print("Suction speed:", suctionw)
-    print("EF before LVAD:", ef_nolvad)
-    print("CO before LVAD:", co_nolvad)
-    print("New EF after LVAD:", new_ef, "New CO:", CO, "New MAP:", MAP, "\n\n")
-
     fig, ax = plt.subplots()
     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
@@ -821,7 +815,7 @@ def lvad_plots(Rm, Ra, Emax, Emin, Vd, Tc, start_v, gamma):
     plt.xlim((0,250))
     #plt.title('Simulated PI-SSL LV Pressure Volume Loop', fontsize=16)
     
-    return plt
+    return plt, ef_nolvad, new_ef, co_nolvad, CO
     
 title = "<h1 style='text-align: center; margin-bottom: 1rem'> Non-Invasive Medical Digital Twins using Physics-Informed Self-Supervised Learning </h1>"
 
@@ -897,11 +891,18 @@ with gr.Blocks() as demo:
 
             with gr.Row():
                 lvad = gr.Plot()
+
+            with gr.Row():
+                EF_o = gr.Number(label="Ejection fraction (EF) before LVAD:")
+                EF_n = gr.Number(label="Ejection fraction (EF) after LVAD:")
+                CO_o = gr.Number(label="Cardiac output before LVAD:")
+                CO_n = gr.Number(label="Cardiac output after LVAD:")
+                #MAP_n = gr.Number(label="Mean arterial pressure (MAP) after LVAD:")
     
     generate_button.click(fn=generate_example, outputs = [video,plot,Rm,Ra,Emax,Emin,Vd,Tc,start_v])
     
     simulation_button.click(fn=pvloop_simulator_plot_only, inputs = [sl1,sl2,sl3,sl4,sl5,sl6,sl7], outputs = [simulation])
     
-    LVAD_button.click(fn=lvad_plots, inputs = [sl1,sl2,sl3,sl4,sl5,sl6,sl7,gamma], outputs = [lvad])
+    LVAD_button.click(fn=lvad_plots, inputs = [sl1,sl2,sl3,sl4,sl5,sl6,sl7,gamma], outputs = [lvad, EF_o, EF_n, CO_o, CO_n])
     
 demo.launch()

app.py

@@ -805,12 +805,6 @@ def lvad_plots(Rm, Ra, Emax, Emin, Vd, Tc, start_v, gamma):
     #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)
       
-    print("\nParameters: Tc, start_v, Emax:", Tc, start_v, Emax)
-    print("Suction speed:", suctionw)
-    print("EF before LVAD:", ef_nolvad)
-    print("CO before LVAD:", co_nolvad)
-    print("New EF after LVAD:", new_ef, "New CO:", CO, "New MAP:", MAP, "\n\n")
-
     fig, ax = plt.subplots()
     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
@@ -821,7 +815,7 @@ def lvad_plots(Rm, Ra, Emax, Emin, Vd, Tc, start_v, gamma):
     plt.xlim((0,250))
     #plt.title('Simulated PI-SSL LV Pressure Volume Loop', fontsize=16)
     
-    return plt
+    return plt, ef_nolvad, new_ef, co_nolvad, CO
     
 title = "<h1 style='text-align: center; margin-bottom: 1rem'> Non-Invasive Medical Digital Twins using Physics-Informed Self-Supervised Learning </h1>"
 
@@ -897,11 +891,18 @@ with gr.Blocks() as demo:
 
             with gr.Row():
                 lvad = gr.Plot()
+
+            with gr.Row():
+                EF_o = gr.Number(label="Ejection fraction (EF) before LVAD:")
+                EF_n = gr.Number(label="Ejection fraction (EF) after LVAD:")
+                CO_o = gr.Number(label="Cardiac output before LVAD:")
+                CO_n = gr.Number(label="Cardiac output after LVAD:")
+                #MAP_n = gr.Number(label="Mean arterial pressure (MAP) after LVAD:")
     
     generate_button.click(fn=generate_example, outputs = [video,plot,Rm,Ra,Emax,Emin,Vd,Tc,start_v])
     
     simulation_button.click(fn=pvloop_simulator_plot_only, inputs = [sl1,sl2,sl3,sl4,sl5,sl6,sl7], outputs = [simulation])
     
-    LVAD_button.click(fn=lvad_plots, inputs = [sl1,sl2,sl3,sl4,sl5,sl6,sl7,gamma], outputs = [lvad])
+    LVAD_button.click(fn=lvad_plots, inputs = [sl1,sl2,sl3,sl4,sl5,sl6,sl7,gamma], outputs = [lvad, EF_o, EF_n, CO_o, CO_n])
     
 demo.launch()