Create app.py

app.py

@@ -0,0 +1,99 @@
+import streamlit as st
+from pytube import YouTube
+
+hide_streamlit_style = """
+<style>
+#MainMenu {visibility: hidden;}
+footer {visibility: hidden;}
+.stApp{
+background-image: linear-gradient(115deg,#FFAF00,#FFC500,#FFD600,#FCED00,#F9F380,#F6F7CD);
+  animation: rotate 7s linear infinite;
+}
+@keyframes rotate {
+  100%{
+    filter: hue-rotate(-360deg)
+  }
+}
+</style>
+"""
+
+# JavaScript code for confetti animation
+confetti_animation_script = """
+<script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/confetti.browser.min.js"></script>
+<script>
+    function triggerConfetti() {
+        confetti({
+            angle: randomInRange(55, 125),
+            particleCount: randomInRange(50, 100),
+            origin: { y: 0.6 },
+            spread: randomInRange(50, 70),
+        });
+    }
+
+    function randomInRange(min, max) {
+        return Math.random() * (max - min) + min;
+    }
+</script>
+"""
+
+st.markdown(hide_streamlit_style, unsafe_allow_html=True)
+
+class YouTubeDownloader:
+    @staticmethod
+    def run():
+        st.header("")
+        url = st.text_input("Enter Here")
+        start_button = st.button("Start")
+
+        if start_button:
+            if url:
+                YouTubeDownloader.validate_url(url)
+                with st.expander("preview video"):
+                    st.video(url)
+                YouTubeDownloader.cleanup()
+                file_ = YouTubeDownloader.download_video(url)
+                st.video(file_)
+                YouTubeDownloader.helper_message()
+                st.markdown(confetti_animation_script, unsafe_allow_html=True)  # Include confetti script
+
+        st.markdown("")
+
+    @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  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()