FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1

WORKDIR /app

# Install system dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends \
        python3 \
        python3-pip \
        ffmpeg \
        git \
        build-essential \
        python3-dev && \
    rm -rf /var/lib/apt/lists/*

RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

COPY . .

RUN pip install --no-cache-dir . whisper-timestamped diart

# Expose port for Hugging Face Spaces
EXPOSE 7860

# Add health check for better debugging
HEALTHCHECK --interval=30s --timeout=5s --start-period=10s \
  CMD curl -f http://localhost:7860/health || exit 1

ENTRYPOINT ["whisperlivekit-server", "--host", "0.0.0.0", "--port", "7860"]

# Default args for CZ transcription
CMD ["--model", "tiny", "--language", "cs"]