from __future__ import unicode_literals
import youtube_dl
from wbb.utils import cust_filter
from pyrogram import filters
from wbb import app, Command
import os
import glob

ydl_opts = {
    'format': 'bestaudio/best',
    'postprocessors': [{
        'key': 'FFmpegExtractAudio',
        'preferredcodec': 'mp3',
        'preferredquality': '192'
    }],
    'postprocessor_args': [
        '-ar', '16000'
    ],
    'prefer_ffmpeg': True,
    'keepvideo': False
}

@app.on_message(cust_filter.command(commands=(["music"])))
async def commit(client, message):
    user_id = message.from_user.id
    user_name = message.from_user.username
    app.set_parse_mode("markdown")
    await message.reply_chat_action("upload_audio")
    link = (message.text.split(None, 1)[1])
    with youtube_dl.YoutubeDL(ydl_opts) as ydl:
        ydl.download([link])
    for filename in glob.glob("*.mp3"):
        m = await message.reply_audio(filename)
        await m.edit_caption(f'''
[Source]({link})''')
        os.remove(filename)