feat(videos): backend features for reviewing video

app/video-processing/Makefile

@@ -43,6 +43,7 @@ run: setup-run ## Run the video processing server as a standalone container.
 		-e AWS_DEFAULT_REGION="ca-central-1" \
 		-e PRIVACYPAL_INPUT_VIDEO_DIR="$(PRIVACYPAL_INPUT_VIDEO_DIR)" \
 		-e PRIVACYPAL_OUTPUT_VIDEO_DIR="$(PRIVACYPAL_OUTPUT_VIDEO_DIR)" \
+		-e VIDPROC_IS_STATELESS="true" \
 		-v "$(LOCAL_INPUT_DIR)":"$(PRIVACYPAL_INPUT_VIDEO_DIR)":z \
 		-v "$(LOCAL_OUTPUT_DIR)":"$(PRIVACYPAL_OUTPUT_VIDEO_DIR)":z \
 		-it --rm $(VID_PROCESS_IMAGE) 2>&1 | tee $(LOG_FILE)

app/video-processing/requirements.txt

@@ -5,3 +5,4 @@ opencv-python-headless==4.8.1.78
 flask==3.0.0
 quart==0.19.3
 uvicorn==0.24.0.post1
+requests==2.22.0

app/video-processing/server.py

@@ -5,6 +5,11 @@
 
 app = Quart(__name__)
 vp = VideoProcessor()
+processes = {}  # key:value is of the form str:mp.Process
+is_stateless = False
+try:
+    is_stateless = True if str(os.environ["VIDPROC_IS_STATELESS"]) == "true" else False # if env variable not defined, will raise KeyError
+except:pass
 
 @app.route("/process_video", methods=["POST"])
 async def handle_request():
@@ -14,12 +19,36 @@ async def handle_request():
             final = f"{out_path}/{file[:-4]}-processed{file[-4:]}"
             if not app.testing: # if we're running Flask unit tests, don't run the video processing method
                 process = mp.Process(target=vp.process_INTERPOLATE, args=(f"{input_path}/{file}", final))  # define a new process pointing to process_INTERPOLATE
+                if is_stateless:
+                    processes[file] = process
                 process.start() # start the process on another thread
                 print(f"Process started on {file}")
-            return "Success: file exists"
+            return "Success: file exists", 200
         else:
-            return "Error: file not found"
-    return "Error: request must be of type POST"
+            return "Error: file not found", 200
+    return "Error: request must be of type POST", 405
+
+@app.route("/check_finished", methods=["POST"])
+async def return_status():
+    if is_stateless:  # only enable this route if we're running in stateless mode
+        if request.method == "POST":
+            file = (await request.data).decode()
+            process: mp.Process
+            try:
+                process = processes[file]
+            except KeyError:
+                return "Invalid filename", 200  # shouldn't ever happen, but just in case
+
+            if process.is_alive():
+                return "false", 200 # return false to the request for "is the video finished processing"
+            else:
+                processes.pop(file) # remove the process from the dictionary
+                return "true", 200  # return true
+        else:
+            return "Error: request must be of type POST", 405
+    else:
+        print("Not running in stateless mode, returning 404")
+        return "", 404
 
 @app.route("/health", methods=["GET"])
 async def return_health():

app/video-processing/video_processor.py

@@ -1,11 +1,16 @@
-import cv2 as cv, os, time, multiprocessing as mp, numpy as np, subprocess as sp, random, boto3
+import cv2 as cv, os, time, multiprocessing as mp, numpy as np, subprocess as sp, random, boto3, requests
 
 class VideoProcessor:
     client: boto3.client
+    is_stateless: bool
     def __init__(self):
         self.client = boto3.client("rekognition")   # request a client of the type 'rekognition' from aws services
+        self.is_stateless = False
+        try:
+            self.is_stateless = True if str(os.environ["VIDPROC_IS_STATELESS"]) == "true" else False # if env variable not defined, will raise KeyError
+        except:pass
 
-    def blur_frame(self, img, rects: list, r: int = 50):
+    def blur_frame(self, img, rects: list, r: int = 25):
         """
         Loads an image and applies a blur on all regions specified by `rects`.
         `img` is an opencv image (ie essentially an np array).
@@ -165,6 +170,8 @@ def process_INTERPOLATE(self, src: str, out: str, keyframe_interval: float = 0.5
         p.wait()
         os.remove(tmp)
         print(f"Done processing {src}.")
+        if not self.is_stateless:
+            requests.post("http://localhost:8080/some/nextjs/api/route", out)   # post to a next.js api route with the output filename to signify processing completion
 
     def img_to_bytes(self, img) -> bytes:
         """