Update Dockerfile

Dockerfile

@@ -2,55 +2,45 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Set environment variables
-ENV PYTHONUNBUFFERED=1
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV STREAMLIT_SERVER_PORT=7860
-ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
+# Set HOME to a writable location
+ENV HOME=/app
+ENV MPLCONFIGDIR=/app/.config/matplotlib
+ENV TORCH_HOME=/app/.torch
+ENV STREAMLIT_HOME=/app/.streamlit
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
+    curl \
+    software-properties-common \
     git \
     libgl1-mesa-glx \
     libglib2.0-0 \
-    wget \
+    libffi-dev \
+    libssl-dev \
+    libbz2-dev \
+    liblzma-dev \
+    libreadline-dev \
+    libsqlite3-dev \
+    zlib1g-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Create writable directories in /tmp
-RUN mkdir -p /tmp/.torch /tmp/.streamlit /tmp/.matplotlib \
-    && chmod 777 /tmp/.torch /tmp/.streamlit /tmp/.matplotlib
+# Create required writable folders
+RUN mkdir -p /app/.torch /app/.streamlit /app/.config/matplotlib
 
-# Install PyTorch first (specify CPU version)
-RUN pip install --no-cache-dir torch==1.13.1+cpu torchvision==0.14.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
-
-# Install numpy specifically to avoid version conflicts
-RUN pip install --no-cache-dir "numpy>=1.21.0,<1.27.0"
-
-# Install detection dependencies - separate critical ones 
-RUN pip install --no-cache-dir cython wheel fvcore iopath
-RUN pip install --no-cache-dir pycocotools matplotlib pyyaml tqdm
-
-# Install Streamlit and OpenCV
-RUN pip install --no-cache-dir streamlit>=1.20.0,<1.46.0 opencv-python-headless>=4.5.0,<4.12.0 pillow>=9.0.0,<12.0.0
-
-# Install detectron2 directly, without going through requirements.txt
-RUN pip install --no-cache-dir 'git+https://github.com/facebookresearch/detectron2.git@v0.6'
-
-# Create app structure
-RUN mkdir -p /app/src
-
-# Copy app files
+# Copy project files
+COPY requirements.txt ./
 COPY src/ ./src/
 
-# Configure Streamlit
-RUN mkdir -p /tmp/.streamlit && \
-    echo '[server]\nheadless = true\nenableCORS = false\nenableXsrfProtection = false' > /tmp/.streamlit/config.toml && \
-    echo '[general]\nemail = ""' > /tmp/.streamlit/credentials.toml && \
-    chmod -R 777 /tmp/.streamlit
+# Install Python dependencies
+RUN pip install --upgrade pip && \
+    pip install "numpy<2.0" && \
+    pip install -r requirements.txt && \
+    pip install torchvision==0.15.2 && \
+    pip install 'git+https://github.com/facebookresearch/detectron2.git'
+
+EXPOSE 8501
 
-# Expose the port Streamlit will run on
-EXPOSE 7860
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-# Run the app
-CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]