sst-hack / Dockerfile
amanmurari's picture
Upload folder using huggingface_hub
0a56fb7 verified
Raw
History Blame Contribute Delete
3.37 kB
FROM python:3.11-slim
LABEL maintainer="OpenEnv Hackathon"
LABEL description="Autonomous Traffic Control – OpenEnv Environment (self-contained)"
# ── Install system deps (as root) ──────────────────────────────────────────
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
&& rm -rf /var/lib/apt/lists/*
# ── Create user for Hugging Face Spaces (UID 1000) ─────────────────────────
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
WORKDIR $HOME/app
# ── Copy packaging manifest first (layer-cache friendly) ──────────────────
COPY --chown=user pyproject.toml README.md ./
# Fix the setuptools package lookup to not scan the root filesystem in Docker
RUN sed -i 's|where = \[".."\]|where = \["."\]|g' pyproject.toml || true
# ── Install Python deps ────────────────────────────────────────────────────
RUN pip install --no-cache-dir \
"openenv-core[core]>=0.2.2" \
"openai>=1.0.0" \
"gradio>=4.0.0" \
"numpy>=1.24.0" \
"python-dotenv>=1.0.0" \
"fastapi>=0.104.0" \
"uvicorn[standard]>=0.24.0" \
"pydantic>=2.5.0" \
"requests>=2.31.0" \
"python-multipart>=0.0.6"
# ── Copy source (self-contained; no root-level deps needed) ───────────────
COPY --chown=user models.py ./traffic_control/models.py
COPY --chown=user environment.py ./traffic_control/environment.py
COPY --chown=user tasks.py ./traffic_control/tasks.py
COPY --chown=user client.py ./traffic_control/client.py
COPY --chown=user __init__.py ./traffic_control/__init__.py
COPY --chown=user inference.py ./traffic_control/inference.py
COPY --chown=user openenv.yaml ./traffic_control/openenv.yaml
COPY --chown=user dashboard.py ./traffic_control/dashboard.py
COPY --chown=user analytics.py ./traffic_control/analytics.py
COPY --chown=user arena.py ./traffic_control/arena.py
COPY --chown=user server/ ./traffic_control/server/
# inference.py must also be at the container root (validator requirement)
COPY --chown=user inference.py ./inference.py
# ── Create package anchor so Python treats traffic_control/ as the package ─
RUN echo "" > ./traffic_control/__init__.py || true
# ── Install the package in editable mode ──────────────────────────────────
RUN pip install --no-cache-dir -e .
# ── Runtime config ────────────────────────────────────────────────────────
ENV PORT=7860
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
EXPOSE 7860
# Single worker is required: Gradio's mount_gradio_app is not fork-safe.
# HF Spaces expects a single healthy process on port 7860.
ENV ENABLE_WEB_INTERFACE=true
CMD ["sh", "-c", \
"uvicorn traffic_control.server.app:app \
--host 0.0.0.0 \
--port ${PORT} \
--workers 1 \
--timeout-keep-alive 75"]