# RunPod AI Orchestrator Template
# Minimal Docker image for ComfyUI + vLLM orchestration
# Models and application code live on network volume at /workspace

FROM runpod/pytorch:2.4.0-py3.11-cuda12.4.1-devel-ubuntu22.04

# Install additional system utilities
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends \
  build-essential \
  python3-dev \
  python3-pip \
  python3-venv \
  git \
  curl \
  wget \
  htop \
  tmux \
  net-tools \
  ffmpeg \
  libavcodec-dev \
  libavformat-dev \
  libavutil-dev \
  libswscale-dev \
  bc \
  jq \
  openssh-server \
  && apt-get clean && rm -rf /var/lib/apt/lists/*

# Install Supervisor for process management
RUN pip install --upgrade pip && pip install --no-cache-dir supervisor

# Install Tailscale for VPN connectivity
RUN curl -fsSL https://tailscale.com/install.sh | sh

# Install yq
RUN wget https://github.com/mikefarah/yq/releases/latest/download/yq_linux_amd64 -O /usr/local/bin/yq && \
  chmod +x /usr/local/bin/yq

# Copy the startup script
COPY start.sh /start.sh
RUN chmod +x /start.sh

# Set working directory to /workspace (network volume mount point)
WORKDIR /workspace

# RunPod calls /start.sh by default
CMD ["/start.sh"]