Update Dockerfile

Dockerfile

@@ -1,38 +1,63 @@
-# Base image: Python 3.9 slim version
-FROM python:3.9-slim
+# Stage 1: Base image with Python
+FROM python:3.9-slim AS base
 
 # Set the working directory in the container
 WORKDIR /app
 
-# Install system dependencies
-# build-essential is needed for some Python packages that compile C code
-# git might be needed if your requirements.txt installs packages from git repos
+# Install essential system dependencies
+# build-essential is for compiling Python packages if needed
+# curl is for the healthcheck
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
-    git \
+    # Add any other specific system dependencies here if errors occur during pip install
     && rm -rf /var/lib/apt/lists/*
 
-# Copy the requirements file into the container
+# Copy the requirements file
 COPY requirements.txt ./
 
 # Install Python dependencies
+# Using --no-cache-dir to keep the image size smaller
 RUN pip3 install --no-cache-dir -r requirements.txt
 
-# Copy your application code into the container
-# If your app is just app.py and maybe a 'css' folder:
-COPY app.py ./
-COPY css/ ./css/
-# If your app code is in a 'src' directory:
-# COPY src/ ./src/
-# Ensure your app.py is then correctly referenced by the ENTRYPOINT, e.g., "src/app.py"
+# Stage 2: Application image (copies only necessary artifacts from base)
+FROM python:3.9-slim AS final
 
-# Expose the port your app runs on (Streamlit default is 8501)
+WORKDIR /app
+
+# Copy Python environment from the base stage (if you had a more complex venv setup, otherwise not strictly needed if pip install is here)
+# For simplicity with direct pip install, we can skip copying venv from another stage if not explicitly created.
+# We'll ensure dependencies are installed directly or ensure this stage also has them.
+# Re-installing system deps or copying from base are options. Let's keep it simple:
+
+RUN apt-get update && apt-get install -y \
+    curl \
+    # Minimal runtime system dependencies
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy installed Python packages from the 'base' stage's site-packages
+# This path might vary slightly depending on the python:3.9-slim image, find it with `python -m site`
+# A common path for global site-packages in slim images.
+# Adjust if your pip install location is different or if you prefer to re-run pip install here.
+COPY --from=base /usr/local/lib/python3.9/site-packages/ /usr/local/lib/python3.9/site-packages/
+COPY --from=base /usr/local/bin/ /usr/local/bin/
+
+# Create necessary directories for your application
+RUN mkdir -p /app/face_detector /app/css /app/images
+
+# Copy your application code and model files
+COPY app.py ./app.py
+COPY deploy.prototxt /app/face_detector/deploy.prototxt
+COPY res10_300x300_ssd_iter_140000.caffemodel /app/face_detector/res10_300x300_ssd_iter_140000.caffemodel
+COPY mask_detector.h5 ./mask_detector.h5
+COPY styles.css /app/css/styles.css
+COPY out.jpg /app/images/out.jpg
+
+# Expose the port Streamlit runs on
 EXPOSE 8501
 
-# Healthcheck for Streamlit (optional but good practice)
+# Healthcheck for Streamlit
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-# Command to run your application
-# This will run: streamlit run app.py --server.port=8501 --server.address=0.0.0.0
+# Set the entrypoint to run the Streamlit application
 ENTRYPOINT ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]