Upload comfyui_setup_script.sh

comfyui_setup_script.sh

@@ -1,402 +1,473 @@
 #!/bin/bash
 
-# =============================================================================
-# ComfyUI Professional Setup Script
-# =============================================================================
-# Autor: Setup automatizado para RunPod/VastAI
-# Versão: 1.0
-# Descrição: Instala e configura ComfyUI com todas as dependências necessárias
-# =============================================================================
-
-set -euo pipefail  # Exit on error, undefined vars, pipe failures
-
-# Cores para output
-readonly RED='\033[0;31m'
-readonly GREEN='\033[0;32m'
-readonly YELLOW='\033[1;33m'
-readonly BLUE='\033[0;34m'
-readonly MAGENTA='\033[0;35m'
-readonly CYAN='\033[0;36m'
-readonly WHITE='\033[1;37m'
-readonly NC='\033[0m' # No Color
-
-# Configurações
-readonly SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
-readonly LOG_FILE="${SCRIPT_DIR}/setup.log"
-readonly COMFY_DIR="/root/comfy"
-readonly TOKEN="0b698de03658fb8df2eeef90bad3ab9a"
-readonly LISTEN_PORT="8818"
-
-# URLs dos arquivos
-declare -A DOWNLOADS=(
-    ["checkpoint"]="https://civitai.com/api/download/models/1761560?type=Model&format=SafeTensor&size=pruned&fp=fp16&token=${TOKEN}"
-    ["lora1"]="https://civitai.com/api/download/models/1426079?type=Model&format=SafeTensor&token=${TOKEN}"
-    ["lora2"]="https://civitai.com/api/download/models/1193082?type=Model&format=SafeTensor&token=${TOKEN}"
-    ["snapshot"]="https://huggingface.co/adbrasi/testedownload/resolve/main/snapshotBRABEX.json"
+# ComfyUI Automated Setup Script
+# Supports LightningAI, RunPod, VastAI, and generic Linux environments
+# Author: Generated for automated ComfyUI deployment
+
+set -e
+set -o pipefail
+
+# ============================================================================
+# GLOBAL CONFIGURATIONS
+# ============================================================================
+
+# Civitai API Token
+readonly CIVITAI_TOKEN="0b698de03658fb8df2eeef90bad3ab9a"
+
+# Custom Nodes to Install
+readonly CUSTOM_NODES=(
+    "https://github.com/Cezarsaint/custom_prompt_manager"
+    "https://github.com/adbrasi/pakreatorio"
 )
 
-declare -A DOWNLOAD_PATHS=(
-    ["checkpoint"]="${COMFY_DIR}/ComfyUI/models/checkpoints"
-    ["lora1"]="${COMFY_DIR}/ComfyUI/models/loras"
-    ["lora2"]="${COMFY_DIR}/ComfyUI/models/loras"
-    ["snapshot"]="${COMFY_DIR}/"
+# Files to Download - Format: "URL|target_subdir|filename"
+# If filename is empty, let aria2c/wget determine from Content-Disposition
+readonly DOWNLOAD_FILES=(
+    "https://civitai.com/api/download/models/1761560?type=Model&format=SafeTensor&size=pruned&fp=fp16|checkpoints|"
+    "https://civitai.com/api/download/models/1426079?type=Model&format=SafeTensor|loras|"
+    "https://civitai.com/api/download/models/1193082?type=Model&format=SafeTensor|loras|"
+    "https://huggingface.co/Kim2091/AnimeSharpV3/resolve/main/2x-AnimeSharpV3.pth|upscale_models|2x-AnimeSharpV3.pth"
+    "https://huggingface.co/adbrasi/testedownload/resolve/main/99coins_anime_girl_face_m_seg.pt|ultralytics/bbox|99coins_anime_girl_face_m_seg.pt"
+    "https://huggingface.co/adbrasi/testedownload/resolve/main/snapshotBRABEX.json|INSTALL_DIR|snapshotBRABEX.json"
 )
 
-# =============================================================================
-# FUNÇÕES UTILITÁRIAS
-# =============================================================================
-
-log() {
-    local level="$1"
-    shift
-    local message="$*"
-    local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
-    
-    case "$level" in
-        "INFO")  echo -e "${CYAN}[INFO]${NC} ${message}" | tee -a "$LOG_FILE" ;;
-        "WARN")  echo -e "${YELLOW}[WARN]${NC} ${message}" | tee -a "$LOG_FILE" ;;
-        "ERROR") echo -e "${RED}[ERROR]${NC} ${message}" | tee -a "$LOG_FILE" ;;
-        "SUCCESS") echo -e "${GREEN}[SUCCESS]${NC} ${message}" | tee -a "$LOG_FILE" ;;
-        "STEP") echo -e "${MAGENTA}[STEP]${NC} ${WHITE}${message}${NC}" | tee -a "$LOG_FILE" ;;
-    esac
-    
-    echo "[$timestamp] [$level] $message" >> "$LOG_FILE"
+# Model directories to create
+readonly MODEL_DIRS=(
+    "checkpoints"
+    "loras"
+    "upscale_models"
+    "ultralytics/bbox"
+)
+
+# Global variables
+BASE_DIR=""
+COMFY_INSTALL_DIR=""
+MODELS_DIR=""
+SUDO_PREFIX=""
+ARIA2C_AVAILABLE=false
+ENV_NAME=""
+
+# ============================================================================
+# UTILITY FUNCTIONS
+# ============================================================================
+
+log_info() {
+    echo "INFO: $1" >&2
 }
 
