#!/bin/bash

# 定义日志目录
LOG_DIR="/home/th/jcq/AI_AutoPlat/logs"
mkdir -p $LOG_DIR

# 定义要运行的程序及其工作目录
declare -A PROGRAMS=(
    ["web_dsj"]="/home/th/jcq/AI_AutoPlat/AI_web_dsj/AI_auto_train.py"
    ["yolov5"]="/home/th/jcq/AI_AutoPlat/yolov5-th/ai_auto_train_yolov5.py"
)

declare -A WORK_DIRS=(
    ["web_dsj"]="/home/th/jcq/AI_AutoPlat/AI_web_dsj"
    ["yolov5"]="/home/th/jcq/AI_AutoPlat/yolov5-th"
)

# 启动所有程序
for name in "${!PROGRAMS[@]}"; do
    program="${PROGRAMS[$name]}"
    work_dir="${WORK_DIRS[$name]}"
    log_file="$LOG_DIR/${name}_$(date +%Y%m%d_%H%M%S).log"
    
    echo "Starting $name (working dir: $work_dir)..."
    (
        cd "$work_dir" || exit 1
        nohup /home/th/anaconda3/envs/yolov5/bin/python "$(basename "$program")" >> "$log_file" 2>&1 &
        echo "PID: $! | Log: $log_file"
    )
done

echo "All services started in background"
echo "To check running processes: ps aux | grep 'python'"
echo "To view logs: tail -f $LOG_DIR/*.log"