|
|
import streamlit as st |
|
|
from pytube import YouTube |
|
|
from streamlit.components.v1 import html |
|
|
|
|
|
html( |
|
|
""" |
|
|
<!DOCTYPE html> |
|
|
<html lang="en"> |
|
|
<head> |
|
|
<meta charset="UTF-8"> |
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0"> |
|
|
<title>Document</title> |
|
|
<style> |
|
|
.m{ |
|
|
background-image: linear-gradient(115deg,#FFAF00,#FFC500,#FFD600,#FCED00,#F9F380,#F6F7CD); |
|
|
animation: rotate 7s linear infinite; |
|
|
|
|
|
} |
|
|
@keyframes rotate { |
|
|
100%{ |
|
|
filter: hue-rotate(-360deg) |
|
|
} |
|
|
} |
|
|
</style> |
|
|
</head> |
|
|
<body> |
|
|
<div class="m"> |
|
|
|
|
|
</div> |
|
|
</body> |
|
|
</html> |
|
|
""" |
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
hide_streamlit_style = """ |
|
|
<style> |
|
|
#MainMenu {visibility: hidden;} |
|
|
footer {visibility: hidden;} |
|
|
|
|
|
|
|
|
</style> |
|
|
""" |
|
|
|
|
|
st.markdown(hide_streamlit_style, unsafe_allow_html=True) |
|
|
|
|
|
class YouTubeDownloader: |
|
|
@staticmethod |
|
|
def run(): |
|
|
st.header("YouTube Video Downloader") |
|
|
url = st.text_input("Enter YouTube URL to download:") |
|
|
start_button = st.button("Start Download") |
|
|
|
|
|
if start_button: |
|
|
if url: |
|
|
YouTubeDownloader.validate_url(url) |
|
|
with st.expander("preview video"): |
|
|
st.video(url) |
|
|
YouTubeDownloader.cleanup() |
|
|
file_ = YouTubeDownloader.download_video(url) |
|
|
st.video(file_) |
|
|
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() |
|
|
) |
|
|
st.success("Downloaded") |
|
|
return local_file |
|
|
|
|
|
@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 the 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() |
|
|
|
