Update ImageStudioTESTER.sh

ImageStudioTESTER.sh

@@ -16,6 +16,24 @@ log_success() { echo -e "${GREEN}[✓]${NC} $1"; }
 log_warn() { echo -e "${YELLOW}[!]${NC} $1"; }
 log_error() { echo -e "${RED}[✗]${NC} $1"; }
 
+drop_caches() {
+    sync
+    echo 3 > /proc/sys/vm/drop_caches || true
+}
+
+flush_cache_if_large() {
+    local file_path="$1"
+    if [ -f "$file_path" ]; then
+        local sz
+        sz=$(stat -c%s "$file_path" 2>/dev/null || echo 0)
+        if [ "$sz" -ge 209715200 ]; then
+            drop_caches
+        fi
+    else
+        drop_caches
+    fi
+}
+
 # Configuração
 COMFY_BASE="/workspace/comfy"
 COMFY_DIR="$COMFY_BASE/ComfyUI"
@@ -25,8 +43,19 @@ COMFY_PORT="8818"
 VENV_DIR="$COMFY_BASE/.venv"
 CIVITAI_TOKEN="4fcb2834969399006a736ee402b061e5"
 
-# Garantir que o diretório base exista para instalar fora de /root
-mkdir -p "$COMFY_BASE"
+export DEBIAN_FRONTEND=noninteractive
+export HF_HOME="/workspace/.hf"
+export HUGGINGFACE_HUB_CACHE="$HF_HOME"
+export TRANSFORMERS_CACHE="$HF_HOME"
+export TMPDIR="/workspace/.tmp"
+export GIT_LFS_SKIP_SMUDGE=1
+export PIP_NO_CACHE_DIR=off
+
+PIP_FLAGS="--no-cache-dir --prefer-binary --only-binary=:all:"
+ARIA="aria2c -c -x 2 -s 2 --disk-cache=0 --file-allocation=none --console-log-level=warn"
+
+# Garantir diretórios necessários para instalar fora de /root e concentrar caches
+mkdir -p "$COMFY_BASE" "$HF_HOME" "$TMPDIR"
 
 # Performance
 export MAX_JOBS=32
@@ -152,6 +181,12 @@ download_hf() {
     else
         wget -q --show-progress -c -O "$target_dir/$filename" "https://huggingface.co/$url"
     fi
+
+    if [ -n "$filename" ]; then
+        flush_cache_if_large "$target_dir/$filename"
+    else
+        drop_caches
+    fi
 }
 
 # Download Mega
@@ -172,6 +207,7 @@ download_mega() {
         log_warn "Mega download falhou"
     fi
     cd - >/dev/null
+    drop_caches
 }
 
 # Download normal
@@ -195,11 +231,11 @@ download_file() {
     # Tentar com aria2c primeiro
     if command -v aria2c >/dev/null 2>&1; then
         if [ -n "$filename" ]; then
-            aria2c -c -x 4 -s 4 --console-log-level=warn --dir="$target_dir" --out="$filename" "$url" || \
+            $ARIA --dir="$target_dir" --out="$filename" "$url" || \
             wget -q --show-progress -c -O "$target_dir/$filename" "$url" || \
             log_error "Download falhou: $filename"
         else
-            aria2c -c -x 4 -s 4 --console-log-level=warn --dir="$target_dir" "$url" || \
+            $ARIA --dir="$target_dir" "$url" || \
             (cd "$target_dir" && wget -q --show-progress -c "$url") || \
             log_error "Download falhou: $url"
         fi
@@ -210,6 +246,12 @@ download_file() {
         (cd "$target_dir" && wget -q --show-progress -c "$url") || \
         log_error "Download falhou: $url"
     fi
+
+    if [ -n "$filename" ]; then
+        flush_cache_if_large "$target_dir/$filename"
+    else
+        drop_caches
+    fi
 }
 
 # Processar downloads
@@ -244,10 +286,21 @@ clone_repo() {
         git clone --depth 1 "$url" "$dest" 2>/dev/null || true
     fi
     
-    # Instalar requirements se existir
-    if [ -f "$dest/requirements.txt" ]; then
-        python -m pip install -q -r "$dest/requirements.txt" 2>/dev/null || true
+}
+
+collect_requirements_and_install() {
+    local base="$COMFY_DIR/custom_nodes"
+    local reqtmp="$TMPDIR/_all_requirements.txt"
+    : > "$reqtmp"
+    if [ -d "$base" ]; then
+        find "$base" -maxdepth 2 -type f -name "requirements.txt" -print0 | xargs -0 -I {} cat "{}" >> "$reqtmp" || true
+        sort -u -o "$reqtmp" "$reqtmp"
+        if [ -s "$reqtmp" ]; then
+            log_info "Instalando requirements combinados dos custom nodes"
+            python -m pip install ${PIP_FLAGS} -r "$reqtmp"
+        fi
     fi
+    rm -f "$reqtmp"
 }
 
 # ========== INSTALAÇÃO ==========
@@ -260,7 +313,9 @@ log_info "========================================="
 log_info "[1/6] Instalando dependências do sistema..."
 
 apt-get update -qq
-apt-get install -y -qq python3-venv aria2 megatools git wget curl 2>/dev/null
+apt-get install -y -qq --no-install-recommends python3-venv aria2 megatools git wget curl 2>/dev/null
+apt-get clean
+rm -rf /var/lib/apt/lists/*
 
 log_success "Dependências do sistema instaladas"
 
@@ -275,8 +330,8 @@ fi
 
 . "$VENV_DIR/bin/activate"
 
-python -m pip install -U pip wheel setuptools -q
-python -m pip install -U "huggingface_hub[cli,hf_transfer]" comfy-cli -q
+python -m pip install ${PIP_FLAGS} -U pip wheel setuptools -q
+python -m pip install ${PIP_FLAGS} -U "huggingface_hub[cli,hf_transfer]" comfy-cli -q
 
 log_success "Ambiente virtual pronto e comfy-cli instalado"
 
@@ -298,7 +353,7 @@ GPU_INFO=$(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null || echo
 
 if [[ "$GPU_INFO" == *"5090"* ]] || [[ "$GPU_INFO" == *"5080"* ]]; then
     log_warn "RTX 5090/5080 detectada - instalando PyTorch nightly"
-    python -m pip install --force-reinstall --pre torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu129
+    python -m pip install ${PIP_FLAGS} --force-reinstall --pre torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu129
 else
     log_info "GPU: ${GPU_INFO:-Não detectada}"
 fi
@@ -323,6 +378,10 @@ for repo in "${CUSTOM_NODES[@]}"; do
     clone_repo "$repo" "$CN_DIR/$node_name"
 done
 
+collect_requirements_and_install
+python -m pip cache purge || true
+conda clean -a -y 2>/dev/null || true
+
 log_success "Custom nodes instalados"
 
 # Verificar e iniciar
@@ -332,5 +391,8 @@ log_info "Iniciando ComfyUI..."
 log_info "URL: http://localhost:$COMFY_PORT"
 log_info "========================================="
 
+rm -rf "$TMPDIR"
+mkdir -p "$TMPDIR"
+
 cd "$COMFY_DIR"
 exec comfy --workspace "$COMFY_DIR" launch -- --listen "$COMFY_HOST" --preview-method latent2rgb --front-end-version Comfy-Org/ComfyUI_frontend@v1.28.8 --port "$COMFY_PORT"