feat: unique directories for each launch

Author: Nigma-Ks

backend/config/paths.py

@@ -1,14 +1,28 @@
 import os
 
+
+def get_process_filepath(uid, filepath):
+    base_prefix = TMP_FILE_DIR + "/"
+    new_prefix = f"{TMP_FILE_DIR}/{uid}_"
+    return filepath.replace(base_prefix, new_prefix)
+
+
+DOCKER_DIR = os.getcwd() + "/docker"
 BASE_DIR = os.getcwd() + "/backend"
+TMP_FILE_DIR = BASE_DIR + "/tmp"
 
 RESOURCES_DIR = os.path.join(BASE_DIR, "resources")
-UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
-RESULTS_DIR = os.path.join(BASE_DIR, "results")
+UPLOAD_DIR = os.path.join(TMP_FILE_DIR, "uploads")
+RESULTS_DIR = os.path.join(TMP_FILE_DIR, "results")
 
 DATASET_FILE = os.path.join(RESOURCES_DIR, "dataset.json")
 LAUNCH_INFO_FILE = os.path.join(UPLOAD_DIR, "launch_info.csv")
 MODEL_ONNX_FILE = os.path.join(UPLOAD_DIR, "model.onnx")
 METHODS_TS_FILE = os.path.join(
     BASE_DIR, "../frontend/src/components/components/Methods.ts"
 )
+COMPSTRAT_RESULTS_DIR = os.path.join(RESULTS_DIR, "compstrat_results")
+ARTIFACTS_AI_CSV_FILE = os.path.join(RESULTS_DIR, "artifacts_run_ai/AI.csv")
+ARTIFACTS_BASELINE_CSV_FILE = os.path.join(
+    RESULTS_DIR, "artifacts_run_baseline/ExecutionTreeContributedCoverage.csv"
+)

backend/main.py

@@ -1,11 +1,18 @@
 from fastapi import FastAPI, UploadFile, File, Form
 from fastapi.middleware.cors import CORSMiddleware
+import shortuuid
 
+from backend.file_utils.files import reset_dir
 from backend.utils.docker_runner import run_pipeline
 from backend.utils.data_uploader import handle_upload
 from backend.utils.methods_handler import Methods
 from backend.utils.results_sender import send_folder_by_email
-from backend.config.paths import RESULTS_DIR, DATASET_FILE, METHODS_TS_FILE
+from backend.config.paths import (
+    RESULTS_DIR,
+    DATASET_FILE,
+    METHODS_TS_FILE,
+    get_process_filepath, BASE_DIR,
+)
 
 dataset_methods = Methods(data_filepath=DATASET_FILE, output_filepath=METHODS_TS_FILE)
 
@@ -20,23 +27,26 @@
 
 
 @app.post("/api/upload")
-async def upload_file(
-    file: UploadFile = File(...),
-    email: str = Form(...),
-    methods: str = Form(...),
-    experiment: str = Form(...),
+async def handle_submit(
+        file: UploadFile = File(...),
+        email: str = Form(...),
+        methods: str = Form(...),
+        experiment: str = Form(...),
 ):
-    handle_upload(file, methods, dataset_methods)
+    task_uid = str(shortuuid.uuid())
+
+    handle_upload(task_uid, file, methods, dataset_methods)
 
-    run_pipeline()
+    run_pipeline(task_uid)
 
     send_folder_by_email(
         email,
-        RESULTS_DIR,
+        get_process_filepath(task_uid, RESULTS_DIR),
         experiment,
         file.filename,
     )
 
+    reset_dir(BASE_DIR)
     return {
         "experiment": experiment,
         "filename": file.filename,