# Usamos una imagen con CUDA para la GPU
FROM nvidia/cuda:12.1.0-base-ubuntu22.04

# Instalamos Python y librerías para que OpenCV no falle
RUN apt-get update && apt-get install -y \
    python3-pip \
    python3-dev \
    git \
    libgl1-mesa-glx \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

# Configuramos el usuario
RUN useradd -m -u 1000 user
USER user
ENV PATH="/home/user/.local/bin:$PATH"
WORKDIR /home/user/app

# Instalamos LeRobot directamente
RUN pip install --no-cache-dir torch torchvision torchaudio
RUN pip install --no-cache-dir "lerobot" "huggingface-hub[cli,hf-transfer]"

# Copiamos los archivos y mantenemos el Space encendido
COPY --chown=user . /home/user/app
CMD ["sleep", "infinity"]