Upload MeitabuPackreatorComfyUi.sh

MeitabuPackreatorComfyUi.sh

@@ -49,6 +49,7 @@ readonly MODEL_DIRS=(
     "loras"
     "upscale_models"
     "ultralytics/bbox"
+    "Nudenet" # ### NOVO ### Adicionado para garantir que a pasta exista
 )
 
 # Workflow configuration
@@ -286,6 +287,63 @@ install_story_diffusion_node() {
     fi
 }
 
+# ============================================================================
+# ### NOVO ### SPECIAL NODE: NUDENET
+# ============================================================================
+
+install_nudenet_node() {
+    log_info "Installing special node: NudeNet..."
+    local node_repo="https://github.com/phuvinh010701/ComfyUI-NudeNet.git"
+    local node_name="ComfyUI-NudeNet"
+    local custom_nodes_dir="${COMFY_INSTALL_DIR}/custom_nodes"
+    local target_dir="${custom_nodes_dir}/${node_name}"
+    local requirements_file="${target_dir}/requirements.txt"
+
+    # Etapa 1: Clonar o repositório se não existir
+    mkdir -p "$custom_nodes_dir"
+    if [ -d "$target_dir" ]; then
+        log_info "NudeNet node directory already exists. Skipping clone."
+    else
+        log_info "Cloning NudeNet repository into $target_dir"
+        if ! git clone "$node_repo" "$target_dir"; then
+            log_error "Failed to clone NudeNet repository."
+            return 1 # Abortar a instalação deste node específico
+        fi
+    fi
+
+    # Etapa 2: Instalar dependências do requirements.txt
+    if [ -f "$requirements_file" ]; then
+        log_info "Installing NudeNet dependencies from $requirements_file..."
+        if ! python3 -m pip install -r "$requirements_file"; then
+            log_warning "Failed to install NudeNet dependencies. This might cause issues."
+        else
+            log_info "NudeNet dependencies installed successfully."
+        fi
+    else
+        log_info "No requirements.txt found for NudeNet. Skipping dependency installation."
+    fi
+
+    # Etapa 3: Baixar o modelo .onnx para a pasta correta
+    log_info "Setting up NudeNet model..."
+    local nudenet_model_dir="${MODELS_DIR}/Nudenet"
+    local nudenet_model_url="https://d2xl8ijk56kv4u.cloudfront.net/models/nudenet.onnx"
+    local nudenet_model_filename="nudenet.onnx"
+    local nudenet_model_path="${nudenet_model_dir}/${nudenet_model_filename}"
+
+    mkdir -p "$nudenet_model_dir" # Garante que o diretório de destino exista
+    
+    if [ -f "$nudenet_model_path" ]; then
+        log_info "NudeNet model already exists at $nudenet_model_path. Skipping download."
+    else
+        log_info "Downloading NudeNet model to $nudenet_model_path..."
+        if ! download_file "$nudenet_model_url" "$nudenet_model_dir" "$nudenet_model_filename"; then
+            log_error "Failed to download the NudeNet model. The node may not work correctly."
+        else
+            log_info "NudeNet model downloaded successfully."
+        fi
+    fi
+}
+
 # ============================================================================
 # DOWNLOAD FUNCTIONS
 # ============================================================================
@@ -363,6 +421,7 @@ main() {
     # Install special nodes and their dependencies
     install_wd14_tagger_node
     install_story_diffusion_node
+    install_nudenet_node # ### NOVO ### Chamada para a nova função
     
     download_all_files
     download_workflow