diff --git a/Dockerfile b/Dockerfile
index 3cd2bf1..f5ea993 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,18 +1,25 @@
 # Stage 1: Build
 FROM python:3.12-slim-bookworm as builder
 
+# Install build dependencies (gcc, Python headers, etc.)
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    gcc \
+    python3-dev \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set the working directory
 WORKDIR /app
+
+# Copy only the requirements file to take advantage of Docker's caching
 COPY requirements.txt .
+
+# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Stage 2: Runtime
 FROM python:3.12-slim-bookworm
 
-# Install system dependencies (if any are needed)
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
-
 # Set the working directory
 WORKDIR /app