diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
new file mode 100644
index 0000000..3994254
--- /dev/null
+++ b/.github/workflows/docker-build.yml
@@ -0,0 +1,22 @@
+name: Build Docker Image
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Build Docker image
+        run: |
+          docker build --build-arg COHERE_API_KEY=${{ secrets.COHERE_API_KEY }} -t studyforge-api .
diff --git a/Dockerfile b/Dockerfile
index 8eb0817..0004143 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,6 +2,9 @@ FROM python:3.11-slim AS base
 
 WORKDIR /app
 
+ARG COHERE_API_KEY
+ENV COHERE_API_KEY=${COHERE_API_KEY}
+
 RUN apt-get update && apt-get install -y \
     build-essential \
     gcc \