ARG UV_VERSION="0.6.14"

FROM ghcr.io/astral-sh/uv:${UV_VERSION} AS uv-fetcher

FROM python:3.11-slim AS runtime

RUN apt-get update && apt-get install -y --no-install-recommends \
    wget build-essential git && \
    rm -rf /var/lib/apt/lists/*

COPY --from=uv-fetcher /uv /uvx /bin/

ENV VIRTUAL_ENV=/modules/.venv
RUN uv venv "$VIRTUAL_ENV" && . "$VIRTUAL_ENV/bin/activate"
ENV PATH="$VIRTUAL_ENV/bin:$PATH"

# Set the working directory to the user's home directory
WORKDIR /app

# Create a non-root user and set permissions
RUN useradd -m -u 1000 appuser && \
    chown -R appuser:appuser /app

# Switch to non-root user
USER appuser

COPY pyproject.toml uv.lock README.md ./
COPY folding-studio ./folding-studio

RUN uv sync --locked --no-install-project --no-dev

COPY folding_studio_demo ./folding_studio_demo
RUN uv sync --locked --no-dev

# COPY pred.cif /app/boltz_results/pred_model_0.cif
# COPY plddt_0.npz /app/boltz_results/plddt_0.npz
COPY spr_af_scores_mapped.csv /app/spr_af_scores_mapped.csv

EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"
ARG FOLDING_PROJECT_CODE="folding_dev"
ENV FOLDING_PROJECT_CODE=$FOLDING_PROJECT_CODE

CMD ["uv", "run", "folding-studio-demo"]