FROM python:3.10-slim

WORKDIR /app

RUN apt-get update && \
    apt-get install -y git curl git-lfs && \
    rm -rf /var/lib/apt/lists/*

RUN uname -a

RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.python.sh | bash

COPY . /app

RUN which git-lfs

RUN pip install graphrag==1.0.0 fastapi uvicorn

COPY search.py /usr/local/lib/python3.10/site-packages/graphrag/query/structured_search/local_search/search.py

RUN chmod +x /app/start.sh

RUN git clone https://huggingface.co/datasets/nameliu/graphrag-data && \
    cd graphrag-data && \
    git checkout master && \
    git-lfs pull

EXPOSE 8080

CMD ["/app/start.sh"]