Update app.py

app.py

@@ -100,17 +100,21 @@ def simulation(input_noise, output_noise, min, max, std, mean, min_pul, max_pul,
     fmu.freeInstance()
     shutil.rmtree(unzipdir, ignore_errors=True)
     result = np.array(rows, dtype=np.dtype([('time', np.float64), ('BIS', np.float64), ('Infusion', np.float64)]))
+    result_baseline = np.load("result.npy")
     df = pd.DataFrame(result)  
+    df_original = pd.DataFrame(result_baseline)
     trace1 = go.Scatter(x=df.index, y=df['BIS'], mode='lines', name='BIS')
     fig1 = go.Figure(data=trace1)
     fig1.update_layout(height=400, width=1200, title_text="BIS evolution")
+    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),)
 
     # Add a line trace for column_2 in the second subplot
     trace2 = go.Scatter(x=df.index, y=df['Infusion'], mode='lines', name='Infusion')
     fig2 = go.Figure(data=trace2)
     fig2.update_layout(height=400, width=1200, title_text="Infusion evolution")
+    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))
 
-    return fig1, fig2  
+    return fig1, fig2    
 import gradio as gr 
 with gr.Blocks() as demo:
     with gr.Row():