Fix: Update Dockerfile to handle permissions and cache directories

Dockerfile

@@ -2,8 +2,17 @@ FROM python:3.10-slim
 
 WORKDIR /code
 
-# Set environment variable to disable pip cache
+# Set environment variables
 ENV PIP_NO_CACHE_DIR=1
+ENV HOME=/home/user
+ENV PYTHONPATH=/code
+ENV HF_HOME=/home/user/huggingface
+ENV YOLO_CONFIG_DIR=/home/user/ultralytics
+
+# Create non-root user
+RUN useradd -m -u 1000 user && \
+    mkdir -p /home/user && \
+    chown -R user:user /home/user
 
 # Install system dependencies with cleanup
 RUN apt-get update && \
@@ -18,6 +27,14 @@ RUN apt-get update && \
     mkdir -p /tmp && \
     chmod 777 /tmp
 
+# Create cache directories with proper permissions
+RUN mkdir -p /home/user/.cache && \
+    mkdir -p /home/user/huggingface && \
+    mkdir -p /home/user/ultralytics && \
+    chown -R user:user /home/user/.cache && \
+    chown -R user:user /home/user/huggingface && \
+    chown -R user:user /home/user/ultralytics
+
 # Copy requirements first
 COPY requirements.txt .
 
@@ -30,14 +47,17 @@ RUN pip install --no-cache-dir --upgrade pip && \
     mkdir -p /tmp && \
     chmod 777 /tmp
 
-# Copy application code
-COPY . .
-
 # Create necessary directories with proper permissions
 RUN mkdir -p /tmp/models/parts/weights/weights && \
-    chmod -R 777 /tmp
+    chmod -R 777 /tmp && \
+    chown -R user:user /tmp/models
+
+# Copy application code
+COPY --chown=user:user . .
+
+# Switch to non-root user
+USER user
 
 EXPOSE 7860
 
-# Added --reload flag for development convenience
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860", "--reload"]
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]