Update Dockerfile

Dockerfile

@@ -1,28 +1,35 @@
-# Enable BuildKit when building: DOCKER_BUILDKIT=1
+# syntax=docker/dockerfile:1
 FROM python:3.10-slim
 
+# Prevent Python from buffering stdout/stderr (better logs)
 ENV PYTHONUNBUFFERED=1
 
 WORKDIR /home/user/app
 
-# System deps
+# System dependencies (minimal), and Git LFS
 RUN apt-get update && \
     apt-get install -y --no-install-recommends git git-lfs ffmpeg libsm6 libxext6 libgl1-mesa-glx && \
     rm -rf /var/lib/apt/lists/* && \
     git lfs install
 
-# Upgrade pip
+# Upgrade pip/setuptools
 RUN pip install --no-cache-dir pip setuptools
 
-# Copy requirements and install
+# Copy requirements early to leverage Docker cache
 COPY requirements.txt .
+
+# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy source
+# Copy application source
 COPY . .
 
-# Expose Flask port
-EXPOSE 8080
+# Expose both common ports (Flask uses 8080, Streamlit uses 8501)
+EXPOSE 8080 8501
 
-# Run Flask app (no reloader conflict)
-CMD ["python", "app.py"]
+# Entrypoint: choose Streamlit if USE_STREAMLIT=1, otherwise run Flask app
+CMD if [ "$USE_STREAMLIT" = "1" ]; then \
+      streamlit run app.py --server.port=8501 --server.enableCORS=false; \
+    else \
+      python app.py; \
+    fi