FROM python:3.11

ENV DEBIAN_FRONTEND=noninteractive
SHELL ["/bin/bash", "-c"]
RUN apt-get update -y \
    && apt-get install -y build-essential python3-dev r-base make apt-utils unzip gpg doxygen git curl aria2 vim screen rsync wget locales gfortran mafft libglew-dev libeigen3-dev \
    libpng-dev libfreetype6-dev libxml2-dev \
    libmsgpack-dev python3-pyqt5.qtopengl libglm-dev libnetcdf-dev \
    && locale-gen en_US.UTF-8 \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /app

COPY . .

RUN pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
RUN pip3 install -r requirements.txt

# Ensure Hugging Face cache directory exists and is writable
RUN mkdir -p /app/.cache && chmod 777 /app/.cache
RUN mkdir -p /app/.cache/offload && chmod 777 /app/.cache/offload

EXPOSE 8501

HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

ENV HF_HOME=/app/.cache

ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]