Update Dockerfile

Dockerfile

@@ -9,22 +9,30 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Copy files
 COPY requirements.txt ./
 COPY src/ ./src/
+COPY .streamlit/ /data/.streamlit/
 
+# Install Python dependencies
 RUN pip3 install -r requirements.txt
 
+# Expose streamlit port
 EXPOSE 8501
 
+# Healthcheck for Hugging Face Spaces
 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"]
-
+# Environment variables for cache & config directories
 ENV TRANSFORMERS_CACHE=/data/cache/transformers
 ENV HF_HOME=/data/cache/huggingface
 ENV TORCH_HOME=/data/cache/torch
 ENV SENTENCE_TRANSFORMERS_HOME=/data/cache/sentence_transformers
 ENV STREAMLIT_CONFIG_DIR=/data/.streamlit
 
-RUN mkdir -p /data/.streamlit /data/cache && \
-    chmod -R 777 /data
+# Create and set permissions on data directories
+RUN mkdir -p /data/cache && \
+    chmod -R 777 /data
+
+# Launch streamlit app
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]