# 1) Base off the prebuilt image that already has diffvg, PyTorch-SVGRender, etc.
FROM ximingxing/svgrender:v1

# 2) Switch to root so we can pip-install
USER root

# 3) Install the HF inference toolkit and any extras your handler needs
RUN pip install --no-cache-dir \
      huggingface_inference_toolkit \
      diffusers \
      transformers \
      pydantic

# 4) Copy your model code (handler.py, any .py files) into /app
WORKDIR /app
COPY . /app

# 5) Expose the port (HF Endpoints listens on 8000)
EXPOSE 8000

# 6) Run the HF Inference Toolkit server,
#    which will automatically pick up handler.py and requirements.txt
ENTRYPOINT ["python", "-u", "-m", "huggingface_inference_toolkit.server"]