FROM python:3.9

WORKDIR /app

RUN apt-get update && apt-get install -y \
    build-essential \
    curl \
    software-properties-common \
    git \
    && rm -rf /var/lib/apt/lists/*

COPY ./requirements.txt /app/requirements.txt
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
RUN wget -O /app/MIN_RESNET101_BMI_Cache_test.pkl "https://github.com/ManavSarkar/Weight-Prediction-using-Machine-Learning/releases/download/dataset/MIN_RESNET101_BMI_Cache_test.pkl"
RUN wget -O /app/gpr_model.pkl "https://github.com/ManavSarkar/Weight-Prediction-using-Machine-Learning/releases/download/dataset/gpr_model.pkl"
RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
COPY . .
EXPOSE 8501

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

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