feat: Modified dockerfiles to use uv instead of pip

Author: anibalrojosan

config/Dockerfile.api

@@ -1,15 +1,42 @@
-FROM python:3.12-slim
+# 1. Use a "builder" stage to install dependencies
+FROM python:3.12-slim AS builder
+
+# Copy the uv binary from the official distroless image
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 
 WORKDIR /app
 
+# Create a virtual environment
+RUN uv venv
+
+# Copy dependency definitions
 COPY pyproject.toml uv.lock ./
-RUN pip install uv && \
-    uv sync --system --no-cache --frozen-lock
 
+# Install dependencies into the virtual environment, excluding the project itself.
+RUN . .venv/bin/activate && uv sync --locked --no-install-project
+
+# Copy the application source code
 COPY src/ ./src/
+
+# Install the project itself into the venv.
+# Since dependencies are already installed, this step is very fast.
+RUN . .venv/bin/activate && uv pip install .
+
+# 2. Use a clean "final" stage for the runtime environment
+FROM python:3.12-slim
+
+WORKDIR /app
+
+# Copy the virtual environment with all dependencies from the builder stage
+COPY --from=builder /app/.venv ./.venv
+
+# Copy other necessary assets
 COPY models/model.joblib ./models/
 COPY data/data.csv ./data/
 
+# Activate the virtual environment by adding it to the PATH
+ENV PATH="/app/.venv/bin:$PATH"
+
 EXPOSE 5000
 
 CMD ["python", "-m", "gunicorn", "--bind", "0.0.0.0:5000", "src.app:app"]

config/Dockerfile.streamlit

@@ -1,13 +1,38 @@
-FROM python:3.12-slim
+# 1. Use a "builder" stage to install dependencies
+FROM python:3.12-slim AS builder
+
+# Copy the uv binary from the official distroless image
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 
 WORKDIR /app
 
+# Create a virtual environment
+RUN uv venv
+
+# Copy dependency definitions
 COPY pyproject.toml uv.lock ./
-RUN pip install uv && \
-    uv sync --system --no-cache --frozen-lock
 
+# Install dependencies into the virtual environment, excluding the project itself.
+RUN . .venv/bin/activate && uv sync --locked --no-install-project
+
+# Copy the application source code
 COPY src/ ./src/
 
+# Install the project itself into the venv.
+# Since dependencies are already installed, this step is very fast.
+RUN . .venv/bin/activate && uv pip install .
+
+# 2. Use a clean "final" stage for the runtime environment
+FROM python:3.12-slim
+
+WORKDIR /app
+
+# Copy the virtual environment with all dependencies from the builder stage
+COPY --from=builder /app/.venv ./.venv
+
+# Activate the virtual environment by adding it to the PATH
+ENV PATH="/app/.venv/bin:$PATH"
+
 EXPOSE 8501
 
 CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]

pyproject.toml

@@ -6,6 +6,7 @@ readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
     "flask>=3.1.2",
+    "gunicorn>=22.0.0",
     "joblib>=1.5.2",
     "pandas>=2.3.3",
     "pytest>=8.4.2",