Update Dockerfile

Dockerfile

@@ -9,20 +9,30 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Create a non-root user
+RUN useradd -m -u 1000 user
 
-# Hugging Face cache config
-ENV TRANSFORMERS_CACHE=/app/cache/huggingface
-ENV HF_HOME=/app/cache/huggingface
-ENV HUGGINGFACE_HUB_CACHE=/app/cache/huggingface
+# Set cache directories to writable locations
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    HF_HOME=/home/user/.cache/huggingface \
+    STREAMLIT_SERVER_HEADLESS=true \
+    STREAMLIT_SERVER_ENABLE_CORS=false \
+    STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION=false \
+    STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
 
-RUN mkdir -p /app/cache/huggingface && chmod -R 777 /app/cache
+# Switch to user
+USER user
 
-RUN pip3 install -r requirements.txt
+# Copy files
+COPY --chown=user requirements.txt ./
+COPY --chown=user src/ ./src/
+
+# Install dependencies
+RUN pip3 install --user --no-cache-dir -r requirements.txt
 
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]