Update app.py

app.py

@@ -1,61 +1,42 @@
 import streamlit as st
-import yt_dlp
 import os
+import yt_dlp
 
-def get_youtube_formats(url):
-    ydl_opts = {
-        'listformats': True,
-    }
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info_dict = ydl.extract_info(url, download=False)
-        formats = info_dict.get('formats', [])
-    return formats
+# Streamlit UI setup
+st.title("YouTube Video Downloader 8K supported")
 
-def download_video(url, format_id=None):
-    ydl_opts = {
-        '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
+# Streamlit input fields
+url = st.text_input("Enter the YouTube URL:")
+quality = st.selectbox("Select the desired quality:", ['4320', '2160', '1440', '1080', '720', '480'])
+save_file_name = st.text_input("Enter the save file name (without extension):")
+
+# Convert quality to the appropriate format string for yt-dlp
+quality_format = f"bestvideo[height<={quality}]+bestaudio/best[height<={quality}]"
 
-st.title("Video Downloader")
+# Define download options based on user input
+ydl_opts = {
+    'format': quality_format,  # Download the specified quality
+    'outtmpl': f'{save_file_name}.%(ext)s',  # Save the video with the specified name
+}
 
-url = st.text_input("Enter the video URL:")
+if st.button("Download"):
+    # Download the video
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
+    
+    # Path to the downloaded video file
+    file_path = f'{save_file_name}.webm'
 
-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']} - {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...")
-            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'
-                )
+    # Check if the file exists
+    if os.path.exists(file_path):
+        # Provide a download link for the file
+        with open(file_path, "rb") as file:
+            btn = st.download_button(
+                label="Download video",
+                data=file,
+                file_name=f"{save_file_name}.webm",
+                mime="video/webm"
+            )
+        st.success('Your video is ready for download.')
     else:
-        st.write("Downloading in best available quality...")
-        if st.button("Download"):
-            st.write("Downloading...")
-            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'
-                )
+        st.error("File not found. Please make sure the download was successful.")