FROM python:3.12.4-slim

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

WORKDIR $HOME/app

RUN pip install --user poetry

COPY --chown=user pyproject.toml ./
RUN /home/user/.local/bin/poetry install --no-root
RUN /home/user/.local/bin/poetry run pip install torch-scatter torch-sparse torch-cluster pyg-lib -f https://data.pyg.org/whl/torch-2.3.1+cu121.html
RUN /home/user/.local/bin/poetry run pip install torch-geometric

COPY --chown=user galis_app.py ./
COPY --chown=user model ./model
COPY --chown=user dataset ./dataset
COPY --chown=user predictor ./predictor
COPY --chown=user llm ./llm
COPY --chown=user embeddings_cache ./embeddings_cache

ENV GOOGLE_API_KEY=""

EXPOSE 7860

CMD ["/home/user/.local/bin/poetry", "run", "streamlit", "run", "galis_app.py", "--server.port=7860", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false", "--server.headless=true"]