VaslNistim / main.py
ManiAz09's picture
Upload folder using huggingface_hub
04ec20b verified
import os
from video_dl import get_video_formats, download_video
from balethon import Client, conditions
from balethon.objects.message import Message
import json
import re
url = "https://www.youtube.com/watch?v=GVPhBSmMuq8&pp=ugUEEgJlbg%3D%3D"
data = get_video_formats(url=url)
# print(json.dumps(data, indent=2))
# download_video(url=url, format_id="278", output="video.mp4", ffmpeg_location=r"C:\ffmpeg-master-latest-win64-gpl\bin\ffmpeg.exe")
bot = Client(os.environ["BotToken"])
def is_youtube_link(event: str) -> bool:
link = False
if event.text:
# Regular expression to match various YouTube URL formats
youtube_regex = r'^(https?://)?(www\.|m\.)?(youtube\.com|youtu\.be)/'
if re.match(youtube_regex, event.text):
link = True
return link
@bot.on_message(condition=conditions.Condition(is_youtube_link))
async def greet(message: Message):
user_message = message.text
formats = get_video_formats(url=user_message)
output_path = download_video(url=user_message, format_id=formats[-1]["format_id"])
# await message.reply("Hello")
await message.reply_video(output_path)
bot.run()