Update app.py

app.py

@@ -34,7 +34,30 @@ import spacy
 
 
 warnings.filterwarnings("ignore")
+def download_youtube(url, choice, res):
+
+  yt = pytube.YouTube(url)
+
+  if choice == 'mp3':
+    audio = yt.streams.filter(only_audio=True).first()
+    print(f"Downloading {audio.title} as MP3")
+    return audio.download()
+
+  elif choice == 'mp4':
+    if res == "720p":
+      video = yt.streams.filter(res="720p").first()
+    elif res == "1080p":  
+      video = yt.streams.filter(res="1080p").first()
+    elif res == "2160p":
+      video = yt.streams.filter(res="2160p").first()
+    else:
+      return "Invalid resolution"
+    
+    print(f"Downloading {video.title} at {video.resolution}")
+    return video.download()
 
+  else:
+    return "Invalid choice"
 def Process_audio(fileName):
   text=''
   txtf=open("The_audio.txt","w+")
@@ -66,7 +89,8 @@ except:
 
 def UrlToAudio(VideoUrl):
   url=VideoUrl
-  os.system("yt-dlp -x --audio-format wav " + url)
+  #os.system("yt-dlp -x --audio-format wav " + url)
+  download_youtube(VideoUrl,"mp3","")
   # load audio and pad/trim it to fit 30 seconds
   base_path = Path(r"")
   for wav_file_path in base_path.glob("*.wav"):