FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime

# Set working directory
WORKDIR /app

# Verify Python version meets requirements (>=3.9.0, <3.12)
RUN python --version

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    libsndfile1 \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

# Install Cython first (crucial for building extensions)
RUN pip install --no-cache-dir Cython packaging

# Clone the repository
RUN git clone https://github.com/PranavDBhat/LIMMITS-24-Coquiai.git /app/LIMMITS-24-Coquiai

# Install only the main requirements (not dev/notebooks requirements)
# This directly uses the requirements.txt file from the repository
RUN cd /app/LIMMITS-24-Coquiai && \
    pip install --no-cache-dir -r requirements.txt

# Install the package in development mode
# This will build the Cython extensions
RUN cd /app/LIMMITS-24-Coquiai && \
    pip install -e .

# Create directories for models, input, and output
RUN mkdir -p /app/models /app/data/input /app/data/output

# Copy the inference script
COPY infer.py /app/

# Set the entrypoint to run the inference script
ENTRYPOINT ["python", "infer.py"]

# Default command (can be overridden)
CMD ["--help"]