Update Dockerfile

Dockerfile

@@ -1,58 +1,74 @@
-# Use CUDA-enabled base image to match HF Spaces CUDA 12.3
-FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
+# Use official Python runtime with CUDA support
+FROM nvidia/cuda:12.1-devel-ubuntu22.04
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    GRADIO_SERVER_NAME="0.0.0.0" \
-    GRADIO_SERVER_PORT="7860" \
-    DISPLAY=:99
-
-# Set working directory
-WORKDIR /app
-
-# Create a non-root user for security
-RUN useradd -m myuser && \
-    chown myuser:myuser /app
+    DEBIAN_FRONTEND=noninteractive \
+    TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX" \
+    FORCE_CUDA="1"
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
+    python3.10 \
+    python3.10-dev \
+    python3-pip \
+    python3.10-venv \
     git \
-    ffmpeg \
-    xvfb \
-    libglib2.0-0 \
-    libgomp1 \
+    wget \
     curl \
-    && rm -rf /var/lib/apt/lists/* \
-    && apt-get clean
+    build-essential \
+    cmake \
+    pkg-config \
+    libopencv-dev \
+    python3-opencv \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    libfontconfig1 \
+    libxrender1 \
+    libgl1-mesa-glx \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first to leverage Docker cache
-COPY requirements.txt .
+# Create and set working directory
+WORKDIR /app
 
-# Install Python dependencies as non-root user
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
+# Create non-root user for security
+RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
+USER appuser
 
-# Copy application code
-COPY --chown=myuser:myuser . .
+# Install Python dependencies
+COPY --chown=appuser:appuser requirements.txt .
 
-# Create necessary directories with proper permissions
-RUN mkdir -p /tmp/gradio /tmp/model_cache && \
-    chmod 777 /tmp/gradio /tmp/model_cache && \
-    chown myuser:myuser /tmp/gradio /tmp/model_cache
+# 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
 
-# Set proper permissions for the app directory
-RUN chmod -R 755 /app
+# Install other requirements
+RUN pip install --no-cache-dir --user -r requirements.txt
 
-# Expose port
-EXPOSE 7860
+# Add user bin to PATH
+ENV PATH="/home/appuser/.local/bin:$PATH"
+
+# Copy application code
+COPY --chown=appuser:appuser . .
+
+# Create necessary directories
+RUN mkdir -p /tmp/model_cache /tmp/processing && \
+    chmod 755 /tmp/model_cache /tmp/processing
 
-# Health check to ensure the app is running
-HEALTHCHECK --interval=30s --timeout=3s \
+# Set environment variables for runtime
+ENV GRADIO_SERVER_NAME="0.0.0.0" \
+    GRADIO_SERVER_PORT="7860" \
+    MODEL_CACHE_DIR="/tmp/model_cache" \
+    TORCH_HOME="/tmp/model_cache"
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:7860/ || exit 1
 
-# Switch to non-root user for security
-USER myuser
+# Expose port
+EXPOSE 7860
 
-# Run the application via start.sh to ensure xvfb
-CMD ["bash", "start.sh"]
+# Run the application
+CMD ["python3", "app.py"]