Create app.py

app.py

@@ -0,0 +1,51 @@
+import streamlit as st
+from pytube import YouTube
+import validators
+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")
+        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
+
+    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!")
+
+            return video, os.path.getsize(video)
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
+            return None, 0
+
+    def display_video(self, video_path):
+        st.video(video_path)
+
+    def cleanup(self, video_path):
+        os.remove(video_path)
+
+if __name__ == "__main__":
+    downloader = YouTubeDownloader()
+    downloader.run()