Using rich progress bar while download video with pytube

[Hetari]

Is there a way to include download progress bars from rich while download video with pytube

This is a normal (simple) code to download video with pytube

from pytube import YouTube
from pytube.cli import on_progress

if __name__ == "__main__":
    url = "https://www.youtube.com/watch?v=dQw4w9WgXcQ"

    print("Downloading video...")
    video = YouTube(url, on_progress_callback=on_progress)

    print("Getting lowest resolution...")
    stream = video.streams.get_lowest_resolution()

    print("Downloading...")
    stream.download()

Now I wonder if I can use the rich progress bar instead. I tried a lot of ways to integrate them but nothing worked. Can anyone replace the on_progress with rich?

This is the on_progress code from pytube itself:
https://github.com/pytube/pytube/blob/a32fff39058a6f7e5e59ecd06a7467b71197ce35/pytube/cli.py#L209-L249