verify_setup.py

#!/usr/bin/env python3
"""Quick setup verification script."""

import subprocess
import sys
import os
from pathlib import Path

def check_virtual_env():
    """Check if virtual environment exists and has correct packages."""
    venv_path = Path("venv")
    if not venv_path.exists():
        print("❌ Virtual environment not found")
        return False
    
    python_path = venv_path / "bin" / "python"
    if not python_path.exists():
        print("❌ Python executable not found in virtual environment")
        return False
    
    # Test import
    try:
        result = subprocess.run([
            str(python_path), "-c", "import ai_arch; import fastapi; import uvicorn"
        ], capture_output=True, text=True)
        
        if result.returncode == 0:
            print("✅ Virtual environment setup correctly")
            return True
        else:
            print(f"❌ Import test failed: {result.stderr}")
            return False
    except Exception as e:
        print(f"❌ Error testing virtual environment: {e}")
        return False

def check_makefile_commands():
    """Check if Makefile commands work."""
    try:
        # Test version command
        result = subprocess.run(["make", "version"], capture_output=True, text=True)
        if result.returncode == 0 and "AI Arch System v0.1.0" in result.stdout:
            print("✅ Make commands use virtual environment correctly")
            return True
        else:
            print(f"❌ Make version command failed: {result.stderr}")
            return False
    except Exception as e:
        print(f"❌ Error testing make commands: {e}")
        return False

def check_port_configuration():
    """Check if port is correctly configured to 6545."""
    files_to_check = [
        ("Makefile", "6545"),
        ("ai_arch/main.py", "6545"),
        ("demo.py", "6545"),
        ("README.md", "6545")
    ]
    
    all_good = True
    for file, expected_port in files_to_check:
        if Path(file).exists():
            content = Path(file).read_text()
            if expected_port in content:
                print(f"✅ Port {expected_port} configured in {file}")
            else:
                print(f"❌ Port {expected_port} not found in {file}")
                all_good = False
        else:
            print(f"⚠️  File {file} not found")
    
    return all_good

def main():
    """Main verification function."""
    print("🔍 AI Architecture System - Setup Verification")
    print("=" * 50)
    
    checks = [
        ("Virtual Environment", check_virtual_env),
        ("Makefile Commands", check_makefile_commands), 
        ("Port Configuration", check_port_configuration),
    ]
    
    all_passed = True
    for check_name, check_func in checks:
        print(f"\n📋 Checking {check_name}...")
        if not check_func():
            all_passed = False
    
    print("\n" + "=" * 50)
    if all_passed:
        print("🎉 All checks passed! System is ready to use.")
        print("\n📖 Quick Start:")
        print("   make dev     # Start development server on port 6545")
        print("   make help    # See all available commands")
        print("   python3 demo.py  # Run the demo")
        return 0
    else:
        print("❌ Some checks failed. Please review the issues above.")
        return 1

if __name__ == "__main__":
    sys.exit(main())