Update Dockerfile

Dockerfile

@@ -4,6 +4,7 @@ WORKDIR /app
 
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
+ENV DOCKER_ENV=true
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -16,13 +17,18 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# Pre-download embedding models
-RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
-RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-mpnet-base-v2')"
+# Pre-download embedding models with correct names
+RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')" || echo "Failed to download all-MiniLM-L6-v2"
+RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-mpnet-base-v2')" || echo "Failed to download all-mpnet-base-v2"
 
-# Create writable directories in /tmp (this is the key change)
-RUN mkdir -p /tmp/researchmate/data /tmp/researchmate/logs /tmp/researchmate/chroma_persist \
-    /tmp/researchmate/uploads /tmp/researchmate/chroma_db && \
+# Create writable directories in /tmp
+RUN mkdir -p /tmp/researchmate/data \
+             /tmp/researchmate/logs \
+             /tmp/researchmate/chroma_persist \
+             /tmp/researchmate/uploads \
+             /tmp/researchmate/chroma_db \
+             /tmp/researchmate/config \
+             /tmp/researchmate/tmp && \
     chmod -R 777 /tmp/researchmate
 
 # Set environment variables for writable paths
@@ -32,9 +38,29 @@ ENV CHROMA_DIR=/tmp/researchmate/chroma_persist
 ENV UPLOADS_DIR=/tmp/researchmate/uploads
 ENV CHROMA_DB_DIR=/tmp/researchmate/chroma_db
 
-# Set matplotlib config directory to writable location
+# Set all cache directories to writable locations
 ENV MPLCONFIGDIR=/tmp/matplotlib
-RUN mkdir -p /tmp/matplotlib && chmod 777 /tmp/matplotlib
+ENV TRANSFORMERS_CACHE=/tmp/transformers
+ENV HF_HOME=/tmp/huggingface
+ENV SENTENCE_TRANSFORMERS_HOME=/tmp/sentence_transformers
+ENV HF_DATASETS_CACHE=/tmp/datasets
+ENV HUGGINGFACE_HUB_CACHE=/tmp/huggingface_hub
+ENV XDG_CACHE_HOME=/tmp/cache
+
+RUN mkdir -p /tmp/matplotlib \
+             /tmp/transformers \
+             /tmp/huggingface \
+             /tmp/sentence_transformers \
+             /tmp/datasets \
+             /tmp/huggingface_hub \
+             /tmp/cache && \
+    chmod -R 777 /tmp/matplotlib \
+                 /tmp/transformers \
+                 /tmp/huggingface \
+                 /tmp/sentence_transformers \
+                 /tmp/datasets \
+                 /tmp/huggingface_hub \
+                 /tmp/cache
 
 # Copy application code
 COPY . .