Update Dockerfile

Dockerfile

@@ -24,30 +24,52 @@ RUN apt-get update && apt-get install -y \
     libreadline-dev \
     libsqlite3-dev \
     zlib1g-dev \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
 # Create temporary directories with proper permissions
 RUN mkdir -p /tmp/.torch /tmp/.streamlit /tmp/.matplotlib \
     && chmod 777 /tmp/.torch /tmp/.streamlit /tmp/.matplotlib
 
-# Copy project files
-COPY requirements.txt ./
-COPY src/ ./src/
+# Upgrade pip and install base packages
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel
 
+# 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
 
-# Copy project files
+# Install numpy with version constraint
+RUN pip install --no-cache-dir "numpy<2.0"
+
+# Copy only requirements first for better caching
 COPY requirements.txt ./
+
+# 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)
+RUN pip install --no-cache-dir 'git+https://github.com/facebookresearch/detectron2.git@v0.6'
+
+# Create the src directory and copy application files
+RUN mkdir -p /app/src
 COPY src/ ./src/
 
-# 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'
+# 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
 
+# Add a healthcheck
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Run as non-root user for better security
+USER 1000
+
+# Command to run the app
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]