Spaces:
Sleeping
Sleeping
File size: 1,691 Bytes
9430c4d 236e101 6de6818 25ead2e a6b0cbe 25ead2e 187461d 9430c4d 236e101 187461d 6de6818 187461d 25ead2e 6de6818 25ead2e 4185844 9430c4d 25ead2e 9430c4d 6de6818 187461d 9430c4d 6de6818 25ead2e |
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 |
import streamlit as st
from pytube import YouTube
import os
import base64
# Function to download YouTube video and return the path of the downloaded file
def download_video(url, selected_quality):
try:
yt = YouTube(url)
st.write("Video Title:", yt.title)
st.write("Downloading...")
stream = yt.streams.filter(progressive=True, file_extension='mp4').get_by_resolution(selected_quality)
download_path = os.path.join(os.path.expanduser("~"), "Downloads")
downloaded_file_path = stream.download(output_path=download_path)
st.success("Download completed successfully!")
return downloaded_file_path
except Exception as e:
st.error(f"An error occurred: {e}")
return None
# Function to create a download link
def create_download_link(file_path):
file_name = os.path.basename(file_path)
with open(file_path, "rb") as f:
bytes = f.read()
b64 = base64.b64encode(bytes).decode()
href = f'<a href="data:file/mp4;base64,{b64}" download="{file_name}">Download {file_name}</a>'
return href
# Streamlit UI
st.title("YouTube Video Downloader")
url = st.text_input("Enter YouTube Video URL:", "")
quality_options = ["1080p", "720p", "480p", "360p"]
selected_quality = st.selectbox("Select Video Quality:", quality_options)
if st.button("Download"):
if url.strip() == "":
st.warning("Please enter a YouTube video URL.")
else:
downloaded_file_path = download_video(url, selected_quality)
if downloaded_file_path:
download_link = create_download_link(downloaded_file_path)
st.markdown(download_link, unsafe_allow_html=True)
|