Simple reboot window or prompt?

[That-Annoying-Guy]

have you tried Show-InstallationRestartPrompt?

FYI: There are a few caveats that people fall in all the time:

Help with EXE and shut down commands - The Toolkit / General Discussion - PSAppDeployToolkit Community