setup.py

"""
setup.py — one-time installer for RL Gridworld App
Run this once before launching rl_gridworld_app.py

    python setup.py
"""

import sys
import subprocess
import importlib
import platform

REQUIRED = {
    "numpy":      "numpy",
    "matplotlib": "matplotlib",
}

def check_python_version():
    major, minor = sys.version_info[:2]
    if (major, minor) < (3, 8):
        print(f"  ✗  Python 3.8+ required (you have {major}.{minor})")
        sys.exit(1)
    print(f"  ✓  Python {major}.{minor} detected")

def check_tkinter():
    try:
        import tkinter  # noqa: F401
        print("  ✓  tkinter is available")
        return True
    except ImportError:
        print("  ✗  tkinter not found")
        os_name = platform.system()
        if os_name == "Linux":
            print("     Fix:  sudo apt install python3-tk   (Debian/Ubuntu)")
            print("           sudo dnf install python3-tkinter  (Fedora/RHEL)")
        elif os_name == "Darwin":
            print("     Fix:  reinstall Python from https://python.org")
            print("           (the official installer ships with tkinter)")
        elif os_name == "Windows":
            print("     Fix:  re-run the Python installer, tick 'tcl/tk and IDLE'")
        return False

def install_packages():
    all_ok = True
    for import_name, pip_name in REQUIRED.items():
        try:
            importlib.import_module(import_name)
            print(f"  ✓  {pip_name} already installed")
        except ImportError:
            print(f"  ↓  Installing {pip_name} …", end="", flush=True)
            result = subprocess.run(
                [sys.executable, "-m", "pip", "install", pip_name],
                capture_output=True, text=True
            )
            if result.returncode == 0:
                print(" done")
            else:
                print(" FAILED")
                print(result.stderr.strip())
                all_ok = False
    return all_ok

def main():
    print("=" * 50)
    print("  RL Gridworld App — Setup")
    print("=" * 50)

    print("\n[1] Checking Python version …")
    check_python_version()

    print("\n[2] Checking tkinter …")
    tk_ok = check_tkinter()

    print("\n[3] Installing Python packages …")
    pkgs_ok = install_packages()

    print("\n" + "=" * 50)
    if pkgs_ok and tk_ok:
        print("  All dependencies ready.")
        print("  Launch the app with:")
        print()
        print("      python rl_gridworld_app.py")
    else:
        print("  Some issues were found — see messages above.")
        print("  Fix them, then re-run this script.")
    print("=" * 50)

if __name__ == "__main__":
    main()