Can you get an on screen notification that lets you know you have been using your phone for x time?
Where x is the limit you have chosen, I.e. 5 or 10 mins it won’t lock you out but it i Something that can break your focus, maybe an interesting thought to change your focus?