Update Dockerfile

Dockerfile

@@ -1,26 +1,36 @@
-# Use an official Python runtime as a parent image
-FROM python:3.10-slim
+# Stage 1: Build
+FROM python:3.10-slim AS builder
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Set environment variables
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONUNBUFFERED=1
+# Set environment variables for user installation
+ENV PATH=/root/.local/bin:$PATH
 
 # Set the working directory in the container
 WORKDIR /app
 
-# Copy the requirements file first
+# Copy the requirements file first for caching
 COPY requirements.txt /app/
 
 # Install dependencies
 RUN pip install --no-cache-dir --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the current directory contents into the container
+# Stage 2: Production
+FROM python:3.10-slim
+
+# Set environment variables for user installation
+ENV PATH=/root/.local/bin:$PATH
+ENV PYTHONUNBUFFERED=1
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy only the necessary files from the builder stage
+COPY --from=builder /root/.local /root/.local
 COPY . /app
 
 # Expose the port that the FastAPI app runs on