add reload labels and fix bug

Author: LokiWager

src/audiokit/refinement/refinement.py

@@ -38,16 +38,22 @@ def _save_file(file_path: str, labels: dict[str, Labeling]):
                 f.write(f"{value.source_file_path}|{value.language}|{value.text_content}\n")
 
     def submit_text(self, source_file_path: str, language: str, text_content: str):
+        """submit text to refinements.list"""
         self.source_file_content[source_file_path] = Labeling(source_file_path, language, text_content)
         self._save_file(self.output_file_path, self.source_file_content)
 
     def load_text(self) -> dict[str, Labeling]:
+        """load current source content from refinements.list"""
+        self.source_file_content = self._load_file(self.output_file_path)
+        return self.source_file_content
+
+    def reload_text(self):
+        """reload source file content from asr.list"""
         self.source_file_content = self._load_file(self.source_file_path)
+        self._save_file(self.output_file_path, self.source_file_content)
         return self.source_file_content
 
     def delete_text(self, source_file_path: str):
+        """delete source file content from refinements.list"""
         self.source_file_content.pop(source_file_path)
         self._save_file(self.output_file_path, self.source_file_content)
-
-    def save_file(self):
-        self._save_file(self.output_file_path, self.source_file_content)

src/rest/rest.py

@@ -21,7 +21,8 @@
 )
 from src.logger import logger
 from src.rest.types import TaskType, TaskCMD
-from src.service.audio import AudioUVR5Params, AudioSlicerParams, AudioASRParams, AudioService, AudioDenoiseParams, AudioRefinementSubmitParams, AudioRefinementDeleteParams
+from src.service.audio import AudioUVR5Params, AudioSlicerParams, AudioASRParams, AudioService, AudioDenoiseParams, AudioRefinementSubmitParams, AudioRefinementDeleteParams, \
+    AudioRefinementReloadParams
 from src.service.file import FileService
 from src.service.namespace import NamespaceService
 from src.service.normalize import NormalizeParams
@@ -60,6 +61,7 @@ async def serve_asset(self, file_path: str) -> Response:
         # Return the file
         return FileResponse(asset_path)
 
+
 class FrontendIndexAPI:
     """Class to handle serving index.html for the root path."""
 
@@ -72,7 +74,7 @@ def _register_routes(self):
         """Register routes to serve index.html."""
         # Serve index.html for any route that is not `/assets`
         self.router.add_api_route("/", self.serve_index, methods=["GET"])
-        #self.router.add_api_route("/{path:path}", self.serve_index, methods=["GET"])
+        # self.router.add_api_route("/{path:path}", self.serve_index, methods=["GET"])
 
     async def serve_index(self, path: str = "") -> Response:
         """Serve index.html for any request (except /assets)."""
@@ -469,6 +471,7 @@ def _register_routes(self):
         self.router.get("/audio/refinement")(self.list_audio_refinement)
         self.router.post("/audio/refinement")(self.update_audio_refinement)
         self.router.delete("/audio/refinement")(self.delete_audio_refinement)
+        self.router.post("/audio/refinement/reload")(self.reload_audio_refinement)
 
     async def audio_uvr5(self, request: AudioUVR5Params):
         if session_manager.exist_running_session():
@@ -532,7 +535,7 @@ async def audio_asr_stop(self, uid: str):
 
     def list_audio_refinement(self, input_dir: str, output_dir: str):
         service = AudioService(source_dir=input_dir, output_dir=output_dir)
-        result = service.refinement_reload()
+        result = service.refinement_load_source()
         if isinstance(result, EaseVoiceResponse):
             return result
         logger.error(f"failed to list audio refinement: {result}")
@@ -554,6 +557,14 @@ def delete_audio_refinement(self, params: AudioRefinementDeleteParams):
         logger.error(f"failed to delete audio refinement: {result}")
         raise HTTPException(status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail=result)
 
