4K_Upscaler

Sleeping

App Files Files Community

Malaji71 commited on Aug 18

Commit

f810e6d

verified ·

1 Parent(s): f5e9a57

Update Dockerfile

Browse files

Files changed (1) hide show

Dockerfile +40 -34

Dockerfile CHANGED Viewed

@@ -7,69 +7,70 @@ WORKDIR /app
 # Prevent interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
-# Install system dependencies for Real-ESRGAN
 RUN apt-get update && apt-get install -y \
     # Build essentials
     build-essential \
     cmake \
     pkg-config \
     # OpenCV dependencies
     libopencv-dev \
-    python3-opencv \
-    # Graphics libraries
     libgl1-mesa-dev \
     libglu1-mesa-dev \
-    # X11 libraries
     libx11-dev \
     libxext-dev \
     libxrender-dev \
-    # Image/video processing
     ffmpeg \
     libavcodec-dev \
     libavformat-dev \
     libswscale-dev \
-    # Additional libraries
-    libglib2.0-0 \
-    libsm6 \
-    libgomp1 \
-    # Download tools
-    wget \
-    curl \
-    git \
-    # Clean up
     && rm -rf /var/lib/apt/lists/* \
     && apt-get clean
-# Upgrade pip and install wheel
 RUN pip install --upgrade pip setuptools wheel
-# Install PyTorch first (CPU version for compatibility)
 RUN pip install --no-cache-dir torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cpu
-# Install core dependencies first
 RUN pip install --no-cache-dir \
     Flask==2.3.3 \
     Werkzeug==2.3.7 \
     Pillow==10.0.1 \
-    numpy==1.24.4 \
     opencv-python-headless==4.8.1.78 \
-    scipy==1.11.4 \
-    tqdm==4.66.1 \
-    pyyaml==6.0.1 \
-    requests==2.31.0
-# Install Real-ESRGAN dependencies step by step
 RUN pip install --no-cache-dir \
     addict==2.4.0 \
     future==0.18.3 \
     lmdb==1.4.1 \
     yapf==0.40.2 \
     filterpy==1.4.5
-# Install BasicSR first (Real-ESRGAN dependency)
 RUN pip install --no-cache-dir basicsr==1.4.2
-# Install remaining Real-ESRGAN dependencies
 RUN pip install --no-cache-dir \
     facexlib==0.3.0 \
     gfpgan==1.3.8
@@ -77,17 +78,21 @@ RUN pip install --no-cache-dir \
 # Finally install Real-ESRGAN
 RUN pip install --no-cache-dir realesrgan==0.3.0
-# Copy requirements for verification
-COPY requirements.txt .
 # Copy application files
-COPY . .
-# Create necessary directories
-RUN mkdir -p /data/uploads /data/outputs /data/models
-# Test Real-ESRGAN installation - FIXED SYNTAX
-RUN python -c "from realesrgan import RealESRGANer; from basicsr.archs.rrdbnet_arch import RRDBNet; print('✅ Real-ESRGAN import successful')"
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
@@ -95,13 +100,14 @@ ENV FLASK_APP=app.py
 ENV FLASK_ENV=production
 ENV TORCH_HOME=/data/models
 ENV OPENCV_LOG_LEVEL=ERROR
 # Expose port
 EXPOSE 7860
 # Health check
 HEALTHCHECK --interval=30s --timeout=30s --start-period=15s --retries=3 \
-    CMD curl -f http://localhost:7860/ || exit 1
 # Run the application
 CMD ["python", "app.py"]

 # Prevent interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     # Build essentials
     build-essential \
     cmake \
     pkg-config \
+    git \
+    wget \
+    curl \
     # OpenCV dependencies
     libopencv-dev \
+    # Graphics and video libraries
     libgl1-mesa-dev \
     libglu1-mesa-dev \
     libx11-dev \
     libxext-dev \
     libxrender-dev \
+    libglib2.0-0 \
+    libsm6 \
+    libgomp1 \
+    # Video processing
     ffmpeg \
     libavcodec-dev \
     libavformat-dev \
     libswscale-dev \
+    # Additional dependencies
+    libjpeg-dev \
+    libpng-dev \
+    libtiff-dev \
     && rm -rf /var/lib/apt/lists/* \
     && apt-get clean
+# Upgrade pip and install essential tools
 RUN pip install --upgrade pip setuptools wheel
+# Install numpy first (critical for all other packages)
+RUN pip install --no-cache-dir numpy==1.24.4
+# Install PyTorch CPU version
 RUN pip install --no-cache-dir torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cpu
+# Install core dependencies
 RUN pip install --no-cache-dir \
     Flask==2.3.3 \
     Werkzeug==2.3.7 \
     Pillow==10.0.1 \
     opencv-python-headless==4.8.1.78 \
+    scipy==1.11.4
+# Install Real-ESRGAN dependencies in correct order
 RUN pip install --no-cache-dir \
+    tqdm==4.66.1 \
+    pyyaml==6.0.1 \
+    requests==2.31.0 \
     addict==2.4.0 \
     future==0.18.3 \
     lmdb==1.4.1 \
     yapf==0.40.2 \
     filterpy==1.4.5
+# Install BasicSR (Real-ESRGAN's core dependency)
 RUN pip install --no-cache-dir basicsr==1.4.2
+# Install face-related dependencies
 RUN pip install --no-cache-dir \
     facexlib==0.3.0 \
     gfpgan==1.3.8
 # Finally install Real-ESRGAN
 RUN pip install --no-cache-dir realesrgan==0.3.0
 # Copy application files
+COPY requirements.txt .
+COPY *.py .
+COPY templates/ templates/ 2>/dev/null || true
+COPY static/ static/ 2>/dev/null || true
+# Create necessary directories with proper permissions
+RUN mkdir -p /data/uploads /data/outputs /data/models && \
+    chmod 755 /data /data/uploads /data/outputs /data/models
+# Test installations
+RUN python -c "import numpy; print(f'NumPy version: {numpy.__version__}')" && \
+    python -c "import torch; print(f'PyTorch version: {torch.__version__}')" && \
+    python -c "import cv2; print(f'OpenCV version: {cv2.__version__}')" && \
+    python -c "from realesrgan import RealESRGANer; from basicsr.archs.rrdbnet_arch import RRDBNet; print('✅ Real-ESRGAN import successful')"
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV FLASK_ENV=production
 ENV TORCH_HOME=/data/models
 ENV OPENCV_LOG_LEVEL=ERROR
+ENV NUMBA_CACHE_DIR=/tmp/numba_cache
 # Expose port
 EXPOSE 7860
 # Health check
 HEALTHCHECK --interval=30s --timeout=30s --start-period=15s --retries=3 \
+    CMD curl -f http://localhost:7860/api/system || exit 1
 # Run the application
 CMD ["python", "app.py"]