FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime

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

# 2. clone Matrix-Game
RUN git clone --depth 1 https://github.com/SkyworkAI/Matrix-Game.git /workspace/Matrix-Game

# 3. install its dependencies & the package itself
RUN pip install --no-cache-dir -r /workspace/Matrix-Game/Matrix-Game-2/requirements.txt \

# 4. gradio UI
RUN pip install --no-cache-dir gradio

# 5. copy the simple interface you already created
COPY app.py /app.py
CMD ["python", "/app.py"]