Update image_Studio.sh

image_Studio.sh

@@ -116,12 +116,9 @@ download_hf() {
     if command -v hf >/dev/null 2>&1; then
         cd "$target_dir"
         hf download "$(dirname "$url")" "$(basename "$url")" --local-dir . 2>/dev/null
-        local result=$?
         cd - >/dev/null
-        return $result
     else
-        wget -q --show-progress -c -O "$target_dir/$filename" "https://huggingface.co/$url" 2>/dev/null
-        return $?
+        wget -q --show-progress -c -O "$target_dir/$filename" "https://huggingface.co/$url"
     fi
 }
 
@@ -139,13 +136,10 @@ download_mega() {
     # Usar megadl simples sem forçar nome
     if megadl "$url" 2>/dev/null; then
         log_success "Mega download OK"
-        cd - >/dev/null
-        return 0
     else
         log_warn "Mega download falhou"
-        cd - >/dev/null
-        return 1
     fi
+    cd - >/dev/null
 }
 
 # Download normal
@@ -169,45 +163,26 @@ 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" 2>/dev/null
+            aria2c -c -x 4 -s 4 --console-log-level=warn --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" 2>/dev/null
+            aria2c -c -x 4 -s 4 --console-log-level=warn --dir="$target_dir" "$url" || \
+            (cd "$target_dir" && wget -q --show-progress -c "$url") || \
+            log_error "Download falhou: $url"
         fi
-        
-        # Se aria2c funcionou, retornar sucesso
-        if [ $? -eq 0 ]; then
-            return 0
-        else
-            log_warn "aria2c falhou, tentando wget..."
-        fi
-    fi
-    
-    # Fallback para wget
-    if [ -n "$filename" ]; then
-        wget -q --show-progress -c -O "$target_dir/$filename" "$url" 2>/dev/null
-    else
-        cd "$target_dir"
-        wget -q --show-progress -c "$url" 2>/dev/null
-        cd - >/dev/null
-    fi
-    
-    # Verificar resultado do wget
-    if [ $? -eq 0 ]; then
-        log_success "Download OK com wget"
-        return 0
+    elif [ -n "$filename" ]; then
+        wget -q --show-progress -c -O "$target_dir/$filename" "$url" || \
+        log_error "Download falhou: $filename"
     else
-        log_error "Download falhou com aria2c e wget: ${filename:-$url}"
-        return 1
+        (cd "$target_dir" && wget -q --show-progress -c "$url") || \
+        log_error "Download falhou: $url"
     fi
 }
 
 # Processar downloads
 process_downloads() {
-    local total=0
-    local failed=0
-    
     for entry in "${DOWNLOAD_FILES[@]}"; do
-        ((total++))
         IFS='|' read -r url type filename <<< "$entry"
         
         # Criar diretório
@@ -216,21 +191,14 @@ process_downloads() {
         
         # Baixar arquivo
         if [[ "$url" == mega://* ]]; then
-            download_mega "$url" "$target_dir" || ((failed++))
+            download_mega "$url" "$target_dir"
         elif [[ "$url" == *.huggingface.co/* ]]; then
             local hf_path="${url#*huggingface.co/}"
-            download_hf "$hf_path" "$target_dir" "$filename" || ((failed++))
+            download_hf "$hf_path" "$target_dir" "$filename"
         else
-            download_file "$url" "$target_dir" "$filename" || ((failed++))
+            download_file "$url" "$target_dir" "$filename"
         fi
     done
-    
-    # Reportar resultado
-    if [ $failed -gt 0 ]; then
-        log_warn "Downloads: $((total-failed))/$total OK ($failed falharam)"
-    else
-        log_success "Todos os $total downloads completados!"
-    fi
 }
 
 # Clone git repo