Update Dockerfile

Dockerfile

@@ -1,3 +1,4 @@
+# Use a minimal Python base image
 FROM python:3.9-slim
 
 # Install system dependencies
@@ -9,25 +10,36 @@ RUN apt-get update && apt-get install -y \
     poppler-utils \
     && rm -rf /var/lib/apt/lists/*
 
-# Set environment variables to avoid permission issues
-ENV MPLCONFIGDIR=/tmp/matplotlib
-ENV YOLO_CONFIG_DIR=/tmp/Ultralytics
-ENV XDG_CACHE_HOME=/tmp
-ENV FONTCONFIG_PATH=/tmp
+# Create a non-root user (Hugging Face Spaces requirement for security)
+RUN useradd -m -u 1000 user
+USER user
+
+# Set environment variables to avoid permission issues and optimize performance
+ENV HOME=/home/user \
+    MPLCONFIGDIR=/tmp/matplotlib \
+    YOLO_CONFIG_DIR=/tmp/ultralytics \
+    XDG_CACHE_HOME=/tmp \
+    FONTCONFIG_PATH=/tmp \
+    PATH=/home/user/.local/bin:$PATH \
+    PYTHONUNBUFFERED=1  # Ensure logs are not buffered for debugging \
+    NUM_WORKERS=2  # Default number of workers (adjustable via Hugging Face settings)
+
+# Create necessary directories
+RUN mkdir -p /tmp/matplotlib /tmp/ultralytics /tmp/fontconfig /tmp/uploads
 
-RUN mkdir -p /tmp/matplotlib /tmp/ultralytics /tmp/fontconfig
 # Set working directory
 WORKDIR /app
 
 # Copy project files
 COPY . .
 
-# Install Python dependencies
-RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
+# Install Python dependencies with performance optimizations
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt && \
+    pip install gunicorn==21.2.0
 
 # Expose Hugging Face Spaces default port
 EXPOSE 7860
 
-# Run FastAPI app with Uvicorn
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run FastAPI app with Uvicorn workers via gunicorn
+CMD ["gunicorn", "-w", "${NUM_WORKERS}", "-k", "uvicorn.workers.UvicornWorker", "app:app", "--bind", "0.0.0.0:7860", "--log-level", "info"]