so if the UK government had wanted to be kind to its citizens, they definitely COULD have retained free roaming too.

UK has no say in EU regulations any longer and cannot mandate EU network providers to give access to anyone. The only thing the UK government could mandate is to allow free roaming of EU mobile customers within its borders.