Update Dockerfile

Dockerfile

@@ -1,20 +1,40 @@
-# Use an official Python runtime as a parent image
+# Stage 1: Build stage (only used if compiling dependencies)
+FROM python:3.9-slim as build-stage
+
+# Set the working directory
+WORKDIR /app
+
+# Install build dependencies (only if needed, otherwise skip this stage)
+# RUN apt-get update && apt-get install -y build-essential gcc
+
+# Copy requirements file
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Stage 2: Final stage
 FROM python:3.9-slim
 
 # Set the working directory in the container
 WORKDIR /app
 
-# Switch to root user to ensure installation permissions
-USER root
+# Copy the contents from the build stage (dependencies only)
+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
 
-# Copy the current directory contents into the container at /app
+# Copy application code
 COPY . /app
 
-# Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+# Remove pip cache and unnecessary files (optional)
+RUN apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+# Add non-root user and switch to it for better security
+RUN useradd -m flaskuser
+USER flaskuser
 
-# Make port 7860 available to the world outside this container
+# Expose the port your Flask app will run on
 EXPOSE 7860
 
-# Run app.py when the container launches
+# Run the Flask app
 CMD ["python", "app.py"]