FROM python:3.11
#
# Install system dependencies
RUN apt-get update && apt-get install -y \
    libgirepository1.0-dev \
    gobject-introspection \
    pkg-config \
    libdbus-1-dev \
    libcairo2-dev \
    ffmpeg \
    git \
    curl \
    wget \
    build-essential \
    python3-dev \
    libgmp-dev \
    libmpfr-dev \
    libmpc-dev \
    libopenblas-dev \
    libsndfile1-dev \
    libavcodec-dev \
    libavformat-dev \
    libswscale-dev \
    libatlas-base-dev \
    && rm -rf /var/lib/apt/lists/*

# Create user and workspace
RUN useradd -m -u 1000 user && \
    mkdir -p /content && \
    chown user:user /content

USER user
ENV PATH="/home/user/.local/bin:$PATH"
WORKDIR /content/

# Clone repositories
RUN git clone https://github.com/noblebarkrr/mvsepless -b alpha_hf
# Install Python dependencies in optimal order
RUN pip install --user --upgrade pip setuptools wheel && \
    pip install --user gdown && \
    pip install --user -r mvsepless/requirements.txt && \
    pip install --user https://github.com/Bebra777228/TrainVocModel-EN/releases/download/fixed-packages/fairseq_fixed-0.13.0-cp311-cp311-linux_x86_64.whl


# Download models
RUN wget -O mvsepless/models/medley_vox/pretrained_models/xlsr_53_56k.pt https://dl.fbaipublicfiles.com/fairseq/wav2vec/xlsr_53_56k.pt
WORKDIR /content/mvsepless/
RUN python download_models.py

# Fix potential permission issues
RUN chmod +x multi_infer.py

EXPOSE 7860
CMD ["python", "multi_infer.py", "-gr", "-hface"]