update

components/services/dataset.py

@@ -52,6 +52,7 @@ class DatasetService:
         self.vectorizer = vectorizer
         self.regulations_path = Path(config.db_config.files.regulations_path)
         self.documents_path = Path(config.db_config.files.documents_path)
+        self.tmp_path= Path(os.environ.get(config.db_config.files.documents_path, '.'))
         logger.info("DatasetService initialized")
 
 
@@ -274,9 +275,10 @@ class DatasetService:
         """
         Получить информацию о процессе обработки датасета.
         """
-        if Path('tmp.json').exists():
+        tmp_file = Path(self.tmp_path / 'tmp.json')
+        if tmp_file.exists():
             try:
-                with open('tmp.json', 'r', encoding='utf-8') as f:
+                with open(tmp_file, 'r', encoding='utf-8') as f:
                     info = json.load(f)
             except Exception as e:
                 logger.warning(f"Error loading processing info: {e}")
@@ -316,7 +318,7 @@ class DatasetService:
         logger.info(f"Uploading ZIP file {file.filename}")
         self.raise_if_processing()
 
-        file_location = Path.cwd() / 'tmp' / 'tmp.zip'
+        file_location = Path(self.tmp_path / 'tmp.json' / 'tmp.zip')
         logger.debug(f"Saving uploaded file to {file_location}")
         file_location.parent.mkdir(parents=True, exist_ok=True)
         with open(file_location, 'wb') as f:
@@ -352,7 +354,7 @@ class DatasetService:
                 status_code=400, detail='Dataset is not draft but trying to apply it'
             )
 
-        TMP_PATH = Path('tmp.json')
+        TMP_PATH = Path(self.tmp_path / 'tmp.json')
 
         def progress_callback(current: int, total: int) -> None:
             log_step = total // 100