python codebase + dockerfile + makefile

.dockerignore

@@ -0,0 +1,6 @@
+# Docker related
+Dockerfile
+docker-compose.yml
+.dockerignore
+
+./postgres_data/*

.gitignore

@@ -0,0 +1,4 @@
+.output
+__pycache__/
+*.py[cod]
+*$py.class

Dockerfile

@@ -0,0 +1,20 @@
+FROM python:3.12-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgl1-mesa-glx
+
+COPY requirements.txt .
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+ 
+EXPOSE 5000-5100
+
+CMD ["python", "src/app.py"]

Makefile

@@ -0,0 +1,22 @@
+IMAGE_NAME := mock-video-feed
+CONTAINER_NAME := mock-video-feed-container
+DOCKERFILE := Dockerfile
+
+build:
+	docker build -t $(IMAGE_NAME) -f $(DOCKERFILE) .
+
+run:
+	docker stop $(CONTAINER_NAME) || true
+	docker rm $(CONTAINER_NAME) || true
+	docker run --name $(CONTAINER_NAME) -d -p 5000-5100:5000-5100 $(IMAGE_NAME)
+
+stop:
+	docker stop $(CONTAINER_NAME) || true
+	docker rm $(CONTAINER_NAME) || true
+
+clean:
+	docker rmi $(IMAGE_NAME) || true
+
+rebuild: stop clean build
+
+.PHONY: build run stop clean rebuild

requirements.txt

@@ -0,0 +1,8 @@
+opencv-python==4.11.0.86
+ultralytics==8.3.133
+Flask==3.1.0
+flask-cors==5.0.1
+PyThreadKiller==3.0.6
+vidgear==0.3.3
+selenium==4.32.0
+webdriver_manager==4.0.2

src/app.py

@@ -0,0 +1,71 @@
+from vidgear.gears import CamGear
+from flask import Flask
+import threading
+from utils import start_html_stream
+import os
+import time
+import cv2
+import socket
+
+def get_video_file():
+    return os.path.join("videos", "classroom.mp4")
+
+video_file = get_video_file()
+
+cap = cv2.VideoCapture(video_file)
+framerate = cap.get(cv2.CAP_PROP_FPS)
+cap.release()
+
+stream = CamGear(source=video_file).start()
+app = Flask(__name__)
+output_frame = [None]
+lock = threading.Lock()
+
+def get_available_port(start_port):
+    port = start_port
+    while True:
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+            if s.connect_ex(('localhost', port)) != 0:
+                return port
+            port += 1
+
+PORT = get_available_port(5000)
+
+@app.route('/')
+def html_stream():
+    return start_html_stream(output_frame, lock)
+
+def start_flask():
+    app.run(host="0.0.0.0", port=PORT, debug=True, use_reloader=False)
+
+def process_stream():
+    global stream
+    while True:
+        start_time = time.time()
+        frame = stream.read()
+        if frame is None:
+            stream.stop()
+            stream = CamGear(source=video_file).start()
+            continue
+        
+        with lock:
+            output_frame[0] = frame.copy()
+        
+        elapsed_time = time.time() - start_time
+        sleep_time = max(1.0 / framerate - elapsed_time, 0)
+        time.sleep(sleep_time)
+
+def main():
+    flask_thread = threading.Thread(target=start_flask)
+    flask_thread.daemon = True
+    flask_thread.start()
+    
+    stream_thread = threading.Thread(target=process_stream)
+    stream_thread.daemon = True
+    stream_thread.start()
+    
+    flask_thread.join()
+    stream_thread.join()
+
+if __name__ == "__main__":
+    main()

src/utils.py

@@ -0,0 +1,72 @@
+import cv2
+import numpy as np
+from selenium import webdriver
+from selenium.webdriver.chrome.service import Service
+from webdriver_manager.chrome import ChromeDriverManager
+from selenium.webdriver.chrome.options import Options
+import time
+from flask import Response
+import cv2
+import time
+
+def start_html_stream(output_frame, lock):
+    def generate():
+        while True:
+            with lock:
+                if output_frame[0] is None:
+                    # TODO: Investigate why if remove this print or time.sleep (but not both at the same time), the stream does not work
+                    # time.sleep(0.5)
+                    print(f"{time.time()} - Frame is None.")
+                    continue
+                (flag, encoded_image) = cv2.imencode(".jpg", output_frame[0])
+                if not flag:
+                    continue
+            yield(b'--frame\r\n' b'Content-Type: image/jpeg\r\n\r\n' + bytearray(encoded_image) + b'\r\n')
+    return Response(generate(), mimetype="multipart/x-mixed-replace; boundary=frame")
+
+def capture_webpage(driver):
+    screenshot = driver.get_screenshot_as_png()
+    image = np.frombuffer(screenshot, dtype=np.uint8)
+    image = cv2.imdecode(image, cv2.IMREAD_COLOR)
+    
+    height, width, _ = image.shape
+    new_width = 1920
+    new_height = int(new_width * height / width)
+    
+    image = cv2.resize(image, (new_width, new_height))
+    
+    if new_height > 1080:
+        start_y = (new_height - 1080) // 2
+        image = image[start_y:start_y + 1080, :]
+    
+    return image
+
+def get_webpage_frames(stream_url):
+    chrome_options = Options()
+    chrome_options.add_argument("--headless")
+    chrome_options.add_argument("--window-size=1920x1080")
+    
+    driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
+    driver.get(stream_url)
+    
+    frames_per_second = 10
+    frame_interval = 1 / frames_per_second
+
+    try:
+        while True:
+            image = capture_webpage(driver)
+            yield image
+            time.sleep(frame_interval)
+    finally:
+        driver.quit()
+
+def get_mjpeg_frames(stream_url):
+    cap = cv2.VideoCapture(stream_url)
+    
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+        yield frame
+    
+    cap.release()

videos/classroom.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a69bd5e39ff0e74286d13bcbfdadddd307b85decd2d9853db7518e0028785e31
+size 13548133