Update app.py

app.py

@@ -114,9 +114,9 @@ def simulation(input_noise, output_noise, min, max, std, mean, min_pul, max_pul,
     fig1.update_layout(height=400, width=1200, title_text="BIS evolution")
 
     # Add a line trace for column_2 in the second subplot
-    trace2 = go.Scatter(x=df.index, y=df['Infusion'], mode='lines', name='Infusion')
+    trace2 = go.Scatter(x=df.index, y=df['Infusion'], mode='lines', name='Infusion rate')
     fig2 = go.Figure(data=trace2)
-    fig2.update_layout(height=400, width=1200, title_text="Infusion evolution")
+    fig2.update_layout(height=400, width=1200, title_text="Infusion rate evolution")
     if show_original:
         if input_noise:
             result_baseline = np.load("result_gaussian.npy")
@@ -124,7 +124,7 @@ def simulation(input_noise, output_noise, min, max, std, mean, min_pul, max_pul,
             result_baseline = np.load("result_pulsive.npy")
         df_original = pd.DataFrame(result_baseline)
         fig1.add_trace(go.Scatter(x=df_original.index, y=df_original['BIS'], mode='lines', name='BIS original', line=dict(color="red"), opacity=0.5))
-        fig2.add_trace(go.Scatter(x=df_original.index, y=df_original['Infusion'], mode='lines', name='Infusion original',  line=dict(color="red"), opacity=0.5))
+        fig2.add_trace(go.Scatter(x=df_original.index, y=df_original['Infusion'], mode='lines', name='Infusion rate original',  line=dict(color="red"), opacity=0.5))
 
     return fig1, fig2    
 import gradio as gr