#!/bin/bash

LOG_DIR="/home/ubuntu/log"
mkdir -p "$LOG_DIR"

echo "[$(date '+%Y-%m-%d %H:%M:%S')] 开始停止旧进程..."

# 杀掉已有进程
pkill -f "main.py" 2>/dev/null && echo "  已停止 ComfyUI" || echo "  ComfyUI 未在运行"
pkill -f "change_cloth_server/change2/service2.py" 2>/dev/null && echo "  已停止 service2" || echo "  service2 未在运行"
pkill -f "change_cloth_server/change3/service3.py" 2>/dev/null && echo "  已停止 service3" || echo "  service3 未在运行"
pkill -f "change_cloth_server/change_app.py" 2>/dev/null && echo "  已停止 change_app" || echo "  change_app 未在运行"

# 等待进程退出
sleep 2

echo "[$(date '+%Y-%m-%d %H:%M:%S')] 开始启动服务..."

# 启动 ComfyUI（使用项目 venv）
COMFYUI_PYTHON="/home/ubuntu/ComfyUI/.venv/bin/python"
if [ -x "$COMFYUI_PYTHON" ]; then
  cd /home/ubuntu/ComfyUI
  nohup "$COMFYUI_PYTHON" main.py >> "$LOG_DIR/comfyui.log" 2>&1 &
  echo "  ComfyUI 已启动 (PID: $!)"
else
  cd /home/ubuntu/ComfyUI
  nohup python3 main.py >> "$LOG_DIR/comfyui.log" 2>&1 &
  echo "  ComfyUI 已启动 (PID: $!)"
fi

# 启动 service2 / service3 / change_app（使用 python3，依赖已 pip install --user）
export PATH="/home/ubuntu/.local/bin:$PATH"
cd /home/ubuntu/change_cloth_server/change2
nohup python3 service2.py >> "$LOG_DIR/service2.log" 2>&1 &
echo "  service2 已启动 (PID: $!)"

cd /home/ubuntu/change_cloth_server/change3
nohup python3 service3.py >> "$LOG_DIR/service3.log" 2>&1 &
echo "  service3 已启动 (PID: $!)"

cd /home/ubuntu/change_cloth_server
nohup python3 change_app.py >> "$LOG_DIR/change_app.log" 2>&1 &
echo "  change_app 已启动 (PID: $!)"

echo "[$(date '+%Y-%m-%d %H:%M:%S')] 所有服务启动完毕，日志目录: $LOG_DIR"