Fixed upload permission

perception_roi_server.py

@@ -38,6 +38,9 @@ DEFAULT_CONF = float(os.environ.get("YOLO_CONF", "0.25"))
 DEFAULT_DEVICE = os.environ.get("YOLO_DEVICE", "auto")
 FAST_DETECT_SCALE = float(os.environ.get("FAST_DETECT_SCALE", "0.65"))
 FAST_DETECT_IMGSZ = int(os.environ.get("FAST_DETECT_IMGSZ", "512"))
+DATA_DIR = os.environ.get("DATA_DIR", "/tmp/roi_demo")
+UPLOAD_DIR = os.path.join(DATA_DIR, "uploads")
+OUTPUT_DIR = os.path.join(DATA_DIR, "outputs")
 
 app = FastAPI(title="ROI Compression Server", version="1.0.0")
 app.add_middleware(
@@ -371,8 +374,8 @@ def _process_job(job: Job):
             scale=max(0.25, min(1.0, (job.target_width / w) if (job.target_width and w) else 1.0)),
         )
 
-        os.makedirs("outputs", exist_ok=True)
-        overlay_path = os.path.join("outputs", f"{job.id}_overlay.mp4")
+        os.makedirs(OUTPUT_DIR, exist_ok=True)
+        overlay_path = os.path.join(OUTPUT_DIR, f"{job.id}_overlay.mp4")
         overlay_writer = _open_writer(overlay_path, w, h, tfps)
 
         with job.lock:
@@ -459,9 +462,9 @@ def _compress_job(job: Job, bandwidth_kbps: int, target_fps: int, target_w: int,
             scale=resolution_scale,
         )
 
-        os.makedirs("outputs", exist_ok=True)
-        compressed_path = os.path.join("outputs", f"{job.id}_compressed_rt.mp4")
-        roi_path = os.path.join("outputs", f"{job.id}_roi_rt.mp4")
+        os.makedirs(OUTPUT_DIR, exist_ok=True)
+        compressed_path = os.path.join(OUTPUT_DIR, f"{job.id}_compressed_rt.mp4")
+        roi_path = os.path.join(OUTPUT_DIR, f"{job.id}_roi_rt.mp4")
 
         compressed_writer = _open_writer(compressed_path, tw, th, tfps)
         roi_writer = _open_writer(roi_path, w, h, tfps)
@@ -564,8 +567,8 @@ async def track_async(
     resolution_scale: float = Form(1.0),
 ):
     job_id = uuid.uuid4().hex[:12]
-    os.makedirs("uploads", exist_ok=True)
-    dst = os.path.join("uploads", f"{job_id}_{os.path.basename(video.filename or 'input.mp4')}")
+    os.makedirs(UPLOAD_DIR, exist_ok=True)
+    dst = os.path.join(UPLOAD_DIR, f"{job_id}_{os.path.basename(video.filename or 'input.mp4')}")
     data = await video.read()
     with open(dst, "wb") as f:
         f.write(data)