# Use NVIDIA CUDA as base for GPU support
FROM ubuntu:22.04

# Avoid prompts during installation
ENV DEBIAN_FRONTEND=noninteractive

# Install system dependencies
RUN apt-get update && apt-get install -y \
    curl \
    git \
    python3 \
    python3-pip \
    wget \
    docker.io \
    sudo \
    && rm -rf /var/lib/apt/lists/*

# Install code-server (VS Code Web)
RUN curl -fsSL https://code-server.dev/install.sh | sh

# Install Hugging Face & ML libraries
RUN pip3 install --no-cache-dir \
    transformers \
    datasets \
    evaluate \
    accelerate \
    torch \
    torchvision \
    torchaudio

# Set the working directory
WORKDIR /home/coder/project

# Expose the code-server port
EXPOSE 8080

# Start code-server without password for easy local access 
# (Note: Add --auth password for security in production)
CMD ["code-server", "--bind-addr", "0.0.0.0:8080", "--auth", "none"]