Disable screen turn-off during phone call
Proximity sensors allow the screen to turn off when close to the ear during a call. It should automatically come back when it's uncovered i.e. when it's not close to the ear.

That said this app seems to promise that functionality -> Smart Screen On Off