fix: df_right from state loaded

app.py

@@ -78,12 +78,12 @@ with col2:
         df_right = st.session_state.df.copy(deep=True)
         if "baseline" in list(df_right.keys()):
             st.markdown(f"A **baseline** feature has been found on your data, do you want to merge it with any of the other features in a new calculated field?")
-            option = st.multiselect('Select a feature to merge', [k for k in list(df_right.keys()) if k != 'baseline'], [[k for k in list(df_right.keys()) if k != 'baseline'][-3]])
+            option = st.multiselect('Select a feature to merge', [k for k in list(df_right.keys()) if k != 'baseline'], [[k for k in list(df_right.keys()) if k != 'baseline'][-4]])
             relative_key = f"relative_{option[0]}"
             add_relative = st.button(f"Add {relative_key}")
             if add_relative:
-                baseline_mean = [sum(s)/len(s) for s in df['baseline']]
-                df_right['relative_pupil_dilation'] = [df[option[0]][i] - baseline_mean[i] for i in range(len(df))]
+                baseline_mean = [sum(s)/len(s) for s in df_right['baseline']]
+                df_right['relative_pupil_dilation'] = [df_right[option[0]][i] - baseline_mean[i] for i in range(len(df_right))]
                 st.markdown("After adding calculated fields")
                 st.dataframe(df_right)
                 with open('myfile.csv') as f: