render / Dockerfile
qbhf2's picture
Update Dockerfile
9fc5909 verified
raw
history blame
4.54 kB
# -------- Base Python with libs Blender needs (CPU headless) --------
FROM python:3.11-slim-bookworm
ENV DEBIAN_FRONTEND=noninteractive
# System deps for Blender GUI-less rendering + runtime fetch tools
RUN apt-get update && apt-get install -y --no-install-recommends \
ca-certificates wget curl bzip2 xz-utils git rsync bash \
libglib2.0-0 libx11-6 libxi6 libxxf86vm1 libxrender1 libxfixes3 \
libxkbcommon0 libxrandr2 libasound2 libxinerama1 libsm6 libice6 \
libgl1 libegl1 libglu1-mesa libdbus-1-3 libxcb1 \
&& rm -rf /var/lib/apt/lists/*
# Local copy of model-viewer (no external loads at runtime)
RUN mkdir -p /app/static && \
wget -q -O /app/static/model-viewer.min.js \
https://unpkg.com/@google/model-viewer/dist/model-viewer.min.js
# -------- Install official Blender (includes OpenImageDenoise) --------
ARG BLENDER_VERSION=4.1.1
ARG BLENDER_MAJOR=4.1
RUN wget -q https://download.blender.org/release/Blender${BLENDER_MAJOR}/blender-${BLENDER_VERSION}-linux-x64.tar.xz \
&& tar -xJf blender-${BLENDER_VERSION}-linux-x64.tar.xz -C /opt \
&& rm blender-${BLENDER_VERSION}-linux-x64.tar.xz \
&& ln -s /opt/blender-${BLENDER_VERSION}-linux-x64/blender /usr/local/bin/blender
# Pillow inside Blender's embedded Python (some scripts import PIL there)
RUN /opt/blender-${BLENDER_VERSION}-linux-x64/${BLENDER_MAJOR}/python/bin/python3.11 -m ensurepip && \
/opt/blender-${BLENDER_VERSION}-linux-x64/${BLENDER_MAJOR}/python/bin/python3.11 -m pip install --no-cache-dir Pillow
# -------- Working dir where repo will be extracted --------
WORKDIR /srv/app
# -------- Startup script: download private repo then run app.py --------
# Required Space settings:
# Secret: GIT_TOKEN (PAT with repo read access)
# Variable: GIT_REPO (e.g. chaous/render OR https://github.com/chaous/render.git)
# Variable: GIT_REF (optional, default "main"; branch/tag/commit)
# Variable: GIT_SUBDIR (optional, if app.py is not at repo root)
RUN printf '%s\n' \
'#!/usr/bin/env bash' \
'set -euo pipefail' \
'export GIT_TERMINAL_PROMPT=0' \
'' \
'export BLENDER_BIN="${BLENDER_BIN:-blender}"' \
'TARGET_ROOT="/srv/app"' \
'REPO="${GIT_REPO:-}"' \
'REF="${GIT_REF:-main}"' \
'SUBDIR="${GIT_SUBDIR:-}"' \
'' \
'if [[ -z "${REPO}" || -z "${GIT_TOKEN:-}" ]]; then' \
' echo "[error] GIT_REPO and GIT_TOKEN must be set in Space settings." >&2' \
' exit 1' \
'fi' \
'' \
'fetch_github_tarball() {' \
' local owner="$1"; local name="$2"; local ref="$3"' \
' echo "[info] Downloading github.com/${owner}/${name}@${ref} tarball..."' \
' local ARCHIVE_URL="https://api.github.com/repos/${owner}/${name}/tarball/${ref}"' \
' wget -q --header="Authorization: token ${GIT_TOKEN}" -O /tmp/repo.tar.gz "${ARCHIVE_URL}"' \
' rm -rf "${TARGET_ROOT:?}/"*' \
' tar -xzf /tmp/repo.tar.gz -C "${TARGET_ROOT}" --strip-components=1' \
' rm -f /tmp/repo.tar.gz' \
'}' \
'' \
'echo "[info] Fetching ${REPO}@${REF}..."' \
'if [[ "${REPO}" == *"://"* ]]; then' \
' # URL form' \
' if [[ "${REPO}" == *"github.com"* ]]; then' \
' path="${REPO#*github.com/}"; path="${path%.git}"' \
' owner="${path%%/*}"; name="${path#*/}"; name="${name%%/*}"' \
' fetch_github_tarball "${owner}" "${name}" "${REF}"' \
' else' \
' echo "[info] Non-GitHub host detected; using git clone."' \
' rm -rf "${TARGET_ROOT:?}/"*' \
' git -c http.extraHeader="Authorization: Bearer ${GIT_TOKEN}" clone --depth 1 --branch "${REF}" "${REPO}" "${TARGET_ROOT}"' \
' fi' \
'else' \
' # owner/repo form' \
' owner="${REPO%%/*}"; name="${REPO#*/}"' \
' fetch_github_tarball "${owner}" "${name}" "${REF}"' \
'fi' \
'' \
'if [[ -n "${SUBDIR}" ]]; then' \
' if [[ -d "${TARGET_ROOT}/${SUBDIR}" ]]; then' \
' cd "${TARGET_ROOT}/${SUBDIR}"' \
' else' \
' echo "[warn] GIT_SUBDIR=${SUBDIR} not found; using repo root."' \
' cd "${TARGET_ROOT}"' \
' fi' \
'else' \
' cd "${TARGET_ROOT}"' \
'fi' \
'' \
'if [[ -f requirements.txt ]]; then' \
' echo "[info] Installing repo requirements..."' \
' pip install --no-cache-dir -r requirements.txt' \
'fi' \
'' \
'if [[ ! -f app.py ]]; then' \
' echo "[error] app.py not found in repo path." >&2' \
' ls -la' \
' exit 1' \
'fi' \
'' \
'echo "[info] Starting app.py..."' \
'exec python app.py' \
> /usr/local/bin/start.sh \
&& chmod +x /usr/local/bin/start.sh
# -------- HF Space runtime env --------
ENV PORT=7860 \
GRADIO_SERVER_NAME=0.0.0.0 \
GRADIO_SERVER_PORT=7860 \
BLENDER_BIN=blender
EXPOSE 7860
CMD ["/usr/local/bin/start.sh"]