FROM debian:11

#Tu token de hf en formato base64
ENV TOKEN=aGZfamlwc3Vsald5UXJRTkhvcWZqRHVweGRMdlNtbEhJbm1OTg==
ENV REPO_ID=HirCoir/piper-laura-neuronal
ENV MODELNAME=es_MX_laura-high
USER root
RUN useradd -m -u 1000 app
RUN apt-get update && apt-get install -y \
    python3 \
    python3-pip \
    python3-venv \
    git espeak-ng
WORKDIR /root
RUN git clone https://github.com/rhasspy/piper
RUN pip install -q cython>=0.29.0 espeak-phonemizer>=1.1.0 librosa>=0.9.2 numpy>=1.19.0 pytorch-lightning~=1.7.0 torch~=1.11.0
RUN pip install -q onnx onnxruntime
RUN pip install -q torchtext==0.12.0

WORKDIR /root/piper/src/python
RUN bash build_monotonic_align.sh
RUN pip install -q torchaudio==0.11.0 torchmetrics==0.11.4

RUN pip install -q huggingface_hub

WORKDIR /root/piper/src/python
COPY *.py .
RUN python3 download-model.py
RUN python3 -m piper_train.export_onnx model.ckpt ${MODELNAME}.onnx
RUN python3 upload-onnx.py
RUN mv ${MODELNAME}.onnx /home/app/${MODELNAME}.onnx
RUN mv ${MODELNAME}.onnx.json /home/app/${MODELNAME}.onnx.json
RUN chmod -R 777 /home/app
RUN pip install flask
USER app
WORKDIR /home/app
COPY server.py .
CMD python3 server.py