-print_banner() {
-    echo -e "${CYAN}"
-    cat << 'EOF'
-    ╔══════════════════════════════════════════════════════════════╗
-    ║                   ComfyUI Setup Script                       ║
-    ║                Professional Installation                      ║
-    ╚══════════════════════════════════════════════════════════════╝
-EOF
-    echo -e "${NC}"
+log_warning() {
+    echo "WARNING: $1" >&2
 }
 
-check_dependencies() {
-    log "STEP" "Verificando dependências do sistema..."
-    
-    local deps=("curl" "wget" "python3" "pip")
-    local missing_deps=()
+log_error() {
+    echo "ERROR: $1" >&2
+}
+
+command_exists() {
+    command -v "$1" >/dev/null 2>&1
+}
+
+is_root() {
+    [ "$(id -u)" -eq 0 ]
+}
+
+# ============================================================================
+# ENVIRONMENT DETECTION
+# ============================================================================
+
+detect_environment() {
+    log_info "Detecting environment..."
+    
+    if [ -d "/teamspace/studios/this_studio" ]; then
+        BASE_DIR="/teamspace/studios/this_studio"
+        ENV_NAME="lightning"
+    elif [ -d "/workspace" ] && [ ! -d "/teamspace/studios/this_studio" ]; then
+        BASE_DIR="/workspace"
+        ENV_NAME="runpod"
+    elif is_root && [ ! -d "/teamspace/studios/this_studio" ] && [ ! -d "/workspace" ]; then
+        BASE_DIR="/root"
+        ENV_NAME="vastai"
+    else
+        BASE_DIR="$HOME"
+        ENV_NAME="generic"
+    fi
     
-    for dep in "${deps[@]}"; do
-        if ! command -v "$dep" &> /dev/null; then
-            missing_deps+=("$dep")
-        fi
-    done
+    COMFY_INSTALL_DIR="${BASE_DIR}/my_comfy_setup"
+    MODELS_DIR="${COMFY_INSTALL_DIR}/models"
     
-    if [[ ${#missing_deps[@]} -ne 0 ]]; then
-        log "ERROR" "Dependências faltando: ${missing_deps[*]}"
-        return 1
+    log_info "Environment detected: $ENV_NAME"
+    log_info "Base directory: $BASE_DIR"
+    log_info "ComfyUI install directory: $COMFY_INSTALL_DIR"
+    log_info "Models directory: $MODELS_DIR"
+}
+
+# ============================================================================
+# PERMISSION MANAGEMENT
+# ============================================================================
+
+setup_permissions() {
+    log_info "Setting up permissions..."
+    
+    if is_root; then
+        SUDO_PREFIX=""
+        log_info "Running as root, no sudo needed"
+    elif command_exists sudo; then
+        SUDO_PREFIX="sudo "
+        log_info "Non-root user with sudo available"
+    else
+        SUDO_PREFIX=""
+        log_warning "Non-root user without sudo, some system operations may fail"
     fi
-    
-    log "SUCCESS" "Todas as dependências básicas estão disponíveis"
 }
 
-install_system_packages() {
-    log "STEP" "Instalando pacotes do sistema..."
+# ============================================================================
+# SYSTEM DEPENDENCIES
+# ============================================================================
+
+install_system_dependencies() {
+    log_info "Installing system dependencies..."
     
-    if command -v apt-get &> /dev/null; then
-        apt-get update -qq
-        apt-get install -y -qq aria2 wget curl
-        log "SUCCESS" "Pacotes do sistema instalados com sucesso"
+    # Update package list
+    if command_exists apt-get; then
+        ${SUDO_PREFIX}apt-get update -qq || log_warning "Failed to update package list"
+        
+        # Install basic dependencies
+        ${SUDO_PREFIX}apt-get install -y -qq python3 python3-pip git || {
+            log_error "Failed to install basic dependencies (python3, pip, git)"
+            exit 1
+        }
+        
+        # Try to install aria2c
+        if ${SUDO_PREFIX}apt-get install -y -qq aria2 >/dev/null 2>&1; then
+            ARIA2C_AVAILABLE=true
+            log_info "aria2c installed successfully"
+        else
+            ARIA2C_AVAILABLE=false
+            log_info "aria2c installation failed, will use wget/curl as fallback"
+        fi
     else
-        log "WARN" "apt-get não encontrado, pulando instalação de pacotes do sistema"
+        log_warning "apt-get not available, skipping system package installation"
+        # Check if aria2c is already available
+        if command_exists aria2c; then
+            ARIA2C_AVAILABLE=true
+            log_info "aria2c already available"
+        fi
     fi
 }
 
+# ============================================================================
+# COMFY-CLI INSTALLATION
+# ============================================================================
+
 install_comfy_cli() {
-    log "STEP" "Instalando comfy-cli..."
+    log_info "Installing comfy-cli..."
     
-    pip install --quiet --upgrade pip
-    pip install --quiet comfy-cli
+    python3 -m pip install --upgrade comfy-cli || {
+        log_error "Failed to install comfy-cli"
+        exit 1
+    }
     
-    if ! command -v comfy &> /dev/null; then
-        log "ERROR" "Falha na instalação do comfy-cli"
-        return 1
+    # Check if comfy is in PATH
+    if ! command_exists comfy; then
+        if [ -d "$HOME/.local/bin" ]; then
+            export PATH="$HOME/.local/bin:$PATH"
+            log_info "Added $HOME/.local/bin to PATH"
+        fi
+        
+        if ! command_exists comfy; then
+            log_error "comfy command not found in PATH after installation"
+            exit 1
+        fi
     fi
     
-    log "SUCCESS" "comfy-cli instalado com sucesso"
+    log_info "comfy-cli version: $(comfy --version 2>/dev/null || echo 'unknown')"
 }
 
-install_comfyui() {
-    log "STEP" "Instalando ComfyUI com suporte NVIDIA..."
-    
-    comfy --skip-prompt install --nvidia
+# ============================================================================
+# COMFYUI CORE INSTALLATION
+# ============================================================================
+
+install_comfyui_core() {
+    log_info "Installing ComfyUI core..."
     
-    if [[ ! -d "$COMFY_DIR" ]]; then
-        log "ERROR" "Diretório do ComfyUI não foi criado"
-        return 1
-    fi
+    comfy --workspace "$COMFY_INSTALL_DIR" --skip-prompt install --nvidia || {
+        log_error "Failed to install ComfyUI core"
+        exit 1
+    }
     
-    log "SUCCESS" "ComfyUI instalado com sucesso"
+    log_info "ComfyUI core installed successfully"
 }
 
-install_custom_nodes() {
-    log "STEP" "Instalando nós customizados..."
-    
-    local nodes=(
-        "https://github.com/Cezarsaint/custom_prompt_manager"
-        "https://github.com/adbrasi/pakreatorio"
-    )
+# ============================================================================
+# DIRECTORY STRUCTURE
+# ============================================================================
+
+create_model_directories() {
+    log_info "Creating model directories..."
     
-    for node in "${nodes[@]}"; do
-        log "INFO" "Instalando nó: $(basename "$node")"
-        if comfy node install "$node"; then
-            log "SUCCESS" "Nó $(basename "$node") instalado com sucesso"
-        else
-            log "WARN" "Falha ao instalar nó: $(basename "$node")"
-        fi
+    # Create model directories
+    for dir in "${MODEL_DIRS[@]}"; do
+        mkdir -p "$MODELS_DIR/$dir"
+        log_info "Created directory: $MODELS_DIR/$dir"
     done
+    
+    # Ensure install directory exists for snapshot
+    mkdir -p "$COMFY_INSTALL_DIR"
 }
 
-create_directories() {
-    log "STEP" "Criando estrutura de diretórios..."
-    
-    local dirs=(
-        "${COMFY_DIR}/ComfyUI/models/checkpoints"
-        "${COMFY_DIR}/ComfyUI/models/loras"
-    )
+# ============================================================================
+# CUSTOM NODES INSTALLATION
+# ============================================================================
+
+install_custom_nodes() {
+    log_info "Installing custom nodes..."
     
-    for dir in "${dirs[@]}"; do
-        mkdir -p "$dir"
-        log "INFO" "Diretório criado: $dir"
+    for node_url in "${CUSTOM_NODES[@]}"; do
+        log_info "Installing custom node: $node_url"
+        comfy --workspace "$COMFY_INSTALL_DIR" node install "$node_url" || {
+            log_warning "Failed to install custom node: $node_url"
+        }
     done
 }
 
-download_files() {
-    log "STEP" "Iniciando downloads paralelos dos modelos..."
+# ============================================================================
+# DOWNLOAD FUNCTIONS
+# ============================================================================
+
+add_civitai_token() {
+    local url="$1"
+    if [[ "$url" == *"civitai.com/api/download"* ]] && [[ "$url" != *"token="* ]]; then
+        echo "${url}&token=${CIVITAI_TOKEN}"
+    else
+        echo "$url"
+    fi
+}
+
+download_file() {
+    local url="$1"
+    local target_dir="$2"
+    local filename="$3"
     
-    local pids=()
-    local download_success=true
+    # Add Civitai token if needed
+    url=$(add_civitai_token "$url")
     
-    # Função para download individual
-    download_file() {
-        local name="$1"
-        local url="$2"
-        local path="$3"
+    local download_cmd=""
+    local success=false
+    
+    # Try aria2c first
+    if $ARIA2C_AVAILABLE && command_exists aria2c; then
+        local aria_opts="-c -s 16 -x 16 -k 1M --console-log-level=warn --summary-interval=0"
         
-        log "INFO" "Baixando $name..."
+        if [ -n "$filename" ]; then
+            download_cmd="aria2c $aria_opts --dir=\"$target_dir\" --out=\"$filename\" \"$url\""
+        else
+            download_cmd="aria2c $aria_opts --dir=\"$target_dir\" --content-disposition=true \"$url\""
+        fi
         
-        if [[ "$name" == "snapshot" ]]; then
-            # Usar wget para o snapshot
-            if wget --quiet --show-progress --progress=bar:force:noscroll \
-                   --timeout=30 --tries=3 \
-                   "$url" -P "$path"; then
-                log "SUCCESS" "Download concluído: $name"
-                return 0
-            else
-                log "ERROR" "Falha no download: $name"
-                return 1
-            fi
+        if eval "$download_cmd" >/dev/null 2>&1; then
+            success=true
+        fi
+    fi
+    
+    # Try wget if aria2c failed or unavailable
+    if ! $success && command_exists wget; then
+        local wget_opts="-q --show-progress -c"
+        
+        if [ -n "$filename" ]; then
+            download_cmd="wget $wget_opts -O \"$target_dir/$filename\" \"$url\""
         else
-            # Usar aria2c para modelos grandes
-            if aria2c --console-log-level=warn \
-                     --summary-interval=10 \
-                     --download-result=hide \
-                     -c -s 16 -x 16 -k 1M \
-                     --timeout=60 --retry-wait=5 --max-tries=3 \
-                     -d "$path" "$url"; then
-                log "SUCCESS" "Download concluído: $name"
-                return 0
+            download_cmd="wget $wget_opts --content-disposition -P \"$target_dir\" \"$url\""
+        fi
+        
+        if eval "$download_cmd"; then
+            success=true
+        fi
+    fi
+    
+    # Try curl as last resort
+    if ! $success && command_exists curl; then
+        log_warning "Using curl fallback for $url - filename handling may be less robust"
+        
+        if [ -n "$filename" ]; then
+            download_cmd="curl -L -C - -o \"$target_dir/$filename\" \"$url\""
+        else
+            # For curl with content-disposition, it's more complex
+            # Try -J -O with --output-dir (if available) or fallback to simpler approach
+            if curl --help 2>/dev/null | grep -q "output-dir"; then
+                download_cmd="curl -L -J -O --output-dir \"$target_dir\" \"$url\""
             else
-                log "ERROR" "Falha no download: $name"
-                return 1
+                # Fallback: extract filename from URL or use generic name
+                local fallback_name
+                fallback_name=$(basename "$url" | cut -d'?' -f1)
+                [ -z "$fallback_name" ] && fallback_name="downloaded_file"
+                download_cmd="curl -L -C - -o \"$target_dir/$fallback_name\" \"$url\""
             fi
         fi
-    }
+        
+        if eval "$download_cmd"; then
+            success=true
+        fi
+    fi
+    
+    if ! $success; then
+        log_error "Failed to download: $url"
+        return 1
+    fi
+    
+    return 0
+}
+
+# ============================================================================
+# PARALLEL DOWNLOADS
+# ============================================================================
+
+download_all_files() {
+    log_info "Starting parallel downloads..."
     
-    # Iniciar downloads em paralelo
-    for name in "${!DOWNLOADS[@]}"; do
-        download_file "$name" "${DOWNLOADS[$name]}" "${DOWNLOAD_PATHS[$name]}" &
+    local pids=()
+    local download_logs=()
+    
+    for file_spec in "${DOWNLOAD_FILES[@]}"; do
+        IFS='|' read -r url target_subdir filename <<< "$file_spec"
+        
+        local target_dir
+        if [ "$target_subdir" = "INSTALL_DIR" ]; then
+            target_dir="$COMFY_INSTALL_DIR"
+        else
+            target_dir="$MODELS_DIR/$target_subdir"
+        fi
+        
+        # Note: Directory creation is handled by download tools automatically
+        # No need to mkdir -p here as it can cause issues
+        
+        local display_name="${filename:-$(basename "$url" | cut -d'?' -f1)}"
+        log_info "Starting download: $display_name -> $target_dir"
+        
+        # Create a temporary log file for this download
+        local log_file
+        log_file=$(mktemp)
+        download_logs+=("$log_file")
+        
+        # Start download in background
+        (
+            if download_file "$url" "$target_dir" "$filename" >"$log_file" 2>&1; then
+                echo "SUCCESS: Downloaded $display_name" >> "$log_file"
+            else
+                echo "FAILED: Failed to download $display_name" >> "$log_file"
+                exit 1
+            fi
+        ) &
+        
         pids+=($!)
     done
     
-    # Aguardar conclusão de todos os downloads
-    for pid in "${pids[@]}"; do
-        if ! wait "$pid"; then
-            download_success=false
+    # Wait for all downloads to complete
+    local failed_count=0
+    for i in "${!pids[@]}"; do
+        local pid=${pids[$i]}
+        local log_file=${download_logs[$i]}
+        
+        if wait "$pid"; then
+            log_info "Download completed successfully"
+        else
+            failed_count=$((failed_count + 1))
+            log_warning "Download failed - check logs"
+        fi
+        
+        # Show download log
+        if [ -f "$log_file" ]; then
+            cat "$log_file" >&2
+            rm -f "$log_file"
         fi
     done
     
-    if [[ "$download_success" == true ]]; then
-        log "SUCCESS" "Todos os downloads concluídos com sucesso"
+    if [ $failed_count -gt 0 ]; then
+        log_warning "$failed_count downloads failed, but continuing..."
     else
-        log "WARN" "Alguns downloads falharam, mas continuando..."
+        log_info "All downloads completed successfully"
     fi
 }
 
+# ============================================================================
+# SNAPSHOT RESTORATION
+# ============================================================================
+
 restore_snapshot() {
-    log "STEP" "Restaurando snapshot..."
+    log_info "Restoring snapshot..."
     
-    local snapshot_file="${COMFY_DIR}/snapshotBRABEX.json"
+    local snapshot_file="$COMFY_INSTALL_DIR/snapshotBRABEX.json"
     
-    if [[ -f "$snapshot_file" ]]; then
-        if comfy node restore-snapshot "$snapshot_file"; then
-            log "SUCCESS" "Snapshot restaurado com sucesso"
+    if [ -f "$snapshot_file" ]; then
+        if comfy --workspace "$COMFY_INSTALL_DIR" node restore-snapshot "$snapshot_file"; then
+            log_info "Snapshot restored successfully"
         else
-            log "WARN" "Falha ao restaurar snapshot"
+            log_warning "Failed to restore snapshot, ComfyUI will start without it"
         fi
     else
-        log "WARN" "Arquivo de snapshot não encontrado"
+        log_warning "Snapshot file not found: $snapshot_file"
     fi
 }
 
-create_launch_script() {
-    log "STEP" "Criando script de inicialização..."
-    
-    local launch_script="${SCRIPT_DIR}/launch-comfyui.sh"
-    
-    cat > "$launch_script" << EOF
-#!/bin/bash
-# Script de inicialização do ComfyUI
-# Gerado automaticamente em $(date)
+# ============================================================================
+# COMFYUI LAUNCH
+# ============================================================================
 
-cd "$COMFY_DIR" || exit 1
-
-echo "Iniciando ComfyUI..."
-echo "Acesse: http://localhost:${LISTEN_PORT}"
-echo "Ou use o endereço público do seu RunPod/VastAI na porta ${LISTEN_PORT}"
-echo ""
-
-comfy launch -- --fast --listen 0.0.0.0 --port ${LISTEN_PORT}
-EOF
-    
-    chmod +x "$launch_script"
-    log "SUCCESS" "Script de inicialização criado: $launch_script"
-}
-
-cleanup() {
-    log "INFO" "Limpando arquivos temporários..."
-    
-    # Limpar cache do pip
-    pip cache purge 2>/dev/null || true
-    
-    # Limpar logs antigos (manter apenas os últimos 5)
-    find "$SCRIPT_DIR" -name "setup.log.*" -type f | head -n -5 | xargs rm -f 2>/dev/null || true
+launch_comfyui() {
+    log_info "Launching ComfyUI on port 8818..."
+    log_info "ComfyUI will be accessible at: http://0.0.0.0:8818"
     
-    log "SUCCESS" "Limpeza concluída"
+    # Launch ComfyUI
+    exec comfy --workspace "$COMFY_INSTALL_DIR" launch -- --fast --listen 0.0.0.0 --port 8818
 }
 
-show_summary() {
-    echo -e "\n${GREEN}╔══════════════════════════════════════════════════════════════╗${NC}"
-    echo -e "${GREEN}║                    INSTALAÇÃO CONCLUÍDA                     ║${NC}"
-    echo -e "${GREEN}╚══════════════════════════════════════════════════════════════╝${NC}"
-    echo -e "\n${CYAN}Logs da instalação:${NC}"
-    echo -e "${WHITE}  $LOG_FILE${NC}"
-    echo -e "\n${YELLOW}Acesse ComfyUI em:${NC}"
-    echo -e "${WHITE}  http://localhost:$LISTEN_PORT${NC}"
-    echo -e "${WHITE}  ou use o endereço público do seu RunPod/VastAI na porta $LISTEN_PORT${NC}\n"
-}
+# ============================================================================
+# MAIN EXECUTION
+# ============================================================================
 
-launch_comfyui() {
-    log "STEP" "Iniciando ComfyUI automaticamente..."
+main() {
+    log_info "Starting ComfyUI automated setup..."
+    log_info "========================================"
     
-    cd "$COMFY_DIR" || {
-        log "ERROR" "Não foi possível acessar o diretório do ComfyUI"
-        return 1
-    }
+    # Step 1: Environment Detection
+    detect_environment
     
-    echo -e "\n${GREEN}╔══════════════════════════════════════════════════════════════╗${NC}"
-    echo -e "${GREEN}║                      INICIANDO COMFYUI                       ║${NC}"
-    echo -e "${GREEN}╚══════════════════════════════════════════════════════════════╝${NC}"
-    echo -e "\n${CYAN}🚀 ComfyUI está iniciando...${NC}"
-    echo -e "${YELLOW}📱 Acesse: http://localhost:${LISTEN_PORT}${NC}"
-    echo -e "${YELLOW}🌐 Ou use o endereço público do seu RunPod/VastAI na porta ${LISTEN_PORT}${NC}"
-    echo -e "\n${WHITE}Para parar o ComfyUI, pressione Ctrl+C${NC}\n"
+    # Step 2: Permission Setup
+    setup_permissions
     
-    # Pequena pausa para o usuário ler as informações
-    sleep 3
+    # Step 3: System Dependencies
+    install_system_dependencies
     
-    log "INFO" "Executando: comfy launch -- --fast --listen 0.0.0.0 --port $LISTEN_PORT"
+    # Step 4: Install comfy-cli
+    install_comfy_cli
     
-    # Executar ComfyUI
-    exec comfy launch -- --fast --listen 0.0.0.0 --port "$LISTEN_PORT"
-}
-
-# =============================================================================
-# FUNÇÃO PRINCIPAL
-# =============================================================================
-
-main() {
-    local start_time=$(date +%s)
-    
-    # Configurar arquivo de log
-    exec > >(tee -a "$LOG_FILE")
-    exec 2>&1
-    
-    print_banner
-    log "INFO" "Iniciando instalação do ComfyUI..."
-    log "INFO" "Log da instalação: $LOG_FILE"
-    
-    # Verificar se já existe uma instalação
-    if [[ -d "$COMFY_DIR" ]] && [[ -f "${COMFY_DIR}/ComfyUI/main.py" ]]; then
-        log "WARN" "ComfyUI já parece estar instalado em $COMFY_DIR"
-        read -p "Deseja continuar mesmo assim? (y/N): " -n 1 -r
-        echo
-        if [[ ! $REPLY =~ ^[Yy]$ ]]; then
-            log "INFO" "Instalação cancelada pelo usuário"
-            exit 0
-        fi
-    fi
+    # Step 5: Install ComfyUI Core
+    install_comfyui_core
     
-    # Executar etapas da instalação
-    local steps=(
-        "check_dependencies"
-        "install_system_packages"
-        "install_comfy_cli"
-        "install_comfyui"
-        "create_directories"
-        "install_custom_nodes"
-        "download_files"
-        "restore_snapshot"
-        "create_launch_script"
-        "cleanup"
-    )
-    
-    local current_step=1
-    local total_steps=${#steps[@]}
-    
-    for step in "${steps[@]}"; do
-        echo -e "\n${BLUE}[$current_step/$total_steps]${NC} Executando: $step"
-        
-        if $step; then
-            log "SUCCESS" "Etapa concluída: $step"
-        else
-            log "ERROR" "Falha na etapa: $step"
-            log "ERROR" "Instalação interrompida"
-            exit 1
-        fi
-        
-        ((current_step++))
-    done
+    # Step 6: Create Directory Structure
+    create_model_directories
+    
+    # Step 7: Install Custom Nodes
+    install_custom_nodes
     
-    local end_time=$(date +%s)
-    local duration=$((end_time - start_time))
+    # Step 8: Download Files in Parallel
+    download_all_files
     
-    log "SUCCESS" "Instalação concluída em ${duration}s"
-    show_summary
+    # Step 9: Restore Snapshot
+    restore_snapshot
     
-    # Iniciar ComfyUI automaticamente
+    # Step 10: Launch ComfyUI
+    log_info "========================================"
+    log_info "Setup completed successfully!"
+    log_info "========================================"
     launch_comfyui
 }
 
-# =============================================================================
-# TRATAMENTO DE SINAIS
-# =============================================================================
-
-trap 'log "ERROR" "Script interrompido pelo usuário"; exit 1' INT TERM
-
-# =============================================================================
-# EXECUÇÃO
-# =============================================================================
+# ============================================================================
+# SCRIPT EXECUTION
+# ============================================================================
 
-# Verificar se está sendo executado como root (recomendado para RunPod/VastAI)
-if [[ $EUID -ne 0 ]]; then
-    log "WARN" "Script não está sendo executado como root"
-    log "WARN" "Algumas operações podem falhar"
-fi
+# Trap to handle script interruption
+trap 'log_error "Script interrupted"; exit 1' INT TERM
 
-# Executar função principal
+# Run main function
 main "$@"