Update Dockerfile

Dockerfile

@@ -20,15 +20,6 @@ RUN mkdir -p /app/.cache/huggingface/transformers \
     && mkdir -p /app/.streamlit \
     && chmod -R 777 /app
 
-# Set Streamlit config to disable CORS and XSRF protection
-RUN echo "\
-[server]\n\
-headless = true\n\
-port = 8501\n\
-enableCORS = false\n\
-enableXsrfProtection = false\n\
-" > /app/.streamlit/config.toml
-
 # Copy files
 COPY requirements.txt ./
 COPY src/ ./src/
@@ -36,9 +27,13 @@ COPY src/ ./src/
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Expose Streamlit's default port
 EXPOSE 8501
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Healthcheck for container status
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
+# Start the Streamlit app
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]