FROM python:3.10

WORKDIR /app

# 1. Install system tools
RUN apt-get update && apt-get install -y git wget && rm -rf /var/lib/apt/lists/*

# 2. Install Python libraries (Added diffusers, transformers, accelerate)
RUN pip install --no-cache-dir \
    torch==2.3.0 \
    torchvision \
    torchaudio \
    onnx \
    onnxscript \
    safetensors \
    huggingface_hub \
    einops \
    sentencepiece \
    ftfy \
    regex \
    easydict \
    gdown \
    diffusers \
    transformers \
    accelerate

# 3. Clone Wan2.1 Repo
RUN git clone https://github.com/Wan-Video/Wan2.1.git

# 4. Copy our conversion script
COPY convert_wan.py /app/convert_wan.py

# 5. Create output directory
RUN mkdir -p /app/output

# 6. RUN command
CMD python convert_wan.py && cd /app/output && python -m http.server 7860