[Improvement] Cleaned data persistence among page reloads

Author: fl0wxr

python/main.py

@@ -202,6 +202,29 @@ def authenticate(credentials: HTTPBasicCredentials = Depends(security)) -> bool:
     return True
 
 
+@app.get("/check_existence_of_clean")
+async def check_existence_of_clean() -> UploadFilesResponse:
+
+    session_fp = "./tmp/session-data/clean/de-identified-files/session.json"
+    proper_dicom_paths = sorted(glob.glob("./tmp/session-data/clean/de-identified-files/**/*.dcm", recursive = True))
+    total_uploaded_file_bytes = 0
+    skip_deidentification = True
+    if os.path.exists(session_fp) and len(proper_dicom_paths) != 0:
+        for dcm_fp in proper_dicom_paths:
+            total_uploaded_file_bytes += sys.getsizeof(dcm_fp)
+            total_uploaded_file_megabytes = "%.1f" % (
+                total_uploaded_file_bytes / (10**3) ** 2
+            )
+    else:
+        total_uploaded_file_megabytes = 0
+
+    return UploadFilesResponse(
+        n_uploaded_files=len(proper_dicom_paths),
+        total_size=total_uploaded_file_megabytes,
+        skip_deidentification=skip_deidentification,
+    )
+
+
 @app.get("/", response_class=HTMLResponse, dependencies=[Depends(authenticate)])
 async def get_root(request: Request) -> HTMLResponse:
     return templates.TemplateResponse("index.html", {"request": request})

python/static/script.js

@@ -400,6 +400,27 @@ nextSlice.addEventListener("click", function () {
     DICOMSlider.dispatchEvent(new Event("input"));
 });
 
+document.addEventListener("DOMContentLoaded", function() {
+    check_existence_of_clean();
+  });
+
+async function check_existence_of_clean()
+{
+    const dcm_files_response = await fetch
+    (
+        '/check_existence_of_clean',
+        {
+            method: 'GET'
+        }
+    );
+    const dcm_files = await dcm_files_response.json();
+    if (dcm_files_response.ok && dcm_files.n_uploaded_files > 0)
+    {
+        skip_deidentification = dcm_files.skip_deidentification
+        await submit_dicom_processing_request();
+    }
+}
+
 function base64torgba(encodedData) {
     const binaryString = window.atob(encodedData);
     const len = binaryString.length;
@@ -611,7 +632,14 @@ async function submit_dicom_processing_request()
         }
     }
     document.body.style.cursor = 'default';
-    showNotification("success", "DICOM files are now de-identified", 3000);
+    if (skip_deidentification)
+    {
+        showNotification("success", "Loaded de-identified DICOM files", 3000);
+    }
+    else
+    {
+        showNotification("success", "DICOM files are now de-identified", 3000);
+    }
 }
 
 async function get_mask_from_file() {