Update Dockerfile

Dockerfile

@@ -2,74 +2,55 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Set up environment variables for writable directories
-ENV HOME=/app
-ENV MPLCONFIGDIR=/tmp/.matplotlib
-ENV TORCH_HOME=/tmp/.torch
-ENV STREAMLIT_HOME=/tmp/.streamlit
-ENV PYTHONPATH=$PYTHONPATH:/app
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV STREAMLIT_SERVER_PORT=7860
+ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
-    curl \
-    software-properties-common \
     git \
     libgl1-mesa-glx \
     libglib2.0-0 \
-    libffi-dev \
-    libssl-dev \
-    libbz2-dev \
-    liblzma-dev \
-    libreadline-dev \
-    libsqlite3-dev \
-    zlib1g-dev \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Create temporary directories with proper permissions
+# Create writable directories in /tmp
 RUN mkdir -p /tmp/.torch /tmp/.streamlit /tmp/.matplotlib \
     && chmod 777 /tmp/.torch /tmp/.streamlit /tmp/.matplotlib
 
-# Upgrade pip and install base packages
-RUN pip install --no-cache-dir --upgrade pip setuptools wheel
+# 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 PyTorch and torchvision first (often the most difficult dependencies)
-RUN pip install --no-cache-dir torch==1.13.1 torchvision==0.14.1 -f https://download.pytorch.org/whl/cpu/torch_stable.html
+# Install numpy specifically to avoid version conflicts
+RUN pip install --no-cache-dir "numpy>=1.21.0,<1.27.0"
 
-# Install numpy with version constraint
-RUN pip install --no-cache-dir "numpy<2.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
 
-# Copy only requirements first for better caching
-COPY requirements.txt ./
+# 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 Python dependencies in smaller batches for better error visibility
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Install detectron2 from GitHub (with specific commit for stability)
+# Install detectron2 directly, without going through requirements.txt
 RUN pip install --no-cache-dir 'git+https://github.com/facebookresearch/detectron2.git@v0.6'
 
-# Create the src directory and copy application files
+# Create app structure
 RUN mkdir -p /app/src
-COPY src/ ./src/
-
-# Set up Streamlit configuration
-RUN mkdir -p /tmp/.streamlit && echo '\
-[general]\n\
-email = ""\n\
-' > /tmp/.streamlit/credentials.toml
 
-# Set proper ownership for application files
-RUN chown -R 1000:1000 /app
-
-# Expose Streamlit port
-EXPOSE 8501
+# Copy app files
+COPY src/ ./src/
 
-# Add a healthcheck
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
+# 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
 
-# Run as non-root user for better security
-USER 1000
+# Expose the port Streamlit will run on
+EXPOSE 7860
 
-# Command to run the app
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Run the app
+CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]