File size: 2,883 Bytes
ee472e7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c54d8a2
 
 
 
 
 
 
 
 
 
 
 
 
ee472e7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c54d8a2
 
ee472e7
 
 
 
 
c54d8a2
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# Build frontend
FROM node:18 as frontend-build
WORKDIR /app
COPY frontend/package*.json ./
RUN npm install
COPY frontend/ ./

RUN npm run build

# Build backend
FROM python:3.12-slim
WORKDIR /app

# Create non-root user
RUN useradd -m -u 1000 user

# Install poetry
RUN pip install poetry

# Create and configure cache directory
RUN mkdir -p /app/.cache && \
    chown -R user:user /app

# Copy and install backend dependencies
COPY backend/pyproject.toml backend/poetry.lock* ./
RUN poetry config virtualenvs.create false \
    && poetry install --no-interaction --no-ansi --no-root --only main

# Isolated venv for the eval worker (heavy ML deps: torch + braindecode from git),
# kept OUT of the web app's pinned stack to avoid version conflicts. Built before
# the code COPY so it stays cached across code-only rebuilds. Non-fatal: if the
# install fails, the web app still builds and the worker is simply disabled.
RUN apt-get update && apt-get install -y --no-install-recommends git \
    && python -m venv /opt/worker-venv \
    && /opt/worker-venv/bin/pip install --no-cache-dir -U pip \
    && (/opt/worker-venv/bin/pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu \
        && /opt/worker-venv/bin/pip install --no-cache-dir huggingface_hub pandas pyarrow open-eeg-bench \
        || echo "WARN: eval-worker deps failed to install; worker disabled at runtime") \
    && chown -R user:user /opt/worker-venv \
    && rm -rf /var/lib/apt/lists/*

# Copy backend code
COPY backend/ .

# Install Node.js and npm
RUN apt-get update && apt-get install -y \
    curl \
    netcat-openbsd \
    && curl -fsSL https://deb.nodesource.com/setup_18.x | bash - \
    && apt-get install -y nodejs \
    && rm -rf /var/lib/apt/lists/*

# Copy frontend server and build
COPY --from=frontend-build /app/build ./frontend/build
COPY --from=frontend-build /app/package*.json ./frontend/
COPY --from=frontend-build /app/server.js ./frontend/

# Install frontend production dependencies
WORKDIR /app/frontend
RUN npm install --production
WORKDIR /app

# Environment variables
ENV HF_HOME=/app/.cache \
    HF_DATASETS_CACHE=/app/.cache \
    INTERNAL_API_PORT=7861 \
    PORT=7860 \
    NODE_ENV=production \
    RUN_EVAL_WORKER=1

# Note: HF_TOKEN should be provided at runtime, not build time
USER user
EXPOSE 7860

# Start the API, the (optional) CPU eval worker, and the frontend. The worker
# runs in its own venv as a background --watch loop. Disable it without a rebuild
# by setting RUN_EVAL_WORKER=0 in the Space settings, then restarting.
CMD ["sh", "-c", "uvicorn app.asgi:app --host 0.0.0.0 --port 7861 & if [ \"$RUN_EVAL_WORKER\" = \"1\" ]; then PYTHONPATH=/app /opt/worker-venv/bin/python -m scripts.run_eval_worker --watch --limit 1 --interval 300 --device cpu & fi; while ! nc -z localhost 7861; do sleep 1; done && cd frontend && npm run serve"]