FROM ghcr.io/agent-axiom/aximo:main

USER root

RUN apt-get update \
    && apt-get install -y --no-install-recommends curl tar \
    && rm -rf /var/lib/apt/lists/*

COPY aximo.toml /app/config/aximo.toml

RUN mkdir -p /var/lib/aximo/models /tmp/aximo-model \
    && curl -fL https://blob.handy.computer/parakeet-v3-int8.tar.gz -o /tmp/parakeet-v3-int8.tar.gz \
    && tar -xzf /tmp/parakeet-v3-int8.tar.gz -C /tmp/aximo-model \
    && if [ -d /tmp/aximo-model/parakeet-tdt-0.6b-v3-int8 ]; then \
         cp -R /tmp/aximo-model/parakeet-tdt-0.6b-v3-int8 /var/lib/aximo/models/; \
       else \
         first_dir="$(find /tmp/aximo-model -mindepth 1 -maxdepth 1 -type d | head -n 1)" && \
         cp -R "$first_dir" /var/lib/aximo/models/parakeet-tdt-0.6b-v3-int8; \
       fi \
    && test -f /var/lib/aximo/models/parakeet-tdt-0.6b-v3-int8/encoder-model.int8.onnx \
    && test -f /var/lib/aximo/models/parakeet-tdt-0.6b-v3-int8/decoder_joint-model.int8.onnx \
    && test -f /var/lib/aximo/models/parakeet-tdt-0.6b-v3-int8/nemo128.onnx \
    && test -f /var/lib/aximo/models/parakeet-tdt-0.6b-v3-int8/vocab.txt \
    && rm -rf /tmp/parakeet-v3-int8.tar.gz /tmp/aximo-model \
    && chown -R aximo:aximo /app /var/lib/aximo

ENV AXIMO_CONFIG=/app/config/aximo.toml
ENV AXIMO_SERVER_HOST=0.0.0.0
ENV AXIMO_SERVER_PORT=7860
ENV ORT_LOG=error

USER aximo

EXPOSE 7860

CMD ["aximo"]