Upload app.py

app.py

@@ -0,0 +1,93 @@
+import streamlit as st
+from pytube import YouTube
+
+class YouTubeDownloader:
+    @staticmethod
+    def run():
+        st.markdown(
+            """
+            <style>
+                body {
+                    margin: 0;
+                    overflow: hidden;
+                    background: linear-gradient(to right, #ffcccb, #b5e7a0);
+                    animation: gradientAnimation 10s infinite alternate;
+                    display: flex;
+                    justify-content: center;
+                    align-items: center;
+                    height: 100vh;
+                }
+
+                @keyframes gradientAnimation {
+                    0% {
+                        background-position: 0% 50%;
+                    }
+                    100% {
+                        background-position: 100% 50%;
+                    }
+                }
+
+                .container {
+                    background-color: rgba(255, 255, 255, 0.8);
+                    padding: 20px;
+                    border-radius: 10px;
+                    text-align: center;
+                }
+            </style>
+            """
+        )
+
+        st.write('<div class="container">')
+        st.header("YouTube Video Downloader")
+        url = st.text_input("Enter YouTube URL to download:")
+        if url:
+            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()
+        st.markdown("YouTube Video Download help")
+        st.write("</div>")
+
+    @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
+
+        if not validators.url(url):
+            st.error("Hi there 👋 URL seems invalid 👽")
+            st.stop()
+
+    @classmethod
+    def cleanup(cls):
+        import pathlib
+        import glob
+
+        junks = glob.glob("*.mp4")
+        for junk in junks:
+            pathlib.Path(junk).unlink()
+
+    @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__":
+    YouTubeDownloader.run()