Upload comfyui_setup_script.sh

comfyui_setup_script.sh

@@ -0,0 +1,378 @@
+#!/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"
+)
+
+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}/"
+)
+
+# =============================================================================
+# 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"
+}
+
+print_banner() {
+    echo -e "${CYAN}"
+    cat << 'EOF'
+    ╔══════════════════════════════════════════════════════════════╗
+    ║                   ComfyUI Setup Script                       ║
+    ║                Professional Installation                      ║
+    ╚══════════════════════════════════════════════════════════════╝
+EOF
+    echo -e "${NC}"
+}
+
+check_dependencies() {
+    log "STEP" "Verificando dependências do sistema..."
+    
+    local deps=("curl" "wget" "python3" "pip")
+    local missing_deps=()
+    
+    for dep in "${deps[@]}"; do
+        if ! command -v "$dep" &> /dev/null; then
+            missing_deps+=("$dep")
+        fi
+    done
+    
+    if [[ ${#missing_deps[@]} -ne 0 ]]; then
+        log "ERROR" "Dependências faltando: ${missing_deps[*]}"
+        return 1
+    fi
+    
+    log "SUCCESS" "Todas as dependências básicas estão disponíveis"
+}
+
+install_system_packages() {
+    log "STEP" "Instalando pacotes do sistema..."
+    
+    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"
+    else
+        log "WARN" "apt-get não encontrado, pulando instalação de pacotes do sistema"
+    fi
+}
+
+install_comfy_cli() {
+    log "STEP" "Instalando comfy-cli..."
+    
+    pip install --quiet --upgrade pip
+    pip install --quiet comfy-cli
+    
+    if ! command -v comfy &> /dev/null; then
+        log "ERROR" "Falha na instalação do comfy-cli"
+        return 1
+    fi
+    
+    log "SUCCESS" "comfy-cli instalado com sucesso"
+}
+
+install_comfyui() {
+    log "STEP" "Instalando ComfyUI com suporte NVIDIA..."
+    
+    comfy --skip-prompt install --nvidia
+    
+    if [[ ! -d "$COMFY_DIR" ]]; then
+        log "ERROR" "Diretório do ComfyUI não foi criado"
+        return 1
+    fi
+    
+    log "SUCCESS" "ComfyUI instalado com sucesso"
+}
+
+install_custom_nodes() {
+    log "STEP" "Instalando nós customizados..."
+    
+    local nodes=(
+        "https://github.com/Cezarsaint/custom_prompt_manager"
+        "https://github.com/adbrasi/pakreatorio"
+    )
+    
+    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
+    done
+}
+
+create_directories() {
+    log "STEP" "Criando estrutura de diretórios..."
+    
+    local dirs=(
+        "${COMFY_DIR}/ComfyUI/models/checkpoints"
+        "${COMFY_DIR}/ComfyUI/models/loras"
+    )
+    
+    for dir in "${dirs[@]}"; do
+        mkdir -p "$dir"
+        log "INFO" "Diretório criado: $dir"
+    done
+}
+
+download_files() {
+    log "STEP" "Iniciando downloads paralelos dos modelos..."
+    
+    local pids=()
+    local download_success=true
+    
+    # Função para download individual
+    download_file() {
+        local name="$1"
+        local url="$2"
+        local path="$3"
+        
+        log "INFO" "Baixando $name..."
+        
+        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
+        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
+            else
+                log "ERROR" "Falha no download: $name"
+                return 1
+            fi
+        fi
+    }
+    
+    # Iniciar downloads em paralelo
+    for name in "${!DOWNLOADS[@]}"; do
+        download_file "$name" "${DOWNLOADS[$name]}" "${DOWNLOAD_PATHS[$name]}" &
+        pids+=($!)
+    done
+    
+    # Aguardar conclusão de todos os downloads
+    for pid in "${pids[@]}"; do
+        if ! wait "$pid"; then
+            download_success=false
+        fi
+    done
+    
+    if [[ "$download_success" == true ]]; then
+        log "SUCCESS" "Todos os downloads concluídos com sucesso"
+    else
+        log "WARN" "Alguns downloads falharam, mas continuando..."
+    fi
+}
+
+restore_snapshot() {
+    log "STEP" "Restaurando snapshot..."
+    
+    local snapshot_file="${COMFY_DIR}/snapshotBRABEX.json"
+    
+    if [[ -f "$snapshot_file" ]]; then
+        if comfy node restore-snapshot "$snapshot_file"; then
+            log "SUCCESS" "Snapshot restaurado com sucesso"
+        else
+            log "WARN" "Falha ao restaurar snapshot"
+        fi
+    else
+        log "WARN" "Arquivo de snapshot não encontrado"
+    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)
+
+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
+    
+    log "SUCCESS" "Limpeza concluída"
+}
+
+show_summary() {
+    echo -e "\n${GREEN}╔══════════════════════════════════════════════════════════════╗${NC}"
+    echo -e "${GREEN}║                    INSTALAÇÃO CONCLUÍDA                     ║${NC}"
+    echo -e "${GREEN}╚══════════════════════════════════════════════════════════════╝${NC}"
+    echo -e "\n${CYAN}Para iniciar o ComfyUI:${NC}"
+    echo -e "${WHITE}  ./launch-comfyui.sh${NC}"
+    echo -e "\n${CYAN}Ou execute manualmente:${NC}"
+    echo -e "${WHITE}  cd $COMFY_DIR && comfy launch -- --fast --listen 0.0.0.0 --port $LISTEN_PORT${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"
+}
+
+# =============================================================================
+# 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
+    
+    # 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
+    
+    local end_time=$(date +%s)
+    local duration=$((end_time - start_time))
+    
+    log "SUCCESS" "Instalação concluída em ${duration}s"
+    show_summary
+}
+
+# =============================================================================
+# TRATAMENTO DE SINAIS
+# =============================================================================
+
+trap 'log "ERROR" "Script interrompido pelo usuário"; exit 1' INT TERM
+
+# =============================================================================
+# EXECUÇÃO
+# =============================================================================
+
+# 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
+
+# Executar função principal
+main "$@"