Spaces:
Runtime error
Runtime error
Commit
·
a25b13a
1
Parent(s):
63a0739
Deleted yt Secondary File
Browse files- yt-downloader.py +0 -94
yt-downloader.py
DELETED
@@ -1,94 +0,0 @@
|
|
1 |
-
import streamlit as st
|
2 |
-
from pytube import Playlist, YouTube
|
3 |
-
import time
|
4 |
-
import os
|
5 |
-
import re
|
6 |
-
|
7 |
-
def sanitize_filename(filename):
|
8 |
-
# Remove invalid characters and replace spaces with underscores
|
9 |
-
return re.sub(r'[\\/*?:"<>|]', "", filename.replace(" ", "_"))
|
10 |
-
|
11 |
-
def download_video(URL, save_folder):
|
12 |
-
try:
|
13 |
-
video = YouTube(URL)
|
14 |
-
video_title = sanitize_filename(video.title)
|
15 |
-
video_filename = f"{video_title}.mp4"
|
16 |
-
file_path = os.path.join(save_folder, video_filename)
|
17 |
-
|
18 |
-
st.write(f"Checking if file {file_path} exists...")
|
19 |
-
if os.path.exists(file_path):
|
20 |
-
st.write("File exists. Skipping download.")
|
21 |
-
return
|
22 |
-
else:
|
23 |
-
st.write("File does not exist. Proceeding with download.")
|
24 |
-
|
25 |
-
st.write(f"\nDownloading your File: {video_title}")
|
26 |
-
|
27 |
-
if not os.path.exists(save_folder):
|
28 |
-
os.makedirs(save_folder)
|
29 |
-
|
30 |
-
st.write("Downloading video...")
|
31 |
-
video.streams.get_highest_resolution().download(output_path=save_folder, filename=video_title)
|
32 |
-
st.write('Hurray! Successfully Downloaded')
|
33 |
-
except Exception as e:
|
34 |
-
st.write("Error occurred while downloading the video:", str(e))
|
35 |
-
|
36 |
-
return True
|
37 |
-
|
38 |
-
|
39 |
-
def download_playlist(playlist_url, save_folder):
|
40 |
-
try:
|
41 |
-
playlist = Playlist(playlist_url)
|
42 |
-
st.write(f'Downloading your Playlist: {playlist.title}')
|
43 |
-
|
44 |
-
start_time = time.time()
|
45 |
-
total_videos = len(playlist.video_urls)
|
46 |
-
downloaded_videos = 0
|
47 |
-
|
48 |
-
for video_url in playlist.video_urls:
|
49 |
-
video = YouTube(video_url)
|
50 |
-
video_title = sanitize_filename(video.title)
|
51 |
-
video_filename = f"{video_title}.mp4"
|
52 |
-
file_path = os.path.join(save_folder, video_filename)
|
53 |
-
|
54 |
-
st.write(f"Checking if file {file_path} exists...")
|
55 |
-
if os.path.exists(file_path):
|
56 |
-
st.write(f"Video '{video_title}' already exists, skipping...")
|
57 |
-
downloaded_videos += 1
|
58 |
-
continue
|
59 |
-
else:
|
60 |
-
st.write("File does not exist. Proceeding with download.")
|
61 |
-
|
62 |
-
st.write(f"Downloading video: {video_title}")
|
63 |
-
video.streams.get_highest_resolution().download(output_path=save_folder, filename=video_title)
|
64 |
-
downloaded_videos += 1
|
65 |
-
st.write(f"Downloaded video {downloaded_videos}/{total_videos}")
|
66 |
-
|
67 |
-
end_time = time.time()
|
68 |
-
|
69 |
-
st.write(f'Hurray! Successfully Downloaded Playlist: {playlist.title}')
|
70 |
-
st.write(f'Total Time Taken: {end_time - start_time:.2f} seconds')
|
71 |
-
except Exception as e:
|
72 |
-
st.write("Error occurred while processing the playlist:", str(e))
|
73 |
-
|
74 |
-
return True
|
75 |
-
|
76 |
-
def main():
|
77 |
-
st.title("YouTube Downloader")
|
78 |
-
download_type = st.selectbox("What do you want to download?", ["Single Video", "Playlist"])
|
79 |
-
|
80 |
-
if download_type == "Single Video":
|
81 |
-
URL = st.text_input("Enter the YouTube video URL")
|
82 |
-
save_folder = st.text_input("Enter the folder path where you want to save the video")
|
83 |
-
if st.button("Download Video"):
|
84 |
-
if download_video(URL, save_folder):
|
85 |
-
st.write("Do you want to download another video?")
|
86 |
-
else:
|
87 |
-
playlist_url = st.text_input("Enter the YouTube playlist URL")
|
88 |
-
save_folder = st.text_input("Enter the folder path where you want to save the videos")
|
89 |
-
if st.button("Download Playlist"):
|
90 |
-
if download_playlist(playlist_url, save_folder):
|
91 |
-
st.write("Do you want to download another playlist?")
|
92 |
-
|
93 |
-
if __name__ == "__main__":
|
94 |
-
main()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|