#!/bin/bash

# 虚拟环境路径 - 请确保与启动脚本中的路径一致
VENV_PATH="$(pwd)/venv"

# 激活虚拟环境（可选，但为了保持一致性）
echo "正在激活Python虚拟环境..."
source "$VENV_PATH/bin/activate"

echo "正在停止所有FLUX风格塑形API服务实例..."

# 查找并停止所有运行中的flux_style_shaper_api.py进程
pids=$(pgrep -f "python3 flux_style_shaper_api/flux_style_shaper_api.py")

if [ -z "$pids" ]; then
  echo "未找到正在运行的FLUX API服务实例"
else
  echo "找到以下进程ID: $pids"
  echo "正在停止服务..."
  
  # 停止所有找到的进程
  for pid in $pids; do
    kill $pid
    echo "已发送停止信号给进程 $pid"
  done
  
  # 等待几秒确认进程已停止
  sleep 3
  
  # 检查是否还有进程在运行
  remaining=$(pgrep -f "python3 flux_style_shaper_api/flux_style_shaper_api.py")
  if [ -z "$remaining" ]; then
    echo "所有服务实例已成功停止"
  else
    echo "以下进程仍在运行: $remaining"
    echo "尝试强制停止..."
    for pid in $remaining; do
      kill -9 $pid
      echo "已强制停止进程 $pid"
    done
  fi
fi

# 退出虚拟环境
deactivate

echo "停止操作完成"