Update Dockerfile

Dockerfile

@@ -1,39 +1,39 @@
 FROM python:3.9-slim
 
-# Set working directory to /app inside the container
 WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
-    software-properties-common \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# Create the .streamlit directory to avoid permission errors
-RUN mkdir -p /app/.streamlit
+# Create directories
+RUN mkdir -p /app/.streamlit /app/.cache
 
-# Copy .streamlit config folder (including config.toml)
-COPY .streamlit/ .streamlit/
+# Set cache environment variables
+ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_HOME=/app/.cache
 
-# Copy app and config files
+# Copy files
+COPY .streamlit/ .streamlit/
 COPY app/ ./
+COPY chromadb_model/ chromadb_model/
 COPY requirements.txt ./
 
 # Install Python dependencies
 RUN pip3 install -r requirements.txt
 
-# Expose port 8501 (Streamlit Spaces requirement)
-EXPOSE 8501
+# Download NLTK data
+RUN python -c "import nltk; nltk.download('punkt'); nltk.download('punkt_tab')"
 
-# Set environment variables for Streamlit
+# Streamlit configuration
 ENV STREAMLIT_CONFIG_DIR=/app/.streamlit
 ENV STREAMLIT_SERVER_HEADLESS=true
 ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
 
-# Healthcheck endpoint for Spaces
+EXPOSE 8501
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-# Run Streamlit with port 8501 (DO NOT override port)
 ENTRYPOINT ["streamlit", "run", "main.py", "--server.address=0.0.0.0"]