FROM rocm/vllm:latest

# Set environment variables to reduce prompts and enable UTF-8
ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1

# Install system dependencies
RUN apt-get update && apt-get install -y wget && rm -rf /var/lib/apt/lists/*

# Upgrade pip and install Python packages
RUN pip install --upgrade pip && \
    pip install diffusers peft==0.17.0 matplotlib jupyter transformers==4.56.1 gradio && \
    pip install huggingface_hub

RUN git clone https://github.com/ROCm/flash-attention.git && cd flash-attention && GPU_ARCHS=gfx942 python setup.py install

RUN pip install spaces easydict ftfy imageio 

RUN git clone https://huggingface.co/spaces/vivienfanghua/wan2.2_enhanced_amd && cd wan2.2_enhanced_amd

WORKDIR /app/wan2.2_enhanced_amd

ENTRYPOINT [ "python", "app.py"]