Update app.py

app.py

@@ -1,23 +1,6 @@
 import streamlit as st
-from pytube import YouTube
 import streamlit.components.v1 as components
 
-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>
@@ -34,76 +17,11 @@ confetti_animation_script = """
     function randomInRange(min, max) {
         return Math.random() * (max - min) + min;
     }
-
-    function startButtonClicked() {
-        triggerConfetti();
-    }
 </script>
 """
 
-components.html(confetti_animation_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", on_click=YouTubeDownloader.start_button_clicked)
-
-        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("")
-
-    @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."
-        )
-
-    @classmethod
-    def start_button_clicked(cls):
-        YouTubeDownloader.trigger_confetti()
-
+components.html(confetti_animation_script, height=600)
 
-if __name__ == "__main__":
-    YouTubeDownloader.run()
+if st.button("Click for Confetti"):
+    st.write("Confetti time!")
+    st.write("You can customize this behavior further as needed.")