Update Dockerfile

Dockerfile

@@ -1,19 +1,13 @@
-# Use official Python runtime with CUDA support
-FROM nvidia/cuda:12.1-devel-ubuntu22.04
+# Use Python 3.9 base image for MatAnyone compatibility
+FROM python:3.9-slim
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    DEBIAN_FRONTEND=noninteractive \
-    TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX" \
-    FORCE_CUDA="1"
+    DEBIAN_FRONTEND=noninteractive
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-    python3.10 \
-    python3.10-dev \
-    python3-pip \
-    python3.10-venv \
     git \
     wget \
     curl \
@@ -28,23 +22,47 @@ RUN apt-get update && apt-get install -y \
     libfontconfig1 \
     libxrender1 \
     libgl1-mesa-glx \
+    libglib2.0-0 \
+    libavformat-dev \
+    libavcodec-dev \
+    libavdevice-dev \
+    libavutil-dev \
+    libswscale-dev \
+    libswresample-dev \
+    libavfilter-dev \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# Create and set working directory
+# Create working directory
 WORKDIR /app
 
-# Create non-root user for security
+# Create non-root user
 RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
 USER appuser
 
-# Install Python dependencies
+# Copy requirements first for better caching
 COPY --chown=appuser:appuser requirements.txt .
 
-# Install PyTorch with CUDA support first (speeds up other installs)
-RUN pip install --no-cache-dir --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
+# Install Python dependencies in stages for better error handling
+RUN pip install --no-cache-dir --user pip setuptools wheel
 
-# Install other requirements
+# Install PyTorch first (most stable approach)
+RUN pip install --no-cache-dir --user \
+    torch==2.0.1 \
+    torchvision==0.15.2 \
+    torchaudio==2.0.1 \
+    --index-url https://download.pytorch.org/whl/cpu
+
+# Install core dependencies
+RUN pip install --no-cache-dir --user \
+    numpy==1.24.3 \
+    opencv-python-headless==4.8.0.76 \
+    Pillow==10.0.1
+
+# Install AV package with build dependencies
+RUN pip install --no-cache-dir --user av==10.0.0
+
+# Install remaining requirements
 RUN pip install --no-cache-dir --user -r requirements.txt
 
 # Add user bin to PATH
@@ -54,8 +72,8 @@ ENV PATH="/home/appuser/.local/bin:$PATH"
 COPY --chown=appuser:appuser . .
 
 # Create necessary directories
-RUN mkdir -p /tmp/model_cache /tmp/processing && \
-    chmod 755 /tmp/model_cache /tmp/processing
+RUN mkdir -p /tmp/model_cache /tmp/processing /tmp/jobs && \
+    chmod 755 /tmp/model_cache /tmp/processing /tmp/jobs
 
 # Set environment variables for runtime
 ENV GRADIO_SERVER_NAME="0.0.0.0" \
@@ -71,4 +89,4 @@ HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
 EXPOSE 7860
 
 # Run the application
-CMD ["python3", "app.py"]
+CMD ["python", "app.py"]