Update image_Studio.sh

image_Studio.sh

@@ -4,6 +4,13 @@
 
 set -euo pipefail
 
+# Detectar se está rodando via pipe (não-interativo)
+if [ ! -t 0 ] || [ ! -t 1 ]; then
+    INTERACTIVE=false
+else
+    INTERACTIVE=true
+fi
+
 # -----------------------------
 # Cores para output
 # -----------------------------
@@ -61,7 +68,13 @@ save_state() {
 
 get_state() {
     if [ -f "$STATE_FILE" ]; then
-        cat "$STATE_FILE"
+        local state=$(cat "$STATE_FILE" 2>/dev/null)
+        # Verificar se é um número válido
+        if [[ "$state" =~ ^[0-9]+$ ]]; then
+            echo "$state"
+        else
+            echo "0"
+        fi
     else
         echo "0"
     fi
@@ -502,7 +515,24 @@ show_menu() {
     
     local current_state=$(get_state)
     
-    if [ "$current_state" != "0" ]; then
+    # Verificar se o estado está vazio ou inválido
+    if [ -z "$current_state" ] || ! [[ "$current_state" =~ ^[0-9]+$ ]]; then
+        log_info "Iniciando nova instalação..."
+        reset_state
+        return 0
+    fi
+    
+    # Se não for interativo, sempre continuar ou começar do zero
+    if [ "$INTERACTIVE" = false ]; then
+        if [ "$current_state" != "0" ] && [ "$current_state" -lt "8" ]; then
+            log_info "Modo não-interativo: continuando instalação anterior (Step: $current_state)"
+        else
+            log_info "Modo não-interativo: iniciando nova instalação"
+        fi
+        return 0
+    fi
+    
+    if [ "$current_state" != "0" ] && [ "$current_state" -lt "8" ]; then
         log_warn "Instalação anterior detectada (Step: $current_state)"
         echo ""
         echo "Opções:"
@@ -510,13 +540,15 @@ show_menu() {
         echo "  2) Reiniciar instalação do zero"
         echo "  3) Sair"
         echo ""
-        read -p "Escolha uma opção [1-3]: " choice
+        
+        # Timeout de 10 segundos, padrão é continuar
+        read -t 10 -p "Escolha uma opção [1-3] (padrão: 1 em 10s): " choice || choice="1"
         
         case $choice in
-            1) return 0 ;;
+            1|"") return 0 ;;
             2) reset_state; return 0 ;;
             3) exit 0 ;;
-            *) echo "Opção inválida"; exit 1 ;;
+            *) log_warn "Opção inválida, continuando..."; return 0 ;;
         esac
     fi
     
@@ -685,16 +717,26 @@ main() {
     log_success " Instalação concluída!"
     log_success "========================================="
     
+    # Se não for interativo ou se for via argumento, iniciar direto
+    if [ "$INTERACTIVE" = false ] || [ "${1:-}" = "--start" ]; then
+        log_info "Iniciando ComfyUI automaticamente..."
+        log_info "URL: http://localhost:$COMFY_PORT"
+        cd "$COMFY_DIR"
+        exec comfy launch -- --listen "$COMFY_HOST" --port "$COMFY_PORT"
+    fi
+    
     echo ""
     echo "Opções:"
     echo "  1) Iniciar ComfyUI agora"
     echo "  2) Resetar estado da instalação"
     echo "  3) Sair"
     echo ""
-    read -p "Escolha uma opção [1-3]: " final_choice
+    
+    # Timeout de 10 segundos, padrão é iniciar
+    read -t 10 -p "Escolha uma opção [1-3] (padrão: 1 em 10s): " final_choice || final_choice="1"
     
     case $final_choice in
-        1)
+        1|"")
             log_info "Iniciando ComfyUI..."
             log_info "URL: http://localhost:$COMFY_PORT"
             cd "$COMFY_DIR"
@@ -708,7 +750,9 @@ main() {
             log_info "Saindo..."
             ;;
         *)
-            log_warn "Opção inválida"
+            log_info "Iniciando ComfyUI (opção padrão)..."
+            cd "$COMFY_DIR"
+            exec comfy launch -- --listen "$COMFY_HOST" --port "$COMFY_PORT"
             ;;
     esac
 }