360

from telegram import Update
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackContext
from pytube import YouTube
import os

# Function to start the bot
def start(update: Update, context: CallbackContext) -> None:
    update.message.reply_text('Hello! Send me a YouTube link, and I will download the video for you.')

# Function to handle messages
def handle_message(update: Update, context: CallbackContext) -> None:
    url = update.message.text
    try:
        yt = YouTube(url)
        stream = yt.streams.get_highest_resolution()
        output_path = 'downloads/'
        if not os.path.exists(output_path):
            os.makedirs(output_path)
        stream.download(output_path=output_path, filename=f'{yt.title}.mp4')
        update.message.reply_text(f'Download completed: {yt.title}.mp4')
    except Exception as e:
        update.message.reply_text(f'Error: {str(e)}')

# Main function to run the bot
def main() -> None:
    # Replace 'YOUR_TOKEN_HERE' with your actual bot token
    updater = Updater("YOUR_TOKEN_HERE")

    dispatcher = updater.dispatcher

    dispatcher.add_handler(CommandHandler("start", start))
    dispatcher.add_handler(MessageHandler(Filters.text & ~Filters.command, handle_message))

    updater.start_polling()
    updater.idle()

if __name__ == '__main__':
    main()