Update setup_comfy_wanDANCE.sh

setup_comfy_wanDANCE.sh

@@ -56,11 +56,14 @@ readonly DOWNLOAD_FILES=(
     "https://huggingface.co/ABDALLALSWAITI/Upscalers/resolve/main/anime/2x-AnimeSharpV2_MoSR_Soft.pth|upscale_models|2x-AnimeSharpV2_MoSR_Soft.pth"
     # Civitai model
     #"https://civitai.com/api/download/models/2122278?type=Model&format=SafeTensor&size=pruned&fp=fp16|checkpoints|raehoshiIllustXL_v60.safetensors"
+    # MEGA downloads (modelos adicionais)
+    "mega://5QRxDDII#MVIEGduGgTDS0DkanTiPQso0O4Im-Ph5QR0JGPbox0M|loras|mega_model_1.safetensors"
+    "mega://8ZQgjZQR#EZav6GgFTX7amc41PGHAM3FOKaTOlfjrvLWs8XpdPNY|loras|mega_model_2.safetensors"
+    "mega://lQRB1ZhT#IBoYK1DJCILHl4ILEgIYUZBVmC2pii-PqtLH38bwJqY|loras|mega_model_3.safetensors"
 )
 
 # Custom nodes para instalar
 readonly CUSTOM_NODES=(
-    "https://github.com/kijai/ComfyUI-Florence2"
     "https://github.com/kijai/ComfyUI-WanVideoWrapper"
     "https://github.com/Fannovel16/ComfyUI-Frame-Interpolation"
     "https://github.com/kijai/ComfyUI-GIMM-VFI"
@@ -68,9 +71,9 @@ readonly CUSTOM_NODES=(
     "https://github.com/kijai/ComfyUI-KJNodes"
     "https://github.com/Fannovel16/comfyui_controlnet_aux"
     "https://github.com/Artificial-Sweetener/comfyui-WhiteRabbit"
-    "https://github.com/shiimizu/ComfyUI_smZNodes"
+    "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes"
+    "https://github.com/yuvraj108c/ComfyUI-Upscaler-Tensorrt"
     "https://github.com/CoreyCorza/ComfyUI-CRZnodes"
-    "https://github.com/yuvraj108c/ComfyUI-Dwpose-Tensorrt"
     "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes"
     "https://github.com/grmchn/ComfyUI-ProportionChanger"
 )
@@ -160,6 +163,89 @@ download_hf() {
     return 1
 }
 
+# Download do MEGA com retry e verificação
+download_mega() {
+    local mega_url="$1"
+    local target_dir="$2"
+    local filename="$3"
+    local max_retries=3
+    local retry_count=0
+
+    # Extrair ID e key do URL MEGA
+    local file_id=$(echo "$mega_url" | sed 's/mega:\/\///' | cut -d'#' -f1)
+    local file_key=$(echo "$mega_url" | cut -d'#' -f2)
+
+    # Construir URL completo do MEGA
+    local full_mega_url="https://mega.nz/file/${file_id}#${file_key}"
+
+    local target_file="$target_dir/$filename"
+
+    # Se arquivo já existe e tem tamanho razoável, pular
+    if [ -f "$target_file" ] && [ $(stat -c%s "$target_file" 2>/dev/null || echo 0) -gt 1000000 ]; then
+        log_success "Arquivo MEGA já existe: $filename"
+        return 0
+    fi
+
+    log_info "Baixando do MEGA: $filename"
+    log_info "  URL: $full_mega_url"
+
+    # Instalar megatools se não existir
+    if ! command_exists megadl; then
+        log_info "Instalando megatools..."
+        apt-get update -y -qq 2>/dev/null
+        apt-get install -y -qq megatools 2>/dev/null
+
+        if ! command_exists megadl; then
+            log_error "Falha ao instalar megatools!"
+            return 1
+        fi
+    fi
+
+    # Tentar baixar com retry
+    while [ $retry_count -lt $max_retries ]; do
+        retry_count=$((retry_count + 1))
+        log_info "Tentativa $retry_count de $max_retries..."
+
+        # Criar diretório temporário para download
+        local temp_dir=$(mktemp -d)
+        cd "$temp_dir"
+
+        # Tentar download com megadl
+        if megadl "$full_mega_url" 2>/dev/null; then
+            # Procurar arquivo baixado
+            local downloaded_file=$(find . -type f \( -name "*.safetensors" -o -name "*.pth" -o -name "*.pt" -o -name "*.bin" \) 2>/dev/null | head -1)
+
+            if [ -n "$downloaded_file" ]; then
+                # Verificar tamanho do arquivo
+                local file_size=$(stat -c%s "$downloaded_file" 2>/dev/null || echo 0)
+                if [ $file_size -gt 1000000 ]; then
+                    # Mover para destino
+                    mv "$downloaded_file" "$target_file"
+                    cd - >/dev/null
+                    rm -rf "$temp_dir"
+                    log_success "Download MEGA concluído: $filename ($(numfmt --to=iec-i --suffix=B $file_size))"
+                    return 0
+                else
+                    log_warn "Arquivo baixado muito pequeno, tentando novamente..."
+                fi
+            fi
+        else
+            log_warn "Falha no download com megadl"
+        fi
+
+        cd - >/dev/null
+        rm -rf "$temp_dir"
+
+        if [ $retry_count -lt $max_retries ]; then
+            log_info "Aguardando 5 segundos antes de tentar novamente..."
+            sleep 5
+        fi
+    done
+
+    log_error "Falha após $max_retries tentativas: $filename"
+    return 1
+}
+
 # Download genérico com aria2c, wget ou curl
 download_file() {
     local url="$1"
@@ -248,12 +334,15 @@ process_downloads() {
         if [[ "$url" == hf://* ]]; then
             # HuggingFace download
             url="${url#hf://}"  # Remove prefixo hf://
-            
+
             # Separar repo e caminho do arquivo
             local repo=$(echo "$url" | cut -d'/' -f1-2)
             local file_path=$(echo "$url" | cut -d'/' -f3-)
-            
+
             download_hf "$repo" "$file_path" "$target_dir" "$filename" || ((failed_count++))
+        elif [[ "$url" == mega://* ]]; then
+            # MEGA download
+            download_mega "$url" "$target_dir" "$filename" || ((failed_count++))
         else
             # Download normal (HTTP/HTTPS)
             download_file "$url" "$target_dir" "$filename" || ((failed_count++))