visual_memory / Dockerfile
kdemon1011's picture
Upload Dockerfile with huggingface_hub
6074ed5 verified
# Visual Memory Gym — Docker image for OpenEnv + Hugging Face Spaces
#
# Single-service Python container on port 8000.
# No database, no external APIs, no additional services.
#
# Build:
# cd visual-memory && docker build -f server/Dockerfile -t openenv-visual-memory .
#
# Run:
# docker run -d --name visual-memory -p 8000:8000 openenv-visual-memory
ARG BASE_IMAGE=ghcr.io/meta-pytorch/openenv-base:latest
FROM ${BASE_IMAGE} AS builder
RUN apt-get update && \
apt-get install -y --no-install-recommends git curl && \
rm -rf /var/lib/apt/lists/*
WORKDIR /app
COPY . /app/env
WORKDIR /app/env
RUN if ! command -v uv >/dev/null 2>&1; then \
curl -LsSf https://astral.sh/uv/install.sh | sh && \
mv /root/.local/bin/uv /usr/local/bin/uv; \
fi
RUN --mount=type=cache,target=/root/.cache/uv \
if [ -f uv.lock ]; then uv sync --frozen --no-install-project --no-editable; \
else uv sync --no-install-project --no-editable; fi
RUN --mount=type=cache,target=/root/.cache/uv \
if [ -f uv.lock ]; then uv sync --frozen --no-editable; \
else uv sync --no-editable; fi
FROM ${BASE_IMAGE}
WORKDIR /app
COPY --from=builder /app/env/.venv /app/.venv
COPY --from=builder /app/env /app/env
ENV PATH="/app/.venv/bin:$PATH"
ENV PYTHONPATH="/app/env:$PYTHONPATH"
ENV ENABLE_WEB_INTERFACE=true
ENV SPACE_ID=huzzle-labs/visual_memory
RUN python -c "\
import re, pathlib;\
src = pathlib.Path('/app/env/README.md').read_text();\
clean = re.sub(r'^---\n.*?\n---\n', '', src, count=1, flags=re.DOTALL);\
pathlib.Path('/app/env/.README_web.md').write_text(clean)"
ENV ENV_README_PATH=/app/env/.README_web.md
EXPOSE 8000
HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"
CMD ["sh", "-c", "cd /app/env && uvicorn server.app:app --host 0.0.0.0 --port 8000"]