#!/usr/bin/env bash
  # =============================================================================
  # OmniShopAgent - 服务健康检查脚本

  # 检查 Streamlit、Search API 等依赖

  # =============================================================================
  set -euo pipefail
  
  SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
  PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
  RED='\033[0;31m'
  GREEN='\033[0;32m'
  YELLOW='\033[1;33m'
  NC='\033[0m'
  
  echo "=========================================="
  echo "OmniShopAgent 服务健康检查"
  echo "=========================================="
  
  # 1. Python 环境
  echo -n "[Python] "
  if command -v python3 &>/dev/null; then
      VER=$(python3 -c 'import sys; v=sys.version_info; print(f"{v.major}.{v.minor}.{v.micro}")' 2>/dev/null)
      if [[ "$VER" == "3.1"* ]] || [[ "$VER" == "3.12"* ]]; then
          echo -e "${GREEN}OK${NC} $VER"
      else
          echo -e "${YELLOW}WARN${NC} $VER (建议 3.12+)"
      fi
  else
      echo -e "${RED}FAIL${NC} 未找到"
  fi
  
  # 2. 虚拟环境
  echo -n "[Virtualenv] "
  if [ -d "$PROJECT_ROOT/venv" ]; then
      echo -e "${GREEN}OK${NC} $PROJECT_ROOT/venv"
  else
      echo -e "${YELLOW}WARN${NC} 未找到 venv"
  fi
  
  # 3. .env 配置
  echo -n "[.env] "
  if [ -f "$PROJECT_ROOT/.env" ]; then
      if grep -q "OPENAI_API_KEY=sk-" "$PROJECT_ROOT/.env" 2>/dev/null; then
          echo -e "${GREEN}OK${NC} 已配置"
      else
          echo -e "${YELLOW}WARN${NC} 请配置 OPENAI_API_KEY"
      fi
  else
      echo -e "${RED}FAIL${NC} 未找到"
  fi
  

  # 4. 数据目录（可选，用于图片上传）

  echo -n "[数据] "
  if [ -d "$PROJECT_ROOT/data/images" ] && [ -f "$PROJECT_ROOT/data/styles.csv" ]; then
      IMG_COUNT=$(find "$PROJECT_ROOT/data/images" -name "*.jpg" 2>/dev/null | wc -l)
      echo -e "${GREEN}OK${NC} $IMG_COUNT 张图片"
  else

      echo -e "${YELLOW}WARN${NC} 未找到 data/images 或 data/styles.csv (可选，用于图片风格分析)"

  fi
  

  # 5. Streamlit

  echo -n "[Streamlit] "
  if pgrep -f "streamlit run app.py" >/dev/null 2>&1; then
      echo -e "${GREEN}OK${NC} 运行中"
  else
      echo -e "${YELLOW}WARN${NC} 未运行 (./scripts/start.sh)"
  fi
  
  echo "=========================================="