Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
+import shutil
 import streamlit as st
-from pySmartDL import SmartDL
+from pytube import YouTube
 import platform
 
 def get_default_download_folder():
@@ -18,15 +19,15 @@ def get_default_download_folder():
 
 def download_video(url):
     try:
-        # Create a SmartDL object with the video URL
-        dl = SmartDL(url, progress_bar=False)
-        # Start the download
-        dl.start()
-        # Get the destination path where the file was downloaded
-        dest_path = dl.get_dest()
+        yt = YouTube(url)
+        streams = yt.streams.filter(progressive=True, file_extension="mp4")
+        highest_res_stream = streams.get_highest_resolution()
+        # Download the video
+        save_path = os.path.join(os.getcwd(), yt.title + '.mp4')
+        highest_res_stream.download(output_path=os.getcwd(), filename=yt.title)
         # Move the downloaded file to the default download folder
         default_download_folder = get_default_download_folder()
-        os.rename(dest_path, os.path.join(default_download_folder, os.path.basename(dest_path)))
+        shutil.move(save_path, os.path.join(default_download_folder, yt.title + '.mp4'))
         st.success("Video downloaded successfully!")
     except Exception as e:
         st.error(f"An error occurred: {e}")
@@ -42,7 +43,7 @@ def main():
         if video_url.strip() == "":
             st.warning("Please enter a valid YouTube URL.")
         else:
-            # Download the video to the default download folder
+            # Download the video and move it to the default download folder
             download_video(video_url)
 
 if __name__ == "__main__":