Your end-users will typically login to their training via your branded tenant - https://YOURMSP.safeaihabits.com
If however they visit the public website and click the login button - they'll be taken to the generic https://app.safeaihabits.com/login
This login will accept their credentials and redirect them to your tenant to continue as normal.