Update Dockerfile

Dockerfile

@@ -1,13 +1,14 @@
 FROM python:3.13-slim
 
-
+# Set working directory
 WORKDIR /app
 
-
+# Set environment variables
 ENV HOME=/app
 ENV XDG_CONFIG_HOME=/app/.streamlit
+ENV NLTK_DATA=/app/nltk_data
 
-
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -15,10 +16,11 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-
+# Create necessary directories
 RUN mkdir -p /app/.streamlit && chmod -R 777 /app/.streamlit
+RUN mkdir -p /app/nltk_data
 
-
+# Copy files
 COPY requirements.txt ./
 COPY src/ ./src/
 COPY src/c_d.csv ./src/
@@ -26,14 +28,17 @@ COPY src/logistic_models.pkl ./src/
 COPY src/tfidf.pkl ./src/
 COPY src/multilabels.pkl ./src/
 
-
+# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Download NLTK data during build
+RUN python -m nltk.downloader punkt averaged_perceptron_tagger wordnet stopwords
 
+# Expose Streamlit port
 EXPOSE 8501
 
-
+# Add healthcheck
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-
+# Run the app
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]