Screen should go to sleep while playing audio

[usman-cas160]

If I play the audio on Telegram X orTtelegram and turn off the screen or go to home page, the audio keeps playing.

If I open Telegram X group, and leave the device then it goes to sleep, same is the case with Telegram.

if I open Telegram X group, and play the audio file then the screen does not go to sleep automaticlly. However, on Telegram the screen goes to sleep while hearing the audio, and once its on sleep the audio keeps playing. This is the real issue that screen should go to sleep while playing audio.