last nits with session management

yourbench_space/app.py

@@ -42,9 +42,10 @@ citation_content = (
     else "# Citation\n\nDocumentation file not found."
 )
 
-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)
+def generate_and_return(hf_org, hf_dataset_name, session_state: gr.State):
+    manager = MANAGERS.get(session_state)
+    session_uid = session_state.value
+    config_path = generate_and_save_config(hf_org, hf_dataset_name, session_uid, manager.config_path)
     for _ in range(5):
         time.sleep(0.5)
         if config_path.exists():
@@ -59,9 +60,11 @@ def generate_and_return(hf_org, hf_dataset_name, uuid: str):
 
 final_dataset = None
 
-def update_process_status(uuid: str):
+def update_process_status(session_state: gr.State):
     """Update process status and include exit details if process has terminated"""
-    manager = MANAGERS.get(uuid)
+    if session_state is None:
+        return gr.update(value=False, label="Not running")
+    manager = MANAGERS.get(session_state.value)
 
     is_running = manager.is_running()
     
@@ -72,12 +75,12 @@ def update_process_status(uuid: str):
     
     return gr.update(value=True, label="Process Status: Running")
 
-def prepare_task(uuid: str, oauth_token: gr.OAuthToken | None, hf_dataset_name: str, _=None):
+def prepare_task(session_uid: str, oauth_token: gr.OAuthToken | None, hf_dataset_name: str, _=None):
     new_env = os.environ.copy()
     if oauth_token:
         new_env["HF_TOKEN"] = oauth_token.token
     new_env["DATASET_PREFIX"] = hf_dataset_name
-    MANAGERS.start_process(uuid, custom_env=new_env)
+    MANAGERS.start_process(session_uid, custom_env=new_env)
 
 
 def update_hf_org_dropdown(oauth_token: gr.OAuthToken | None):

yourbench_space/utils.py

@@ -118,6 +118,8 @@ class SubprocessManagerGroup:
         self.managers[uid].kill_process()
 
     def read_and_get_output(self, uid: Union[str, gr.State]):
+        if uid is None:
+            return "", []
         uid = SubprocessManagerGroup.grab_uuid(uid)
         return self.managers[uid].read_and_get_output()
 
@@ -125,9 +127,9 @@ class SubprocessManagerGroup:
 class SubprocessManager:
     def __init__(self, session_uid: str):
         self.session_uid = session_uid
-        self.path = pathlib.Path(f"/app/{session_uid}")
+        self.path = pathlib.Path(f"app/{session_uid}")
         self.path.mkdir(parents=True, exist_ok=True)
-        self.config_path = pathlib.Path(f"/app/{session_uid}/config.yml")
+        self.config_path = pathlib.Path(f"{self.path}/config.yml")
         self.command = ["uv", "run", "yourbench", f"--config", str(self.config_path)]
         self.process = None
         self.output_stream = io.StringIO()