updated for session state

multi_env_compare.py

@@ -45,6 +45,10 @@ def perform_multi_env_analysis(uploaded_dataframes):
     # Initialize session state for selected functional areas if it doesn't exist
     if 'selected_functional_areas' not in st.session_state:
         st.session_state.selected_functional_areas = ["All"]
+    
+    # Make sure functional_areas_multiselect is also initialized
+    if 'functional_areas_multiselect' not in st.session_state:
+        st.session_state.functional_areas_multiselect = st.session_state.selected_functional_areas
 
     # Select functional areas to display, using session state
     selected_functional_areas = st.multiselect(
@@ -61,7 +65,7 @@ def perform_multi_env_analysis(uploaded_dataframes):
         st.success("Functional area selection updated!")
         time.sleep(0.5)  # Add a small delay for better user experience
         st.rerun()  # Rerun the app to reflect the changes
-
+    
     if "All" in selected_functional_areas:
         selected_functional_areas = combined_data['Functional area'].unique()