Update app.py

app.py

@@ -1,52 +1,56 @@
 import streamlit as st
 from pytube import YouTube
-import validators
-from tqdm import tqdm
-import os
 
 class YouTubeDownloader:
-    def __init__(self):
-        st.set_page_config(page_title="YouTube Video Downloader", layout="wide")
-        self.output_dir = "downloads"
-        os.makedirs(self.output_dir, exist_ok=True)
-
-    def run(self):
-        st.title("YouTube Video Downloader")
+    @staticmethod
+    def run():
+        st.header("YouTube Video Downloader")
         url = st.text_input("Enter YouTube URL to download:")
         if url:
-            if self.validate_url(url):
-                video, file_size = self.download_video(url)
-                if video:
-                    self.display_video(video)
-                    self.cleanup(video)
-                else:
-                    st.error("Download failed. Please try again.")
-
-    def validate_url(self, url):
-        if not validators.url(url):
-            st.error("Invalid URL. Please enter a valid YouTube URL.")
-            return False
-        return True
+            YouTubeDownloader.validate_url(url)
+            with st.expander("Preview video"):
+                st.video(url)
+            if st.button("Download"):
+                YouTubeDownloader.cleanup()
+                file_ = YouTubeDownloader.download_video(url)
+                st.video(file_)
+                YouTubeDownloader.helper_message()
+
+    @staticmethod
+    def download_video(url):
+        with st.spinner("Downloading..."):
+            local_file = (
+                YouTube(url)
+                .streams.filter(progressive=True, file_extension="mp4")
+                .first()
+                .download()
+            )
+            st.success("Downloaded")
+        return local_file
+
+    @staticmethod
+    def validate_url(url):
+        import validators
 
-    def download_video(self, url):
-        st.info("Downloading... Please wait.")
-        try:
-            yt = YouTube(url)
-            stream = yt.streams.get_highest_resolution()
-            video = stream.download(output_path=self.output_dir, filename="video")
-            st.success("Downloaded successfully!")
+        if not validators.url(url):
+            st.error("Hi there 👋 URL seems invalid 👽")
+            st.stop()
 
-            return video, os.path.getsize(video)
-        except Exception as e:
-            st.error(f"An error occurred: {str(e)}")
-            return None, 0
+    @classmethod
+    def cleanup(cls):
+        import pathlib
+        import glob
 
-    def display_video(self, video_path):
-        st.video(video_path)
+        junks = glob.glob("*.mp4")
+        for junk in junks:
+            pathlib.Path(junk).unlink()
 
-    def cleanup(self, video_path):
-        os.remove(video_path)
+    @classmethod
+    def helper_message(cls):
+        st.write(
+            "> To save the video to the local computer, "
+            "click the vertical ... icon (aka hamburger button) in the bottom-right corner (in the video above) and click download."
+        )
 
 if __name__ == "__main__":
-    downloader = YouTubeDownloader()
-    downloader.run()
+    YouTubeDownloader.run()