#!/usr/bin/expect
set timeout 30
set host "156.225.27.52"
set port "5204"
set user "root"
set password $env(DEPLOY_SSH_PASSWORD)

spawn ssh -o StrictHostKeyChecking=no -p $port $user@$host "echo 'Connected successfully' && nvidia-smi"
expect {
    "password:" {
        send "$password\r"
        exp_continue
    }
    "yes/no" {
        send "yes\r"
        exp_continue
    }
    eof
}