FROM rocm/pytorch:rocm6.3.2_ubuntu24.04_py3.12

WORKDIR /app

RUN apt-get update && apt-get install -y ffmpeg && rm -rf /var/lib/apt/lists/*

RUN pip install --no-cache-dir \
    torch transformers accelerate \
    boson_multimodal \
    soundfile librosa \
    pyannote.audio \
    llama-cpp-python \
    numpy aiofiles \
    sqlalchemy aiosqlite \
    pydantic pydantic-settings \
    celery[redis] redis

ENV PYTHONPATH=/app
ENV HUGGINGFACE_HUB_CACHE=/data/hf_cache

COPY backend/ .

CMD ["celery", "-A", "app.tasks.celery_app", "worker", "--loglevel=info", "--concurrency=1"]