Spaces:
Runtime error
Runtime error
File size: 3,571 Bytes
63a0739 |
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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
import streamlit as st
from pytube import Playlist, YouTube
import time
import os
import re
def sanitize_filename(filename):
# Remove invalid characters and replace spaces with underscores
return re.sub(r'[\\/*?:"<>|]', "", filename.replace(" ", "_"))
def download_video(URL, save_folder):
try:
video = YouTube(URL)
video_title = sanitize_filename(video.title)
video_filename = f"{video_title}.mp4"
file_path = os.path.join(save_folder, video_filename)
st.write(f"Checking if file {file_path} exists...")
if os.path.exists(file_path):
st.write("File exists. Skipping download.")
return
else:
st.write("File does not exist. Proceeding with download.")
st.write(f"\nDownloading your File: {video_title}")
if not os.path.exists(save_folder):
os.makedirs(save_folder)
st.write("Downloading video...")
video.streams.get_highest_resolution().download(output_path=save_folder, filename=video_title)
st.write('Hurray! Successfully Downloaded')
except Exception as e:
st.write("Error occurred while downloading the video:", str(e))
return True
def download_playlist(playlist_url, save_folder):
try:
playlist = Playlist(playlist_url)
st.write(f'Downloading your Playlist: {playlist.title}')
start_time = time.time()
total_videos = len(playlist.video_urls)
downloaded_videos = 0
for video_url in playlist.video_urls:
video = YouTube(video_url)
video_title = sanitize_filename(video.title)
video_filename = f"{video_title}.mp4"
file_path = os.path.join(save_folder, video_filename)
st.write(f"Checking if file {file_path} exists...")
if os.path.exists(file_path):
st.write(f"Video '{video_title}' already exists, skipping...")
downloaded_videos += 1
continue
else:
st.write("File does not exist. Proceeding with download.")
st.write(f"Downloading video: {video_title}")
video.streams.get_highest_resolution().download(output_path=save_folder, filename=video_title)
downloaded_videos += 1
st.write(f"Downloaded video {downloaded_videos}/{total_videos}")
end_time = time.time()
st.write(f'Hurray! Successfully Downloaded Playlist: {playlist.title}')
st.write(f'Total Time Taken: {end_time - start_time:.2f} seconds')
except Exception as e:
st.write("Error occurred while processing the playlist:", str(e))
return True
def main():
st.title("YouTube Downloader")
download_type = st.selectbox("What do you want to download?", ["Single Video", "Playlist"])
if download_type == "Single Video":
URL = st.text_input("Enter the YouTube video URL")
save_folder = st.text_input("Enter the folder path where you want to save the video")
if st.button("Download Video"):
if download_video(URL, save_folder):
st.write("Do you want to download another video?")
else:
playlist_url = st.text_input("Enter the YouTube playlist URL")
save_folder = st.text_input("Enter the folder path where you want to save the videos")
if st.button("Download Playlist"):
if download_playlist(playlist_url, save_folder):
st.write("Do you want to download another playlist?")
if __name__ == "__main__":
main()
|