Update Dockerfile

Dockerfile

@@ -1,6 +1,9 @@
 # Use the official Jupyter Docker image
 FROM jupyter/base-notebook:latest
 
+# Switch to the root user to perform privileged actions
+USER root
+
 # Install Jupyter Lab and additional packages
 RUN pip install --no-cache-dir \
     jupyterlab \
@@ -13,8 +16,14 @@ RUN pip install --no-cache-dir \
     scikit-learn \
     pandas
 
-# Expose the port Jupyter Lab will run on
-EXPOSE 8888
+# Copy the app.py script to the container
+COPY app.py /home/jovyan/app.py
+
+# Expose the ports Jupyter Lab and the redirect server will run on
+EXPOSE 8888 8000
+
+# Switch back to the notebook user
+USER $NB_UID
 
-# Start Jupyter Lab
-CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root", "--NotebookApp.token=''"]
+# Start the servers using app.py
+CMD ["python", "/home/jovyan/app.py"]