Customize your Screen On per-app. When your selected apps gain focus, the Smart Screen On will activate, and when they go in the background, it will deactivate.
You can also choose the Always On so your screen will never lock, no mater what.
Completely free with no in-app purchases or annoying ads.