From 26b9efc08af025870aaf04f93d2a0ebfeede5cde Mon Sep 17 00:00:00 2001
From: Skullheadx <94652084+Skullheadx@users.noreply.github.com>
Date: Wed, 22 Nov 2023 18:38:31 -0500
Subject: [PATCH] Update main.py

---
 main.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 05c1ce4..fa0371b 100644
--- a/main.py
+++ b/main.py
@@ -1,22 +1,31 @@
 from pytube import YouTube
 
-SAVE_PATH = "D:/Youtube/CodingClub/"
+SAVE_PATH = "D:/Youtube/"
 
 with open('links_file.txt', 'r') as f:
     links = f.read().split('\n')
     if links[-1] == "":
         links = links[:-1]
 
+VIDEO = False
 
-def download(link):
+def download_video(link):
     yt = YouTube(link)
     mp4_files = yt.streams.filter(file_extension="mp4")
     mp4_files = mp4_files.get_highest_resolution()
     mp4_files.download(output_path=SAVE_PATH)
     print("Download is completed successfully")
+def download_audio(link):
+    yt = YouTube(link)
+    audio = yt.streams.filter(only_audio=True).get_audio_only()
+    audio.download(output_path=SAVE_PATH)
+    print("Download is completed successfully")
 
 
 for i in links:
-    download(i)
+    if VIDEO:
+        download_video(i)
+    else:
+        download_audio(i)
 
-print('Videos Downloaded!')
+print('Items Downloaded!')
-- 
2.54.0