#!/usr/bin/expect

set host "connect.nmb2.seetacloud.com"
set port "15054"
set user "root"
set password "9Dml+WZeqp5b"
set remote_dir "/root/autodl-tmp/book-rec-with-LLMs"

# Install dependencies if needed
set cmd_pip "/root/miniconda3/bin/pip install lightgbm pandas tqdm scikit-learn"

# Run Evaluate
# We need to set PYTHONPATH because evaluation script imports src.
set cmd_eval "export PYTHONPATH=. && /root/miniconda3/bin/python scripts/evaluate.py"

spawn ssh -p $port $user@$host "cd $remote_dir; $cmd_pip; $cmd_eval"
set timeout 1200

expect {
    "password:" {
        send "$password\r"
        exp_continue
    }
    "continue connecting" {
        send "yes\r"
        exp_continue
    }
    eof
}