+    def reload_audio_refinement(self, params: AudioRefinementReloadParams):
+        service = AudioService(source_dir=params.source_dir, output_dir=params.output_dir)
+        result = service.refinement_reload_source()
+        if isinstance(result, EaseVoiceResponse):
+            return result
+        logger.error(f"failed to reload audio refinement: {result}")
+        raise HTTPException(status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail=result)
+
 
 class EaseVoiceAPI:
     def __init__(self):
@@ -608,7 +619,6 @@ async def lifespan_context(app: FastAPI) -> AsyncGenerator[None, None]:
 frontend_index_api = FrontendIndexAPI(frontend_dir)
 app.include_router(frontend_index_api.router)
 
-
 tensorboard_service = TensorBoardService()
 tensorboard_api = TensorBoardAPI()
 app.include_router(tensorboard_api.router)
@@ -639,7 +649,6 @@ async def lifespan_context(app: FastAPI) -> AsyncGenerator[None, None]:
 easevoice_api = EaseVoiceAPI()
 app.include_router(easevoice_api.router, prefix="/apis/v1")
 
-
 # Function to print all routing information
 # def print_routes(app: FastAPI):
 #     for route in app.routes:
@@ -649,4 +658,4 @@ async def lifespan_context(app: FastAPI) -> AsyncGenerator[None, None]:
 #         print("-" * 40)
 #
 # # Call the function to print all routes
-# print_routes(app)
+# print_routes(app)

src/service/audio.py

@@ -71,6 +71,11 @@ class AudioRefinementDeleteParams:
     output_dir: str
     source_file_path: str
 
+@dataclass
+class AudioRefinementReloadParams:
+    source_dir: str
+    output_dir: str
+
 
 class AudioService(object):
     def __init__(self, source_dir: str, output_dir: str):
@@ -177,7 +182,6 @@ def slicer(
                 EaseVoiceResponse(ResponseStatus.FAILED, "Slice Failed", {"file_name": name})
         return EaseVoiceResponse(ResponseStatus.SUCCESS, "Slice Success", data)
 
-
     def denoise(self) -> EaseVoiceResponse:
         os.makedirs(os.path.join(self.output_dir, denoises_output), exist_ok=True)
         trace_data = {}
@@ -221,22 +225,26 @@ def _get_files(self, output_path: str):
                 files.append(file_path)
         return files
 
+    def refinement_reload_source(self) -> EaseVoiceResponse:
+        try:
+            resp = self.refinement.reload_text()
+            return EaseVoiceResponse(ResponseStatus.SUCCESS, "Reload Source Success", resp)
+        except Exception as e:
+            return EaseVoiceResponse(ResponseStatus.FAILED, "Reload Source Failed", {
+                "error": str(e)
+            })
+
     def refinement_load_source(self) -> EaseVoiceResponse:
         os.makedirs(os.path.join(self.output_dir, refinements_output), exist_ok=True)
         if len(self.refinement.source_file_content) == 0:
             self.refinement.load_text()
-            self.refinement.save_file()
         resp = self.refinement.source_file_content
         return EaseVoiceResponse(ResponseStatus.SUCCESS, "Load Source Success", resp)
 
     def refinement_submit_text(self, index: str, language: str, text_content: str) -> EaseVoiceResponse:
-        self.refinement.submit_text(index, language.upper(), text_content)
+        self.refinement.submit_text(index, language.lower(), text_content)
         return EaseVoiceResponse(ResponseStatus.SUCCESS, "Submit Text Success", self.refinement.source_file_content)
 
     def refinement_delete_text(self, file_index: str):
         self.refinement.delete_text(file_index)
         return EaseVoiceResponse(ResponseStatus.SUCCESS, "Delete Text Success", self.refinement.source_file_content)
-
-    def refinement_reload(self):
-        self.refinement.load_text()
-        return EaseVoiceResponse(ResponseStatus.SUCCESS, "Reload Success", self.refinement.source_file_content)