BYOC Quickstart

mkdir byoc-quickstart && cd byoc-quickstart

"""
processor.py - BYOC green-tint pipeline for Livepeer.

Applies a green-channel boost to every incoming video frame.
CPU only, no model loading.
"""

import asyncio
import logging

from pytrickle import FrameProcessor, run_processor

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


class GreenTintProcessor(FrameProcessor):
    """Applies a green channel boost to every frame.

    Input and output: raw RGB bytes (width * height * 3 bytes per frame).
    """

    async def initialize(self):
        """Called once when the container starts. Load models here."""
        logger.info("GreenTintProcessor: initialised")

    async def process(self, frame: bytes) -> bytes:
        """Receive one frame of raw bytes, return transformed bytes."""
        data = bytearray(frame)
        # Boost the green channel (offset 1 in each RGB triplet)
        for i in range(1, len(data), 3):
            data[i] = min(255, data[i] + 80)
        return bytes(data)

    async def shutdown(self):
        """Called when the job ends."""
        logger.info("GreenTintProcessor: shutdown")


if __name__ == "__main__":
    asyncio.run(run_processor(GreenTintProcessor(), port=8000))

FROM python:3.11-slim

WORKDIR /app

RUN apt-get update && apt-get install -y --no-install-recommends git \
    && rm -rf /var/lib/apt/lists/*

RUN pip install --no-cache-dir \
    git+https://github.com/livepeer/pytrickle.git

COPY processor.py ./processor.py

EXPOSE 8000

ENTRYPOINT ["python", "processor.py"]

docker build -t byoc-green-tint:latest .
docker images | grep byoc-green-tint

docker run --name byoc-test -p 8000:8000 byoc-green-tint:latest

curl http://localhost:8000/api/stream/status

docker stop byoc-test && docker rm byoc-test

docker run --name byoc-green-tint --network host byoc-green-tint:latest

./livepeer \
  -orchestrator \
  -network offchain \
  -serviceAddr 127.0.0.1:8936 \
  -cliAddr 127.0.0.1:7936 \
  -datadir ~/.lpData-orch \
  -byoc \
  -byocContainerURL http://127.0.0.1:8000 \
  -byocModelID green-tint-cpu

./livepeer \
  -gateway \
  -network offchain \
  -orchAddr 127.0.0.1:8936 \
  -httpAddr 0.0.0.0:8935 \
  -httpIngest \
  -cliAddr 127.0.0.1:5935 \
  -datadir ~/.lpData-gw

import requests

GATEWAY_URL = "http://localhost:8935"
MODEL_ID = "green-tint-cpu"

# Black 640x480 RGB frame
frame = bytes(640 * 480 * 3)

response = requests.post(
    f"{GATEWAY_URL}/live/video-to-video",
    headers={
        "Content-Type": "application/octet-stream",
        "X-Model-Id": MODEL_ID,
    },
    data=frame,
)

print(f"Status: {response.status_code}")
print(f"Response size: {len(response.content)} bytes")

# Confirm the green-channel boost happened
g_channel_sum = sum(response.content[1::3])
print(f"Green channel sum: {g_channel_sum} (input was 0)")

python send_job.py