FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04

# Metadata
LABEL maintainer="marcosremar2"
LABEL description="Ultravox Pipeline - RunPod GPU Worker with Snapshot Auto-Deploy"
LABEL version="1.0"

# Install required tools
RUN apt-get update && apt-get install -y \
    wget \
    curl \
    zstd \
    git \
    && rm -rf /var/lib/apt/lists/*

# Create directories
RUN mkdir -p /root/ultravox-pipeline /var/log

# Copy initialization script
COPY scripts/runpod_init.sh /root/runpod_init.sh
RUN chmod +x /root/runpod_init.sh

# Environment variables (can be overridden at runtime)
ENV SNAPSHOT_URL="https://huggingface.co/marcosremar2/ultravox-snapshots/resolve/main/ultravox-snapshot-latest.tar.zst"
ENV ULTRAVOX_PROFILE="gpu-prod"
ENV ULTRAVOX_SM_AUTHORIZED="1"

# Expose Service Manager port
EXPOSE 8888

# Entrypoint: Run initialization script
ENTRYPOINT ["/root/runpod_init.sh"]