Environment Setup

#!/usr/bin/env python3
"""Verify AI development environment setup.
Run: uv run python verify_setup.py
"""
import sys
import shutil

def header(msg: str) -> None:
    print(f"\n{'='*50}")
    print(f"  {msg}")
    print(f"{'='*50}")

def check(label: str, ok: bool, fix: str = "") -> bool:
    status = "OK" if ok else "MISSING"
    color = "\033[92m" if ok else "\033[91m"
    reset = "\033[0m"
    print(f"  {color}[{status}]{reset} {label}")
    if not ok and fix:
        print(f"         Fix: {fix}")
    return ok

def main():
    all_ok = True

    header("1. Python Version")
    v = sys.version_info
    print(f"  Python {v.major}.{v.minor}.{v.micro}")
    all_ok &= check("Python >= 3.10", v >= (3, 10), "uv python install 3.12")

    header("2. Package Manager")
    has_uv = shutil.which("uv") is not None
    check("uv installed", has_uv, "curl -LsSf https://astral.sh/uv/install.sh | sh")

    header("3. Core Packages")
    packages = [
        ("openai", "openai", "uv add openai"),
        ("anthropic", "anthropic", "uv add anthropic"),
        ("transformers", "transformers", "uv add transformers"),
        ("sentence_transformers", "sentence-transformers", "uv add sentence-transformers"),
        ("tiktoken", "tiktoken", "uv add tiktoken"),
        ("torch", "torch", "uv add torch"),
        ("dotenv", "python-dotenv", "uv add python-dotenv"),
        ("numpy", "numpy", "uv add numpy"),
        ("pandas", "pandas", "uv add pandas"),
    ]
    for import_name, display_name, fix_cmd in packages:
        try:
            __import__(import_name)
            check(display_name, True)
        except ImportError:
            all_ok &= check(display_name, False, fix_cmd)

    header("4. GPU Availability")
    try:
        import torch
        if torch.cuda.is_available():
            name = torch.cuda.get_device_name(0)
            vram = torch.cuda.get_device_properties(0).total_mem / 1e9
            check(f"CUDA GPU: {name} ({vram:.0f} GB)", True)
        elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
            check("Apple MPS (Metal)", True)
        else:
            check("GPU", False, "CPU only — fine for API work, see GPU section")
    except ImportError:
        check("GPU check", False, "Install torch first")

    header("5. API Keys (.env)")
    try:
        from dotenv import load_dotenv
        import os
        load_dotenv()
        for key, name in [
            ("OPENAI_API_KEY", "OpenAI"),
            ("ANTHROPIC_API_KEY", "Anthropic"),
            ("HF_TOKEN", "Hugging Face"),
        ]:
            val = os.getenv(key, "")
            if val:
                masked = val[:8] + "..." + val[-4:]
                check(f"{name}: {masked}", True)
            else:
                check(f"{name}", False, f"Add {key}=... to .env file")
    except ImportError:
        check("python-dotenv", False, "uv add python-dotenv")

    header("6. Quick API Test")
    try:
        import os
        from dotenv import load_dotenv
        load_dotenv()
        if os.getenv("OPENAI_API_KEY"):
            from openai import OpenAI
            client = OpenAI()
            r = client.chat.completions.create(
                model="gpt-4o-mini",
                messages=[{"role": "user", "content": "Say 'setup verified' in 2 words"}],
                max_tokens=10,
            )
            check(f"OpenAI API: {r.choices[0].message.content}", True)
        else:
            check("OpenAI API", False, "Set OPENAI_API_KEY to test")
    except Exception as e:
        all_ok &= check(f"OpenAI API: {e}", False)

    header("Result")
    if all_ok:
        print("  All checks passed. You're ready to build.")
    else:
        print("  Some checks failed. Fix the items above before continuing.")
    print()

if __name__ == "__main__":
    main()