Update app.py

add download button

app.py

@@ -1,30 +1,44 @@
 import streamlit as st
 from pytube import YouTube
 import os
+import base64
 
-# Function to download YouTube video
+# Function to download YouTube video and return the path of the downloaded file
 def download_video(url, selected_quality):
     try:
         yt = YouTube(url)
         st.write("Video Title:", yt.title)
         st.write("Downloading...")
         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)
+        download_path = os.path.join(os.path.expanduser("~"), "Downloads")
+        downloaded_file_path = stream.download(output_path=download_path)
         st.success("Download completed successfully!")
+        return downloaded_file_path
     except Exception as e:
         st.error(f"An error occurred: {e}")
+        return None
+
+# Function to create a download link
+def create_download_link(file_path):
+    file_name = os.path.basename(file_path)
+    with open(file_path, "rb") as f:
+        bytes = f.read()
+        b64 = base64.b64encode(bytes).decode()
+        href = f'<a href="data:file/mp4;base64,{b64}" download="{file_name}">Download {file_name}</a>'
+        return href
 
 # Streamlit UI
 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:
-        download_video(url, selected_quality)
+        downloaded_file_path = download_video(url, selected_quality)
+        if downloaded_file_path:
+            download_link = create_download_link(downloaded_file_path)
+            st.markdown(download_link, unsafe_allow_html=True)