#!/usr/bin/expect

set timeout -1
set host "156.225.27.52"
set port "5204"
set user "root"
set password "Jsjj@8888"

spawn ssh -p $port $user@$host "
    echo '--- GPU Status ---'
    nvidia-smi --query-gpu=name,memory.total,memory.free --format=csv,noheader
    
    echo '--- Conda Environment ---'
    source ~/miniconda3/bin/activate
    conda env list | grep dreamomni2
    
    echo '--- Installed Packages (Top 5) ---'
    conda activate dreamomni2
    pip list | grep -E 'torch|diffusers|transformers|accelerate|modelscope'
    
    echo '--- Model Files ---'
    ls -lh ~/ai-projects/DreamOmni2/checkpoints/DreamOmni2 | head -n 5
    
    echo '--- Code Files ---'
    ls -F ~/ai-projects/DreamOmni2
"

expect {
    "password:" {
        send "$password\r"
        exp_continue
    }
    "yes/no" {
        send "yes\r"
        exp_continue
    }
    eof
}