Feature Enhancement for YTDL

[DeadMan30]

Hello,

I have been working on a feature that allows users to select multiple video qualities. I’ve written a separate piece of code that prompts the user for their input URL and presents options for selecting the desired quality via buttons. Once the user makes a selection, yt-dlp will handle the download accordingly.

I’ve tested this feature on my local machine, and it works smoothly—I can select the video format, and it downloads in that format as expected.

However, I’m experiencing some challenges integrating this feature into your existing code. I would appreciate your guidance on how to adapt my implementation to fit within the Telegram-Leecher-Bot. Alternatively, if you prefer, you can use my code to update the bot yourself.

Updated Features:

• Users can select different video qualities.
• Videos download in the user's selected format with the best available audio quality.

Could you please review this feature and let me know how we can proceed with the integration?

[DeadMan30]

import yt_dlp
from pyrogram import Client, filters
from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup

API_ID = YOUR_API_KEY
API_HASH = "YOUR_API_HASH"
BOT_TOKEN = "YOUR_BOT_TOKEN"

app = Client("my_bot", api_id=API_ID, api_hash=API_HASH, bot_token=BOT_TOKEN)

user_links = {}

def get_formats(link):
    with yt_dlp.YoutubeDL() as ydl:
        info = ydl.extract_info(link, download=False)
        return info["formats"]
.
def create_format_buttons(formats):
    buttons = []

    def bytes_to_human_readable(size_in_bytes):
        if isinstance(size_in_bytes, int) and size_in_bytes > 0:
            if size_in_bytes >= (1024 ** 3):
                return f"{size_in_bytes / (1024 ** 3):.2f} GB"
            else:
                return f"{size_in_bytes / (1024 ** 2):.2f} MB"
        return "Unknown"

    for fmt in formats:
        format_note = fmt.get('format_note', 'N/A')
        file_size = fmt.get('filesize', fmt.get('filesize_approx', 'Unknown'))

        file_size = bytes_to_human_readable(file_size)

        if (
            fmt['ext'] not in ['mhtml', 'unknown', 'm4a', 'opus', 'aac']
            and '-drc' not in fmt['format_id']
            and format_note not in ['ultralow', 'low', 'medium', 'N/A', 'Default']
        ):
            button_text = f"{format_note} | {fmt['ext']} | {file_size}"
            buttons.append(InlineKeyboardButton(text=button_text, callback_data=fmt['format_id']))

    keyboard = InlineKeyboardMarkup([buttons[i:i + 2] for i in range(0, len(buttons), 2)])
    return keyboard

@app.on_message(filters.text & filters.private)
async def send_format_buttons(client, message):
    link = message.text
    user_links[message.from_user.id] = link
    formats = get_formats(link)
    keyboard = create_format_buttons(formats)

    await message.reply("Choose Video Format:", reply_markup=keyboard)
    
@app.on_callback_query()
async def callback_handler(client, callback_query):
    selected_format = callback_query.data
    await callback_query.message.delete()

    link = user_links.get(callback_query.from_user.id)
    
    if link:
        await download_video(link, selected_format)

async def download_video(link, format_id):
    ydl_opts = {
        'format': f'{format_id}+bestaudio/best',
        'outtmpl': '%(uploader)s - %(title)s (%(height)sp).%(ext)s',
        'allow_playlist_files': True,
        'writethumbnail': True,
        'concurrent_fragments': 4,
        'overwrites': True,
        'writesubtitles': 'srt',
        'extractor_args': {
            'subtitlesformat': 'srt'
        }
    }

    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
        ydl.download([link])

app.run()