|
|
import streamlit as st |
|
|
from pytube import YouTube |
|
|
import streamlit.components.v1 as components |
|
|
|
|
|
components.html("<html><body><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><div class="m"></div></body></html>") |
|
|
|
|
|
|
|
|
|
|
|
animation_symbol = "❄" |
|
|
|
|
|
st.markdown( |
|
|
f""" |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
<div class="snowflake">{animation_symbol}</div> |
|
|
""", |
|
|
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() |
|
|
|
