Updated

Dockerfile

@@ -1,31 +1,43 @@
-# ----------------------
-# DevAssist AI Dockerfile
-# ----------------------
-
-# Base image
+# Use lightweight Python image
 FROM python:3.10-slim
 
+# Create a non-root user
+RUN useradd -m appuser
+
 # Set working directory
 WORKDIR /app
 
-# Create directories for caches and Chroma DB (writable by default user)
-RUN mkdir -p /app/huggingface_cache /app/Dev_Assist_SME_Builder_DB
+# Give appuser ownership of /app
+RUN chown -R appuser:appuser /app
+
+# Switch to appuser
+USER appuser
 
-# Copy requirements first (to leverage Docker cache)
-COPY requirements.txt .
+# Install system dependencies (needed for some models)
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements first for caching
+COPY --chown=appuser:appuser requirements.txt .
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy all project files
-COPY . .
+# Copy project files
+COPY --chown=appuser:appuser . .
+
+# Ensure HuggingFace cache directory exists and is writable
+RUN mkdir -p /app/huggingface_cache && chmod -R 777 /app/huggingface_cache
 
 # Expose FastAPI port
 EXPOSE 7860
 
-# Set environment variables for cache directories (HuggingFace & Chroma)
+# Environment variables for HuggingFace
 ENV HF_HOME=/app/huggingface_cache
-ENV CHROMA_DB_DIR=/app/Dev_Assist_SME_Builder_DB
+ENV TRANSFORMERS_CACHE=/app/huggingface_cache
 
-# Default command to start FastAPI via Uvicorn
+# Run FastAPI
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]