File size: 2,306 Bytes
b789b2e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
import streamlit as st
from pytube import YouTube
class YouTubeDownloader:
@staticmethod
def run():
st.header("YouTube Video Downloader")
url = st.text_input("Enter YouTube URL to download:")
if url:
YouTubeDownloader.validate_url(url)
with st.expander("Preview Video"):
st.video(url)
if st.button("Download Video"):
YouTubeDownloader.cleanup()
file_path = YouTubeDownloader.download_video(url)
st.success("Downloaded")
st.download_button(
label="Download Here",
key="download_button",
on_click=YouTubeDownloader.download_button_callback,
args=(file_path,),
)
YouTubeDownloader.helper_message()
st.markdown("YouTube Video Download help")
@staticmethod
def download_video(url):
with st.spinner("Downloading..."):
local_file = (
YouTube(url)
.streams.filter(progressive=True, file_extension="mp4")
.first()
.download()
)
return local_file
@staticmethod
def download_button_callback(file_path):
with open(file_path, "rb") as file:
file_data = file.read()
st.download_button(
label="Click to Download",
data=file_data,
key="file_download",
file_name="downloaded_video.mp4",
mime="video/mp4",
)
@staticmethod
def validate_url(url):
import validators
if not validators.url(url):
st.error("Hi there 👋 URL seems invalid 👽")
st.stop()
@classmethod
def cleanup(cls):
import pathlib
import glob
junks = glob.glob("*.mp4")
for junk in junks:
pathlib.Path(junk).unlink()
@classmethod
def helper_message(cls):
st.write(
"> To save the video to local computer, "
"click the vertical ... icon (aka hamburger button) in the bottom-right corner (in the video above) and click download."
)
if __name__ == "__main__":
YouTubeDownloader.run()
|