feat: update score

web_server.py

@@ -118,11 +118,11 @@ def upload_submission_info(team_id: str, user_submission_info: Dict[str, Any]):
     )
 
 
-def update_submission_status(team_id: str, submission_id: str, status: int):
+def update_submission_data(team_id: str, submission_id: str, data: Dict[str, Any]):
     user_submission_info = download_submission_info(team_id)
     for submission in user_submission_info["submissions"]:
         if submission["submission_id"] == submission_id:
-            submission["status"] = status
+            submission.update(data)
             break
     upload_submission_info(team_id, user_submission_info)
 
@@ -314,6 +314,26 @@ class EnvHandler:
         with self._lock:
             self._log_list.append(log_message)
 
+    def calculate_score(self) -> Dict[str, Any]:
+        """
+        Calculate the score based on the current environment state.
+        Returns:
+            Dict[str, Any]: The score dictionary.
+        """
+        if not self._done:
+            raise ValueError("Environment is not done yet. Cannot calculate score.")
+        
+        all_score_json = []
+        for scene_config in self.scene_list:
+            scene_output = os.path.join(self.base_output, scene_config.name)
+            with open(os.path.join(scene_output, 'eval.json'), 'r') as f:
+                score_json = json.load(f)
+            all_score_json.append(score_json)
+
+        rc = np.mean([score['rc'] for score in all_score_json]).round(4)
+        hdscore = np.mean([score['hdscore'] for score in all_score_json]).round(4)
+        return {"rc": rc, "hdscore": hdscore}
+
 
 class EnvHandlerManager:
     def __init__(self):
@@ -417,7 +437,7 @@ def _get_env_handler(
     
     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)
+        update_submission_data(token_info["team_id"], submission_id, {"status": SubmissionStatus.PROCESSING.value})
 
     env_handler = env_manager.get_env_handler(submission_id)
     if env_handler is None:
@@ -487,7 +507,8 @@ def execute_action_endpoint(
         token_info = get_token_info(auth_token)
         env_manager.close_env_handler(token_info["submission_id"])
         delete_client_space(token_info["client_space_id"])
-        update_submission_status(token_info["team_id"], token_info["submission_id"], SubmissionStatus.SUCCESS.value)
+        final_score = env_handler.calculate_score()
+        update_submission_data(token_info["team_id"], token_info["submission_id"], {"status": SubmissionStatus.SUCCESS.value, "score": final_score})
         hf_api.upload_folder(
             repo_id=COMPETITION_ID,
             folder_path=env_handler.base_output,