Update Dockerfile

Dockerfile

@@ -23,16 +23,18 @@ RUN mkdir -p $NLTK_DATA
 # Download NLTK data as root (with permissions) and handle SSL issues
 RUN python -c "import ssl; ssl._create_default_https_context = ssl._create_unverified_context; import nltk; nltk.download('stopwords', download_dir='/usr/local/share/nltk_data'); nltk.download('wordnet', download_dir='/usr/local/share/nltk_data'); nltk.download('omw-1.4', download_dir='/usr/local/share/nltk_data')"
 
-# Copy the application code
-COPY . .
+# Create a non-root user first
+RUN useradd -m -u 1000 user
 
-# Create necessary directories with proper permissions
-RUN mkdir -p models artifacts vectorizers
-RUN chmod -R 755 models artifacts vectorizers
+# Copy the application code
+COPY --chown=user:user . .
 
-# Create a non-root user for security
-RUN useradd -m -u 1000 user
+# Create necessary directories with proper permissions for the user
+RUN mkdir -p /app/models /app/artifacts /app/vectorizers
 RUN chown -R user:user /app
+RUN chmod -R 755 /app
+
+# Switch to non-root user
 USER user
 
 # Set environment variables
@@ -44,5 +46,8 @@ ENV HOME=/home/user \
 # Expose the port that Streamlit runs on
 EXPOSE 8501
 
+# Health check
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+
 # Run the application
 CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]