Fix: Add python-dotenv dependency and fix matplotlib permissions with non-root user

Dockerfile

@@ -1,6 +1,9 @@
 # Optimized Dockerfile for Hugging Face Spaces - Reduced size
 FROM python:3.10-slim
 
+# Create non-root user
+RUN useradd -m -u 1000 user
+
 # Set working directory
 WORKDIR /app
 
@@ -13,23 +16,26 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && apt-get clean
 
 # Copy requirements first for better caching
-COPY requirements.txt ./
+COPY --chown=user:user requirements.txt ./
+
+# Switch to user before installing Python packages
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
 
 # Install Python dependencies with optimizations
 RUN pip install --no-cache-dir -r requirements.txt && \
-    pip cache purge && \
-    rm -rf /root/.cache/pip
+    pip cache purge
 
 # Download model checkpoint from Google Drive
 RUN pip install --no-cache-dir gdown && \
     gdown --id 1ftzxTJUnlxpQFqPlaUozG_JUbl1Qi5tQ -O /app/model_checkpoint.ckpt && \
     pip uninstall -y gdown && \
-    rm -rf /root/.cache/pip
+    pip cache purge
 
 # Copy application files
-COPY app.py inference_core.py ./
-COPY scripts/ ./scripts/
-COPY configs/ ./configs/
+COPY --chown=user:user app.py inference_core.py ./
+COPY --chown=user:user scripts/ ./scripts/
+COPY --chown=user:user configs/ ./configs/
 
 # Create output directories
 RUN mkdir -p api_inference_pred_masks_pipeline \
@@ -42,7 +48,8 @@ EXPOSE 7860
 # Set environment variables
 ENV PYTHONUNBUFFERED=1 \
     PORT=7860 \
-    PYTHONDONTWRITEBYTECODE=1
+    PYTHONDONTWRITEBYTECODE=1 \
+    MPLCONFIGDIR=/tmp/matplotlib
 
 # Start Flask app
 CMD ["python", "app.py"]

requirements.txt

@@ -23,6 +23,7 @@ cloudinary
 # Configuration
 omegaconf
 pyyaml
+python-dotenv
 
 # Utilities
 requests