Update Dockerfile

Dockerfile

@@ -14,6 +14,17 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Create data directory and set permissions
+RUN mkdir -p /code/data && \
+    chmod 777 /code/data
+
+# Create matplotlib config directory and set permissions
+RUN mkdir -p /tmp/matplotlib && \
+    chmod 777 /tmp/matplotlib
+
+# Set matplotlib config directory
+ENV MPLCONFIGDIR=/tmp/matplotlib
+
 # Copy requirements file
 COPY requirements.txt .
 
@@ -23,15 +34,19 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application code
 COPY . .
 
-# Create data directory
-RUN mkdir -p /code/data
-
 # Set environment variables
 ENV DATA_PATH=/code/data
 
 # Expose port
 EXPOSE 7860
 
+# Create non-root user
+RUN useradd -m -u 1000 appuser && \
+    chown -R appuser:appuser /code
+
+# Switch to non-root user
+USER appuser
+
 # Create entrypoint script
 RUN echo '#!/bin/bash\n\
 python app.py --data_path=/code/data' > /code/entrypoint.sh && \