Support for super-resolution node with opencv-cuda (#65)

* added built cv2 from github

* add opencv-cuda install to dockerfile

---------

Co-authored-by: Elite <[email protected]>

Author: JJassonn69

.gitignore

@@ -22,3 +22,6 @@ launch.json
 nodes/web/static/*
 .cursor/
 !nodes/web/static/.gitkeep
+
+# opencv files
+opencv

docker/Dockerfile.base

@@ -58,6 +58,9 @@ RUN conda run -n comfystream --no-capture-output --cwd /workspace/comfystream py
 # Install additional dependencies
 RUN conda run -n comfystream --no-capture-output pip install --upgrade tensorrt-cu12-bindings tensorrt-cu12-libs --root-user-action=ignore
 
+# Setup opencv with CUDA support
+RUN conda run -n comfystream --no-capture-output bash /workspace/comfystream/docker/entrypoint.sh --opencv-cuda
+
 # Configure no environment activation by default
 RUN conda config --set auto_activate_base false && \
     conda init bash

docker/entrypoint.sh

@@ -23,6 +23,7 @@ show_help() {
   echo "Options:"
   echo "  --download-models       Download default models"
   echo "  --build-engines         Build TensorRT engines for default models"
+  echo "  --opencv-cuda           Setup OpenCV with CUDA support"
   echo "  --server                Start the Comfystream server, UI and ComfyUI"
   echo "  --help                  Show this help message"
   echo ""
@@ -70,6 +71,55 @@ if [ "$1" = "--build-engines" ]; then
   shift
 fi
 
+if [ "$1" = "--opencv-cuda" ]; then
+  cd /workspace/comfystream
+  conda activate comfystream
+  
+  # Check if OpenCV CUDA build already exists
+  if [ ! -f "/workspace/comfystream/opencv-cuda-release.tar.gz" ]; then
+    # Download and extract OpenCV CUDA build
+    DOWNLOAD_NAME="opencv-cuda-release.tar.gz"
+    wget -q -O "$DOWNLOAD_NAME" https://github.com/JJassonn69/ComfyUI-Stream-Pack/releases/download/v1.0/opencv-cuda-release.tar.gz
+    tar -xzf "$DOWNLOAD_NAME" -C /workspace/comfystream/
+    rm "$DOWNLOAD_NAME"
+  else
+    echo "OpenCV CUDA build already exists, skipping download."
+  fi
+
+  # Install required libraries
+  apt-get update && apt-get install -y \
+    libgflags-dev \
+    libgoogle-glog-dev \
+    libjpeg-dev \
+    libavcodec-dev \
+    libavformat-dev \
+    libavutil-dev \
+    libswscale-dev
+
+  # Remove existing cv2 package
+  SITE_PACKAGES_DIR="/workspace/miniconda3/envs/comfystream/lib/python3.11/site-packages"
+  rm -rf "${SITE_PACKAGES_DIR}/cv2"*
+
+  # Copy new cv2 package
+  cp -r /workspace/comfystream/cv2 "${SITE_PACKAGES_DIR}/"
+
+  # Handle library dependencies
+  CONDA_ENV_LIB="/workspace/miniconda3/envs/comfystream/lib"
+  
+  # Remove existing libstdc++ and copy system one
+  rm -f "${CONDA_ENV_LIB}/libstdc++.so"*
+  cp /usr/lib/x86_64-linux-gnu/libstdc++.so* "${CONDA_ENV_LIB}/"
+
+  # Copy OpenCV libraries
+  cp /workspace/comfystream/opencv/build/lib/libopencv_* /usr/lib/x86_64-linux-gnu/
+
+  # remove the opencv-contrib and cv2 folders
+  rm -rf /workspace/comfystream/opencv_contrib
+  rm -rf /workspace/comfystream/cv2
+
+  echo "OpenCV CUDA installation completed"
+  shift
+fi
 
 if [ "$1" = "--server" ]; then
   /usr/bin/supervisord -c /etc/supervisor/supervisord.conf