Merge pull request #27 from UMEssen/26-move-default-dicom-folder-to-temp

Remove DICOMs after computation and bump up version

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "body-organ-analysis"
-version = "0.1.1"
+version = "0.1.2"
 license = "Apache-2.0"
 description = "BOA is a tool for segmentation of CT scans developed by the SHIP-AI group at the Institute for Artificial Intelligence in Medicine (https://ship-ai.ikim.nrw/). Combining the TotalSegmentator and the Body Composition Analysis, this tool is capable of analyzing medical images and identifying the different structures within the human body, including bones, muscles, organs, and blood vessels."
 authors = ["Giulia Baldini <[email protected]>"]

scripts/celery_task.py

@@ -8,6 +8,7 @@
 from typing import Any, Dict, List, Optional
 
 import requests
+import shutil
 from celery import Celery, bootsteps
 from celery.signals import worker_ready, worker_shutdown
 from requests.exceptions import HTTPError
@@ -228,6 +229,7 @@ def analyze_stable_series(resource_id: str) -> Dict[str, Optional[str]]:
         )
         stats["download_time"] = download_time
         stats["save_persistent_time"] = time() - start_store
+    shutil.rmtree(input_data_folder)
     logger.info(f"Entire pipeline: DONE in {time() - start_init:0.5f}s")
 
     stats["task_id"] = analyze_stable_series.request.id