Upload causvid_installer.sh

causvid_installer.sh

@@ -0,0 +1,220 @@
+#!/bin/bash
+
+# CausVid ComfyUI Installer
+# Simple installer for CausVid workflow without model downloads
+
+set -e
+set -o pipefail
+
+# ============================================================================
+# GLOBAL CONFIGURATIONS
+# ============================================================================
+
+readonly WORKFLOW_URL="https://huggingface.co/adbrasi/testedownload/resolve/main/CausVid-WAN-14B-Flow.json"
+readonly WORKSPACE_DIR="/workspace/ComfyUI"
+readonly WORKFLOW_FILE="/workspace/CausVid-WAN-14B-Flow.json"
+
+# ============================================================================
+# UTILITY FUNCTIONS
+# ============================================================================
+
+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
+}
+
+# ============================================================================
+# CLEANUP FUNCTION
+# ============================================================================
+
+cleanup_existing_installation() {
+    log_info "Checking for existing installations..."
+    
+    if [ -d "$WORKSPACE_DIR" ]; then
+        log_warning "Found existing directory: $WORKSPACE_DIR"
+        log_info "Removing existing installation to avoid conflicts..."
+        rm -rf "$WORKSPACE_DIR"
+    fi
+    
+    # Also clean up any potential leftover directories
+    if [ -d "/workspace/ComfyUi" ]; then
+        log_info "Removing /workspace/ComfyUi..."
+        rm -rf "/workspace/ComfyUi"
+    fi
+    
+    if [ -d "/workspace/comfyui" ]; then
+        log_info "Removing /workspace/comfyui..."
+        rm -rf "/workspace/comfyui"
+    fi
+}
+
+# ============================================================================
+# COMFY-CLI INSTALLATION
+# ============================================================================
+
+install_comfy_cli() {
+    log_info "Installing/upgrading comfy-cli..."
+    
+    python3 -m pip install --upgrade comfy-cli || {
+        log_error "Failed to install comfy-cli"
+        exit 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_info "comfy-cli version: $(comfy --version 2>/dev/null || echo 'unknown')"
+}
+
+# ============================================================================
+# COMFYUI INSTALLATION
+# ============================================================================
+
+install_comfyui() {
+    log_info "Installing ComfyUI to $WORKSPACE_DIR..."
+    
+    # Ensure parent directory exists
+    mkdir -p "$(dirname "$WORKSPACE_DIR")"
+    
+    # Install ComfyUI with fast dependencies
+    comfy --workspace "$WORKSPACE_DIR" --skip-prompt install --nvidia --fast-deps || {
+        log_error "Failed to install ComfyUI"
+        exit 1
+    }
+    
+    log_info "ComfyUI installed successfully"
+}
+
+# ============================================================================
+# WORKFLOW DOWNLOAD
+# ============================================================================
+
+download_workflow() {
+    log_info "Downloading CausVid workflow..."
+    
+    local success=false
+    
+    # Try different download methods
+    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
+        if aria2c -q -o "$(basename "$WORKFLOW_FILE")" -d "$(dirname "$WORKFLOW_FILE")" "$WORKFLOW_URL"; then
+            success=true
+        fi
+    else
+        log_error "No download tool available (wget, curl, or aria2c)"
+        exit 1
+    fi
+    
+    if ! $success; then
+        log_error "Failed to download workflow file"
+        exit 1
+    fi
+    
+    if [ ! -f "$WORKFLOW_FILE" ]; then
+        log_error "Workflow file was not downloaded successfully"
+        exit 1
+    fi
+    
+    log_info "Workflow downloaded to: $WORKFLOW_FILE"
+}
+
+# ============================================================================
+# WORKFLOW DEPENDENCIES
+# ============================================================================
+
+install_workflow_dependencies() {
+    log_info "Installing workflow dependencies..."
+    
+    if [ ! -f "$WORKFLOW_FILE" ]; then
+        log_error "Workflow file not found: $WORKFLOW_FILE"
+        exit 1
+    fi
+    
+    comfy node install-deps --workflow="$WORKFLOW_FILE" || {
+        log_warning "Some workflow dependencies may have failed to install"
+        log_info "This is often normal - ComfyUI will show missing nodes on startup"
+    }
+    
+    log_info "Workflow dependencies installation completed"
+}
+
+# ============================================================================
+# LAUNCH COMFYUI
+# ============================================================================
+
+launch_comfyui() {
+    log_info "Launching ComfyUI on port 8818..."
+    log_info "ComfyUI will be accessible at: http://0.0.0.0:8818"
+    log_info "Workflow file is available at: $WORKFLOW_FILE"
+    
+    # Launch ComfyUI
+    exec comfy --workspace "$WORKSPACE_DIR" launch -- --fast --listen 0.0.0.0 --port 8818
+}
+
+# ============================================================================
+# MAIN EXECUTION
+# ============================================================================
+
+main() {
+    log_info "Starting CausVid ComfyUI installation..."
+    log_info "======================================"
+    
+    # Step 1: Clean up any existing installations
+    cleanup_existing_installation
+    
+    # Step 2: Install comfy-cli
+    install_comfy_cli
+    
+    # Step 3: Install ComfyUI
+    install_comfyui
+    
+    # Step 4: Download workflow
+    download_workflow
+    
+    # Step 5: Install workflow dependencies
+    install_workflow_dependencies
+    
+    # Step 6: Launch ComfyUI
+    log_info "======================================"
+    log_info "Installation completed successfully!"
+    log_info "======================================"
+    launch_comfyui
+}
+
+# ============================================================================
+# SCRIPT EXECUTION
+# ============================================================================
+
+# Trap to handle script interruption
+trap 'log_error "Script interrupted"; exit 1' INT TERM
+
+# Run main function
+main "$@"