fix docker cache handling

Dockerfile

@@ -2,7 +2,7 @@ FROM ubuntu:22.04 AS build
 WORKDIR /src
 
 RUN apt-get update && \
-  apt-get install -y build-essential cmake git curl && \
+  apt-get install -y build-essential cmake git curl ninja-build && \
   rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
 
 ARG CRISPASR_REPO=https://github.com/CrispStrobe/CrispASR
@@ -12,8 +12,10 @@ RUN git clone --depth 1 "${CRISPASR_REPO}" /src/CrispASR && \
   git -C /src/CrispASR fetch --depth 1 origin "${CRISPASR_REF}" && \
   git -C /src/CrispASR checkout --detach FETCH_HEAD
 WORKDIR /src/CrispASR
-RUN cmake -B build -DWHISPER_BUILD_TESTS=OFF && \
-  cmake --build build -j"$(nproc)" --target whisper-cli
+ARG CRISPASR_BUILD_JOBS
+RUN jobs="${CRISPASR_BUILD_JOBS:-$(nproc)}" && \
+  cmake -S . -B build -G Ninja -DWHISPER_BUILD_TESTS=OFF && \
+  cmake --build build -j"${jobs}" --target whisper-cli
 
 FROM ubuntu:22.04 AS runtime
 WORKDIR /space
@@ -35,6 +37,7 @@ RUN chmod +x /space/start.sh && \
 
 ENV PATH=/opt/crispasr/build/bin:$PATH
 ENV CRISPASR_SERVER_URL=http://127.0.0.1:8080
+ENV CRISPASR_CACHE_DIR=/cache
 ENV GRADIO_SERVER_NAME=0.0.0.0
 ENV GRADIO_SERVER_PORT=7860
 

README.md

@@ -19,6 +19,7 @@ This folder contains a Hugging Face Docker Space wrapper for CrispASR:
 - `CRISPASR_BACKEND=whisper` or another backend name
 - `CRISPASR_LANGUAGE=en`
 - `CRISPASR_AUTO_DOWNLOAD=1`
+- `CRISPASR_CACHE_DIR=/cache`
 - `CRISPASR_EXTRA_ARGS=`
 
 ## Local build
@@ -30,3 +31,15 @@ docker run --rm -p 7860:7860 -p 8080:8080 \
   -v "$PWD/models:/models" \
   crispasr-hf-space
 ```
+
+For auto-downloads, mount a writable cache volume if you want models to survive container restarts:
+
+```bash
+docker volume create crispasr-cache
+docker run --rm -p 7860:7860 -p 8080:8080 \
+  -e CRISPASR_AUTO_DOWNLOAD=1 \
+  -v crispasr-cache:/cache \
+  crispasr-hf-space
+```
+
+Build parallelism can be tuned with `--build-arg CRISPASR_BUILD_JOBS=8`.

start.sh

@@ -15,10 +15,29 @@ MODEL_PATH="${CRISPASR_MODEL:-/models/model.gguf}"
 LANGUAGE="${CRISPASR_LANGUAGE:-en}"
 BACKEND="${CRISPASR_BACKEND:-whisper}"
 AUTO_DOWNLOAD="${CRISPASR_AUTO_DOWNLOAD:-1}"
+CACHE_DIR="${CRISPASR_CACHE_DIR:-/cache}"
 EXTRA_ARGS="${CRISPASR_EXTRA_ARGS:-}"
 
+ensure_writable_dir() {
+    local dir="$1"
+    if [[ ! -d "$dir" ]]; then
+        if ! mkdir -p "$dir" 2>/dev/null; then
+            log "ERROR: could not create cache directory '$dir'"
+            exit 70
+        fi
+    fi
+    if [[ ! -w "$dir" ]]; then
+        log "ERROR: cache directory '$dir' is not writable by uid=$(id -u) gid=$(id -g)"
+        ls -ld "$dir" >&2 || true
+        log "If this is a bind mount, chown the host directory or set CRISPASR_CACHE_DIR to a writable path."
+        exit 70
+    fi
+}
+
+ensure_writable_dir "$CACHE_DIR"
+
 declare -a cmd
-cmd=(crispasr --server --host "$SERVER_HOST" --port "$SERVER_PORT" -l "$LANGUAGE")
+cmd=(crispasr --server --host "$SERVER_HOST" --port "$SERVER_PORT" -l "$LANGUAGE" --cache-dir "$CACHE_DIR")
 
 if [[ "$AUTO_DOWNLOAD" == "1" ]]; then
     cmd+=(-m auto --auto-download)
@@ -35,7 +54,7 @@ if [[ -n "$EXTRA_ARGS" ]]; then
 fi
 
 log "startup"
-log "server_host=$SERVER_HOST server_port=$SERVER_PORT backend=$BACKEND language=$LANGUAGE auto_download=$AUTO_DOWNLOAD"
+log "server_host=$SERVER_HOST server_port=$SERVER_PORT backend=$BACKEND language=$LANGUAGE auto_download=$AUTO_DOWNLOAD cache_dir=$CACHE_DIR"
 if [[ "$AUTO_DOWNLOAD" == "1" ]]; then
     log "model=auto"
 else