Update app/main_IO.py

app/main_IO.py

@@ -45,6 +45,23 @@ DEFAULT_SESSION_STATE = {
 }
 
 
+def initialise_session_state():
+    """
+    Initializes session state variables on a new session.
+    """
+    if 'session_initialized' not in st.session_state:
+        # New session detected
+        for key, default_val in DEFAULT_SESSION_STATE.items():
+            st.session_state[key] = default_val
+
+        # Clear questions_to_download on session start
+        st.session_state['questions_to_download'] = {}
+
+        # Mark session as initialized
+        st.session_state['session_initialized'] = True
+
+
+"""
 def initialise_session_state():
     """
     Initializes the session state variables if not already set.
@@ -52,6 +69,7 @@ def initialise_session_state():
     for key, default_val in DEFAULT_SESSION_STATE.items():
         if key not in st.session_state:
             st.session_state[key] = default_val
+"""
 
 
 def reset_session_state_on_upload():