Maybe switching it up could work. What if learning happened at home with the use of AI and "homework" happened in class under supervision?
https://en.m.wikipedia.org/wiki/Flipped_classroom
I wrote my master thesis about that.
It‘s an old idea.