Merge pull request #125 from surfiniaburger/main

Updated dockerfile

Author: pankit-eng

scripts/download_dataset.py

@@ -6,13 +6,14 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
-# Install all necessary Python packages for the server, including gunicorn
-RUN pip install --no-cache-dir \
-    fastapi>=0.104.0 \
-    "uvicorn[standard]>=0.24.0" \
-    requests>=2.25.0 \
-    wsproto>=1.0.0 \
-    gunicorn
+# Set the working directory
+WORKDIR /app
+
+# Copy requirements file and install dependencies. This is done in a separate
+# step to leverage Docker's layer caching. Dependencies are only re-installed
+# when the requirements.txt file changes.
+COPY src/envs/dipg_safety_env/server/requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Set the working directory and PYTHONPATH inside the container
 WORKDIR /app
@@ -31,4 +32,4 @@ HEALTHCHECK --interval=60s --timeout=10s --start-period=180s --retries=3 \
 
 
 # Note: The DIPG_DATASET_PATH must be provided when running this container.
-CMD ["gunicorn", "-w", "4", "-k", "uvicorn.workers.UvicornWorker", "-b", "0.0.0.0:8000", "envs.dipg_safety_env.server.app:app"]
+CMD ["gunicorn", "-w", "4", "-k", "uvicorn.workers.UvicornWorker", "-b", "0.0.0.0:8000", "envs.dipg_safety_env.server.app:app"]

src/envs/dipg_safety_env/server/Dockerfile

@@ -0,0 +1,5 @@
+fastapi==0.104.0
+uvicorn[standard]==0.24.0
+requests==2.25.0
+wsproto==1.0.0
+gunicorn==22.0.0