assets/docker/Dockerfile.rocm

FROM python:3.12-slim

EXPOSE 7860

RUN apt-get update && apt-get install -y ffmpeg && apt-get install -y libportaudio2 && apt-get clean

WORKDIR /app

COPY . .

RUN python3 -m venv /app/.venv && \
    . /app/.venv/bin/activate && \
    python -c "from main.app.install import remove_onnxruntime;remove_onnxruntime()" && \
    pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir six packaging python-dateutil platformdirs pywin32 && \
    pip install --no-cache-dir torch torchaudio torchvision --index-url https://download.pytorch.org/whl/rocm7.2 && \
    pip install --no-cache-dir onnxruntime-rocm && \
    pip install --no-cache-dir -r requirements.txt && \
    pip install --no-cache-dir faiss-cpu==1.13.2

VOLUME [ "/app/assets" ]
ENV PATH="/app/.venv/bin:$PATH"

CMD ["python3", "main/app/app.py", "--client"]