Update Dockerfile

Dockerfile

@@ -1,31 +1,46 @@
 FROM python:3.9-slim
 
+# Set working directory
 WORKDIR /app
 
-# RUN apt-get update && apt-get install -y \
-#     build-essential \
-#     curl \
-#     software-properties-common \
-#     git \
-#     && rm -rf /var/lib/apt/lists/*
+# Avoid interactive prompts
+ENV DEBIAN_FRONTEND=noninteractive
+ENV STREAMLIT_HOME=/app/.streamlit
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
-    curl \
     git \
+    curl \
     && rm -rf /var/lib/apt/lists/*
-# RUN apt-get update && apt-get install -y \
-#     build-essential \
-#     curl \
-#     && rm -rf /var/lib/apt/lists/*
-    
+
+# Copy requirements and install Python dependencies
 COPY requirements.txt .
+RUN pip install --upgrade pip && pip install -r requirements.txt
+
+# Create directories for Streamlit and HuggingFace to avoid permission errors
+RUN mkdir -p $STREAMLIT_HOME && mkdir -p $TRANSFORMERS_CACHE
+RUN echo "\
+[server]\n\
+headless = true\n\
+enableCORS = false\n\
+port = 8501\n\
+" > $STREAMLIT_HOME/config.toml
 
-RUN pip3 install -r requirements.txt
+# Download Hugging Face model (pre-cache)
+RUN python -c "from transformers import AutoModel, AutoTokenizer; \
+               AutoModel.from_pretrained('indobenchmark/indobert-base-p1'); \
+               AutoTokenizer.from_pretrained('indobenchmark/indobert-base-p1')"
 
+# Copy project files
 COPY . .
 
+# Expose Streamlit port
 EXPOSE 8501
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Healthcheck (optional)
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-ENTRYPOINT ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Run Streamlit
+ENTRYPOINT ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]