FROM python:3.11-slim

# Mettre à jour pip
RUN pip install -U pip

ARG TARGETPLATFORM

# Mise à jour des paquets
RUN apt-get update

# Installation des dépendances
RUN apt-get install --no-install-recommends -y curl ffmpeg

# Installation de Rust pour certaines configurations (exemple : non linux/amd64)
RUN if [ "$TARGETPLATFORM" != "linux/amd64" ]; then \
        apt-get install --no-install-recommends -y build-essential && \
        curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y; \
    fi

# Nettoyage des caches
RUN apt-get clean && rm -rf /var/lib/apt/lists/*

ENV PATH="/root/.cargo/bin:${PATH}"

WORKDIR /app
RUN mkdir -p voices config

ARG USE_ROCM
ENV USE_ROCM=${USE_ROCM}

COPY requirements*.txt /app/
RUN if [ "${USE_ROCM}" = "1" ]; then mv /app/requirements-rocm.txt /app/requirements.txt; fi
RUN pip install -r requirements.txt

COPY *.py *.sh *.default.yaml README.md LICENSE /app/

ARG PRELOAD_MODEL
ENV PRELOAD_MODEL=${PRELOAD_MODEL}
ENV TTS_HOME=voices
ENV HF_HOME=voices
ENV COQUI_TOS_AGREED=1

CMD bash startup.sh