Update Dockerfile

Dockerfile

@@ -1,8 +1,8 @@
-# Stage 1: Build stage (only used if compiling dependencies)
+# Stage 1: Build stage (if needed for compiling specific dependencies)
 FROM python:3.9-slim as build-stage
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y libmagic1 file
+# Install system dependencies required for Python packages
+RUN apt-get update && apt-get install -y build-essential
 
 # Set the working directory
 WORKDIR /app
@@ -13,13 +13,17 @@ COPY requirements.txt .
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+
 # Stage 2: Final stage
 FROM python:3.9-slim
 
+# Install system dependencies required at runtime
+RUN apt-get update && apt-get install -y libmagic1 file && apt-get clean
+
 # Set the working directory in the container
 WORKDIR /app
 
-# Copy the contents from the build stage (dependencies only)
+# Copy Python dependencies from the build stage
 COPY --from=build-stage /usr/local/lib/python3.9/site-packages /usr/local/lib/python3.9/site-packages
 COPY --from=build-stage /usr/local/bin /usr/local/bin
 
@@ -29,12 +33,9 @@ RUN useradd -m flaskuser
 # Copy application code
 COPY . /app
 
-# Change ownership of /app to flaskuser after user is created
+# Change ownership of /app to flaskuser
 RUN chown -R flaskuser:flaskuser /app
 
-# Remove pip cache and unnecessary files (optional)
-RUN apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
-
 # Switch to the non-root user
 USER flaskuser
 
@@ -42,5 +43,4 @@ USER flaskuser
 EXPOSE 7860
 
 # Run the Flask app using gunicorn with an infinite request timeout
-# CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--timeout", "0", "app:app"]
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--timeout", "0", "--access-logfile", "-", "--error-logfile", "-", "app:app"]
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--timeout", "0", "--access-logfile", "-", "--error-logfile", "-", "app:app"]