#!/usr/bin/expect

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

# 定义测试指令
# 这是一个更通用的测试命令，先尝试列出文件确认脚本名
puts "--> Starting DreamOmni2 Inference Test..."

spawn ssh -p $port $user@$host
expect {
    "password:" { send "$password\r" }
    "yes/no" { send "yes\r"; exp_continue }
}

expect "#"
send "source ~/miniconda3/bin/activate\r"
send "conda activate dreamomni2\r"
send "cd ~/ai-projects/DreamOmni2\r"

# 1. 先检查有哪些脚本
puts "--> Checking available scripts..."
send "ls -F\r"
expect "#"

# 2. 运行示例
puts "--> Running inference_gen.py (Text-to-Image Generation)..."
# 注意：这里我们使用默认参数运行，或者查看脚本头部了解参数
# 假设 inference_gen.py 是官方提供的生成脚本
send "python inference_gen.py\r"

# 等待输出，这可能需要很长时间（加载模型+生成）
expect {
    "Traceback" {
        puts "\n[ERROR] Python Traceback detected!"
    }
    "Error" {
        puts "\n[ERROR] General error detected!"
    }
    "Saved" {
        puts "\n[SUCCESS] Image saved successfully!"
    }
    "Done" {
         puts "\n[SUCCESS] Process completed!"
    }
    "#" {
        # Catch prompt return if no specific message
    }
}

# 3. 检查是否有结果生成
puts "--> Checking for generated images..."
send "ls -lh *.png *.jpg example_input/gen_tests/ 2>/dev/null\r"
expect "#"

send "exit\r"
expect eof

puts "Test run finished."