FROM python:3.11-slim

ENV PYTHONUNBUFFERED=1 \
    PORT=7860 \
    PIP_NO_CACHE_DIR=1 \
    HF_HUB_DISABLE_XET=1

RUN apt-get update && apt-get install -y --no-install-recommends \
    git curl ca-certificates ffmpeg build-essential linux-libc-dev \
    libgl1 libglib2.0-0 libegl1 libosmesa6 libgomp1

WORKDIR /app

COPY cosmos-framework /app/cosmos-framework
COPY assets /app/assets
COPY start.sh /app/start.sh

RUN python -m pip install --upgrade pip uv \
 && uv pip install --system torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cpu \
 && uv pip install --system -e /app/cosmos-framework --no-deps \
 && uv pip install --system \
    accelerate av cattrs diffusers einops hydra-core imageio imageio-ffmpeg loguru msgpack \
    obstore omegaconf pydantic requests scipy termcolor 'transformers>=4.57.1,<5.0.0' tyro websockets \
    boto3 pyyaml lerobot==0.4.4 pandas pyarrow 'datasets>=2.19.0' \
    viser mujoco pin trimesh opencv-python-headless huggingface_hub hf_transfer iopath \
 && uv pip install --system torchcodec==0.2.* --index-url https://download.pytorch.org/whl/cpu

CMD ["bash", "/app/start.sh"]