Just manual settings I could use to say "always open this site in tab group X" or "prompt to move this site to tab groups Y or Z if I open it somewhere else (but with the option not to)" would probably make it usable-enough for me.
The tab groups extension for Chrome works well for automating moving tabs to the correct group. https://chrome.google.com/webstore/detail/tab-groups-extensi...
I try and be disciplined but like you end up having to manually stop myself and drag tabs back to the ‘proper’ group.