Update app.py

app.py

@@ -3,13 +3,13 @@ from pytube import YouTube
 import os
 
 # Function to download YouTube video
-def download_video(url, title):
+def download_video(url, selected_quality):
     try:
         yt = YouTube(url)
-        st.write("Video Title:", title)
+        st.write("Video Title:", yt.title)
         st.write("Downloading...")
-        stream = yt.streams.filter(progressive=True, file_extension='mp4').first()
-        download_path = os.path.join(os.path.expanduser("~\Downloads"), f"{title}.mp4")
+        stream = yt.streams.filter(progressive=True, file_extension='mp4').get_by_resolution(selected_quality)
+        download_path = os.path.join(os.path.expanduser("~\Downloads"))
         stream.download(output_path=download_path)
         st.success("Download completed successfully!")
     except Exception as e:
@@ -19,17 +19,12 @@ def download_video(url, title):
 st.title("YouTube Video Downloader")
 
 url = st.text_input("Enter YouTube Video URL:", "")
+quality_options = ["1080p", "720p", "480p", "360p"]
+
+selected_quality = st.selectbox("Select Video Quality:", quality_options)
 
 if st.button("Download"):
     if url.strip() == "":
         st.warning("Please enter a YouTube video URL.")
     else:
-        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)
+        download_video(url, selected_quality)