fix session state management

yourbench_space/app.py

@@ -42,9 +42,9 @@ citation_content = (
     else "# Citation\n\nDocumentation file not found."
 )
 
-def generate_and_return(hf_org, hf_dataset_name, session_state: gr.State):
-    manager = MANAGERS.get(session_state.value)
-    config_path = generate_and_save_config(hf_org, hf_dataset_name, session_state.value, manager.config_path)
+def generate_and_return(hf_org, hf_dataset_name, uuid: str):
+    manager = MANAGERS.get(uuid)
+    config_path = generate_and_save_config(hf_org, hf_dataset_name, uuid, manager.config_path)
     for _ in range(5):
         time.sleep(0.5)
         if config_path.exists():
@@ -59,9 +59,9 @@ def generate_and_return(hf_org, hf_dataset_name, session_state: gr.State):
 
 final_dataset = None
 
-def update_process_status(session_state: gr.State):
+def update_process_status(uuid: str):
     """Update process status and include exit details if process has terminated"""
-    manager = MANAGERS.get(session_state.value)
+    manager = MANAGERS.get(uuid)
 
     is_running = manager.is_running()
     
@@ -72,8 +72,8 @@ def update_process_status(session_state: gr.State):
     
     return gr.update(value=True, label="Process Status: Running")
 
-def prepare_task(session_state: gr.State, oauth_token: gr.OAuthToken | None, hf_dataset_name: str, _=None):
-    manager = MANAGERS.get(session_state.value)
+def prepare_task(uuid: str, oauth_token: gr.OAuthToken | None, hf_dataset_name: str, _=None):
+    manager = MANAGERS.get(uuid)
     new_env = os.environ.copy()
     if oauth_token:
         new_env["HF_TOKEN"] = oauth_token.token
@@ -134,7 +134,7 @@ def run_evaluation_pipeline(oauth_token: gr.OAuthToken | None, org_name, eval_na
 with gr.Blocks(theme=gr.themes.Default()) as app:
     # We initialize the session state with the user randomly generated uuid
     # Using uuid4 makes collision cases extremely unlikely even for concurrent users
-    session_state = gr.State(uuid.uuid4(), delete_callback=lambda uid: MANAGERS.remove(uid))
+    session_state = gr.State(str(uuid.uuid4()), delete_callback=lambda uid: MANAGERS.remove(uid))
     MANAGERS.create(session_state.value)
 
     gr.Markdown(project_description)

yourbench_space/utils.py

@@ -24,9 +24,8 @@ STAGES = [
 ]
 
 
-def save_files(session_state: gr.State, files: List[pathlib.Path]) -> str:
+def save_files(uuid: str, files: List[pathlib.Path]) -> str:
     """Save uploaded files to the UPLOAD_DIRECTORY/uuid safely"""
-    uuid = session_state.value
     saved_paths = []
 
     for file in files: