Update app.py

app.py

@@ -34,9 +34,15 @@ 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:
@@ -44,7 +50,7 @@ class YouTubeDownloader:
     def run():
         st.header("")
         url = st.text_input("Enter Here")
-        start_button = st.button("Start")
+        start_button = st.button("Start", on_click=YouTubeDownloader.start_button_clicked)
 
         if start_button:
             if url:
@@ -56,9 +62,6 @@ class YouTubeDownloader:
                 st.video(file_)
                 YouTubeDownloader.helper_message()
 
-                # Use components.html to inject the HTML directly
-                components.html(confetti_animation_script, height=0)
-
         st.markdown("")
 
     @staticmethod
@@ -78,7 +81,7 @@ class YouTubeDownloader:
         import validators
 
         if not validators.url(url):
-            st.error("Hi there 👋 URL seems invalid 👽")
+            st.error("Hi there 👋 URL seems invalid 👽")
             st.stop()
 
     @classmethod
@@ -97,6 +100,10 @@ class YouTubeDownloader:
             "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()
+
 
 if __name__ == "__main__":
     YouTubeDownloader.run()