Update app.py

app.py

@@ -3,10 +3,10 @@ from pytube import YouTube
 import os
 
 # Function to download YouTube video
-def download_video(url):
+def download_video(url, title):
     try:
         yt = YouTube(url)
-        st.write("Video Title:", yt.title)
+        st.write("Video Title:", title)
         st.write("Downloading...")
         stream = yt.streams.filter(progressive=True, file_extension='mp4').first()
         stream.download(output_path=os.path.expanduser("~\Downloads"))
@@ -23,4 +23,12 @@ if st.button("Download"):
     if url.strip() == "":
         st.warning("Please enter a YouTube video URL.")
     else:
-        download_video(url)
+        yt = YouTube(url)
+        title = yt.title
+        download_video(url, title)
+
+if 'url' in st.session_state and st.session_state.url.strip() != "":
+    yt = YouTube(st.session_state.url)
+    title = yt.title
+    if st.button("Download Video"):
+        download_video(st.session_state.url, title)