Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import yt_dlp
+import os
 
 def get_youtube_formats(url):
     ydl_opts = {
@@ -12,10 +13,12 @@ def get_youtube_formats(url):
 
 def download_video(url, format_id=None):
     ydl_opts = {
-        'format': format_id if format_id else 'best'
+        'format': format_id if format_id else 'best',
+        'outtmpl': 'downloaded_video.%(ext)s'
     }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([url])
+    return 'downloaded_video.mp4'  # Assuming MP4 format for simplicity
 
 st.title("Video Downloader")
 
@@ -25,18 +28,34 @@ if url:
     if "youtube.com" in url or "youtu.be" in url:
         st.write("Fetching available formats from YouTube...")
         formats = get_youtube_formats(url)
-        format_options = [f"{f['format_id']} - {f.get('format_note', 'No note')} - {f['ext']}" for f in formats]
+        format_options = [f"{f['format_id']} - {f.get('format_note', 'No note')} - {f['ext']} - {f['filesize'] or 'Unknown size'}" for f in formats]
         
         format_choice = st.selectbox("Select video quality:", format_options)
         selected_format_id = format_choice.split(' ')[0]
         
         if st.button("Download"):
             st.write("Downloading...")
-            download_video(url, format_id=selected_format_id)
+            downloaded_file = download_video(url, format_id=selected_format_id)
             st.success("Download complete!")
+            
+            with open(downloaded_file, 'rb') as file:
+                btn = st.download_button(
+                    label="Download Video",
+                    data=file,
+                    file_name=downloaded_file,
+                    mime='video/mp4'
+                )
     else:
         st.write("Downloading in best available quality...")
         if st.button("Download"):
             st.write("Downloading...")
-            download_video(url)
+            downloaded_file = download_video(url)
             st.success("Download complete!")
+            
+            with open(downloaded_file, 'rb') as file:
+                btn = st.download_button(
+                    label="Download Video",
+                    data=file,
+                    file_name=downloaded_file,
+                    mime='video/mp4'
+                )