Upload ComfyWAN_installer.sh

ComfyWAN_installer.sh

@@ -0,0 +1,226 @@
+#!/bin/bash
+
+# ============================================================================
+# Instalador Dinâmico para ComfyUI
+# Instala o ComfyUI e suas dependências no diretório atual.
+# ============================================================================
+
+set -e
+set -o pipefail
+
+# ============================================================================
+# CONFIGURAÇÕES GLOBAIS
+# ============================================================================
+
+# URL do workflow a ser baixado
+readonly WORKFLOW_URL="https://huggingface.co/adbrasi/testedownload/resolve/main/modo%20balanco3.json"
+
+# --- Caminhos Dinâmicos ---
+# Define o diretório de instalação como o diretório atual onde o script é executado
+readonly INSTALL_BASE_DIR="$(pwd)"
+# O ComfyUI será instalado em uma subpasta "ComfyUI"
+readonly WORKSPACE_DIR="$INSTALL_BASE_DIR/ComfyUI"
+# O arquivo de workflow será salvo na raiz do diretório de instalação
+readonly WORKFLOW_FILENAME="workflow.json"
+readonly WORKFLOW_FILE="$INSTALL_BASE_DIR/$WORKFLOW_FILENAME"
+
+# ============================================================================
+# FUNÇÕES DE UTILIDADE
+# ============================================================================
+
+log_info() {
+    echo "INFO: $1" >&2
+}
+
+log_warning() {
+    echo "WARNING: $1" >&2
+}
+
+log_error() {
+    echo "ERROR: $1" >&2
+}
+
+command_exists() {
+    command -v "$1" >/dev/null 2>&1
+}
+
+# ============================================================================
+# FUNÇÃO DE LIMPEZA
+# ============================================================================
+
+cleanup_existing_installation() {
+    log_info "Verificando instalações existentes em: $WORKSPACE_DIR"
+    
+    if [ -d "$WORKSPACE_DIR" ]; then
+        log_warning "Diretório existente encontrado: $WORKSPACE_DIR"
+        log_info "Removendo a instalação existente para evitar conflitos..."
+        rm -rf "$WORKSPACE_DIR"
+        log_info "Diretório removido."
+    else
+        log_info "Nenhuma instalação anterior encontrada no local. Continuando..."
+    fi
+}
+
+# ============================================================================
+# INSTALAÇÃO DO COMFY-CLI
+# ============================================================================
+
+install_comfy_cli() {
+    log_info "Instalando/atualizando comfy-cli..."
+    
+    python3 -m pip install --upgrade comfy-cli || {
+        log_error "Falha ao instalar comfy-cli"
+        exit 1
+    }
+    
+    # Verifica se o 'comfy' está no PATH
+    if ! command_exists comfy; then
+        if [ -d "$HOME/.local/bin" ] && [[ ":$PATH:" != *":$HOME/.local/bin:"* ]]; then
+            export PATH="$HOME/.local/bin:$PATH"
+            log_info "Adicionado $HOME/.local/bin ao PATH para esta sessão."
+        fi
+        
+        if ! command_exists comfy; then
+            log_error "Comando 'comfy' não encontrado no PATH após a instalação."
+            log_error "Verifique se '$HOME/.local/bin' está em seu PATH."
+            exit 1
+        fi
+    fi
+    
+    log_info "comfy-cli version: $(comfy --version 2>/dev/null || echo 'desconhecida')"
+}
+
+# ============================================================================
+# INSTALAÇÃO DO COMFYUI
+# ============================================================================
+
+install_comfyui() {
+    log_info "Instalando ComfyUI em: $WORKSPACE_DIR"
+    
+    # Instala ComfyUI com dependências rápidas
+    comfy --workspace "$WORKSPACE_DIR" --skip-prompt install --nvidia --fast-deps || {
+        log_error "Falha ao instalar ComfyUI"
+        exit 1
+    }
+    
+    log_info "ComfyUI instalado com sucesso."
+}
+
+# ============================================================================
+# DOWNLOAD DO WORKFLOW
+# ============================================================================
+
+download_workflow() {
+    log_info "Baixando o workflow de: $WORKFLOW_URL"
+    log_info "Salvando como: $WORKFLOW_FILE"
+    
+    local success=false
+    
+    # Tenta diferentes métodos de download
+    if command_exists wget; then
+        if wget -q --show-progress -O "$WORKFLOW_FILE" "$WORKFLOW_URL"; then
+            success=true
+        fi
+    elif command_exists curl; then
+        if curl -L -o "$WORKFLOW_FILE" "$WORKFLOW_URL"; then
+            success=true
+        fi
+    elif command_exists aria2c; then
+        # aria2c precisa do nome do arquivo e diretório separados
+        if aria2c -q -o "$(basename "$WORKFLOW_FILE")" -d "$(dirname "$WORKFLOW_FILE")" "$WORKFLOW_URL"; then
+            success=true
+        fi
+    else
+        log_error "Nenhuma ferramenta de download disponível (wget, curl ou aria2c)."
+        exit 1
+    fi
+    
+    if ! $success; then
+        log_error "Falha ao baixar o arquivo do workflow."
+        exit 1
+    fi
+    
+    if [ ! -f "$WORKFLOW_FILE" ]; then
+        log_error "O arquivo do workflow não foi baixado com sucesso."
+        exit 1
+    fi
+    
+    log_info "Workflow baixado com sucesso."
+}
+
+# ============================================================================
+# DEPENDÊNCIAS DO WORKFLOW
+# ============================================================================
+
+install_workflow_dependencies() {
+    log_info "Instalando dependências do workflow a partir de: $WORKFLOW_FILE"
+    
+    if [ ! -f "$WORKFLOW_FILE" ]; then
+        log_error "Arquivo de workflow não encontrado: $WORKFLOW_FILE"
+        exit 1
+    fi
+    
+    # Usa o workspace correto para instalar as dependências
+    comfy --workspace "$WORKSPACE_DIR" node install-deps --workflow="$WORKFLOW_FILE" || {
+        log_warning "Algumas dependências do workflow podem ter falhado ao instalar."
+        log_info "Isso pode ser normal. O ComfyUI mostrará os nós ausentes ao iniciar."
+    }
+    
+    log_info "Instalação de dependências do workflow concluída."
+}
+
+# ============================================================================
+# INICIAR O COMFYUI
+# ============================================================================
+
+launch_comfyui() {
+    log_info "Iniciando ComfyUI na porta 8818..."
+    log_info "Workspace do ComfyUI: $WORKSPACE_DIR"
+    log_info "Arquivo de workflow disponível em: $WORKFLOW_FILE"
+    log_info "Para usar o workflow, arraste e solte o arquivo '$WORKFLOW_FILENAME' na interface do ComfyUI."
+    log_info "Acesse o ComfyUI em: http://127.0.0.1:8818 (ou o IP da sua máquina)"
+    
+    # Inicia o ComfyUI
+    exec comfy --workspace "$WORKSPACE_DIR" launch -- --fast --listen 0.0.0.0 --port 8818
+}
+
+# ============================================================================
+# EXECUÇÃO PRINCIPAL
+# ============================================================================
+
+main() {
+    log_info "Iniciando instalação dinâmica do ComfyUI..."
+    log_info "Diretório de instalação base: $INSTALL_BASE_DIR"
+    log_info "================================================="
+    
+    # Passo 1: Limpar instalações existentes no diretório alvo
+    cleanup_existing_installation
+    
+    # Passo 2: Instalar comfy-cli
+    install_comfy_cli
+    
+    # Passo 3: Instalar ComfyUI
+    install_comfyui
+    
+    # Passo 4: Baixar o workflow
+    download_workflow
+    
+    # Passo 5: Instalar dependências do workflow
+    install_workflow_dependencies
+    
+    # Passo 6: Iniciar o ComfyUI
+    log_info "================================================="
+    log_info "Instalação concluída com sucesso!"
+    log_info "================================================="
+    launch_comfyui
+}
+
+# ============================================================================
+# EXECUÇÃO DO SCRIPT
+# ============================================================================
+
+# Captura interrupções (Ctrl+C)
+trap 'log_error "Script interrompido pelo usuário."; exit 1' INT TERM
+
+# Executa a função principal
+main "$@"