feat: fix bug

web_server.py

@@ -294,6 +294,17 @@ class EnvHandlerManager:
         os.makedirs(output, exist_ok=True)
         return EnvHandler(cfg, output)
 
+    def exists_env_handler(self, env_id: str) -> bool:
+        """
+        Check if the environment handler for the given environment ID exists.
+        Args:
+            env_id (str): The environment ID.
+        Returns:
+            bool: True if the environment handler exists, False otherwise.
+        """
+        with self._lock:
+            return env_id in self._env_handlers
+
     def get_env_handler(self, env_id: str) -> EnvHandler:
         """
         Get the environment handler for the given environment ID.
@@ -338,7 +349,12 @@ def _get_env_handler(
         token_info = get_token_info(token)
     except Exception:
         raise HTTPException(status_code=401)
-    env_handler = env_manager.get_env_handler(token_info["submission_id"])
+    
+    submission_id = token_info["submission_id"]
+    if not env_manager.exists_env_handler(submission_id):
+        update_submission_status(token_info["team_id"], submission_id, SubmissionStatus.PROCESSING.value)
+
+    env_handler = env_manager.get_env_handler(submission_id)
     if env_handler is None:
         raise HTTPException(status_code=404, detail="Environment handler already closed.")
     return env_handler