Update Dockerfile

Dockerfile

@@ -1,10 +1,17 @@
 FROM python:3.10
 
+# Set working directory
 WORKDIR /app
 
-ENV XDG_CONFIG_HOME=/tmp/.config
-ENV XDG_CACHE_HOME=/tmp/.cache
+# === ENV VARS ===
+ENV XDG_CONFIG_HOME=/tmp/.config \
+    XDG_CACHE_HOME=/tmp/.cache \
+    HF_HOME=/tmp/huggingface \
+    HF_HUB_CACHE=/tmp/huggingface/hub \
+    STREAMLIT_HOME=/tmp/.streamlit \
+    PYTHONUNBUFFERED=1
 
+# === Install system packages ===
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -17,14 +24,22 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# === Create writable directories ===
+RUN mkdir -p $XDG_CONFIG_HOME/streamlit $XDG_CACHE_HOME $HF_HUB_CACHE $STREAMLIT_HOME
+
+# === Add basic Streamlit config ===
+RUN echo "[server]\nheadless = true\nenableCORS = false\ndefaultPort = 8501" > $XDG_CONFIG_HOME/streamlit/config.toml
+
+# === Copy app code ===
 COPY requirements.txt ./
 COPY src/ ./src/
 
-RUN pip3 install -r requirements.txt
-
+# === Install Python dependencies ===
+RUN pip3 install --no-cache-dir -r requirements.txt
 
+# === Expose port & healthcheck ===
 EXPOSE 8501
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-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"]
+# === Run Streamlit app ===
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]