src/envs/gym_env/server/Dockerfile

# Dockerfile for Atari Environment
# This image provides Atari 2600 games via the Arcade Learning Environment (ALE)

# Configurable base image - defaults to local build, can be overridden for CI/CD
# Base image provides: fastapi, uvicorn, requests, curl, PYTHONPATH=/app/src
#
# Local build: docker build -t envtorch-base:latest -f src/core/containers/images/Dockerfile .
#              docker build -f src/envs/atari_env/server/Dockerfile -t atari-env:latest .
#
# CI/CD build: docker build --build-arg BASE_IMAGE=ghcr.io/meta-pytorch/openenv-base:latest \
#              -f src/envs/atari_env/server/Dockerfile -t atari-env:latest .
ARG BASE_IMAGE=openenv-base:latest
FROM ${BASE_IMAGE}

# Install dependencies
COPY src/envs/gym_env/server/requirements.txt /tmp/requirements.txt
RUN apt-get update && apt-get install -y \
    swig \
    build-essential \
    python3-dev

RUN pip install -r /tmp/requirements.txt && rm /tmp/requirements.txt

# Copy OpenEnv core (base image already set WORKDIR=/app)
COPY src/core/ /app/src/core/

# Copy Gym environment code
COPY  src/envs/gym_env/ /app/src/envs/gym_env/

# Copy README for web interface documentation
COPY src/envs/gym_env/README.md /app/README.md

ARG GYM_ENVIRONMENT_ID="MountainCarContinuous-v0"
ARG GYM_RENDER_MODE="rgb_array"
ENV ADDITIONAL_PARAMETERS_YAML_FILE=""


# --- Runtime environment with defaults ---
# These ENV lines set defaults but still allow runtime overrides
ENV GYM_ENVIRONMENT_ID=${GYM_ENVIRONMENT_ID}
ENV ADDITIONAL_PARAMETERS=${ADDITIONAL_PARAMETERS_YAML_FILE}

# Expose port
EXPOSE 8000

# Health check
HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
    CMD curl -f http://localhost:8000/health || exit 1

# Run the FastAPI server
CMD ["uvicorn", "envs.gym_env.server.app:app", "--host", "0.0.0.0", "--port", "8000"]