Update app.py

app.py

@@ -1,43 +1,58 @@
 import streamlit as st
 import yt_dlp
-import os
 
-def download_video(url, download_path):
+# Function to get video info (formats, quality, etc.)
+def get_video_info(url):
     ydl_opts = {
-        'format': 'best',  # Download the best quality by default
-        'outtmpl': os.path.join(download_path, '%(title)s.%(ext)s'),
-        'noplaylist': True,  # Don't download playlists
+        'quiet': True,
+        'noplaylist': True,
     }
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info = ydl.extract_info(url, download=False)
+    return info
 
+# Function to download video
+def download_video(url, format):
+    ydl_opts = {
+        'format': format,
+        'outtmpl': '%(title)s.%(ext)s',  # Name the file based on the video title
+        'noplaylist': True,
+    }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info_dict = ydl.extract_info(url, download=False)
-        formats = info_dict.get('formats', [])
-        best_quality = formats[0]
-        for f in formats:
-            if f['height'] > best_quality['height']:
-                best_quality = f
-        download_url = best_quality['url']
-        file_extension = best_quality['ext']
-        
-        # Initiate the download
-        ydl_opts['format'] = best_quality['format_id']
-        ydl_opts['outtmpl'] = os.path.join(download_path, '%(title)s.' + file_extension)
         ydl.download([url])
 
-        return f"Downloaded: {info_dict['title']}.{file_extension}"
+# Streamlit layout
+st.title("Video Downloader App")
+
+# User input for video URL
+url = st.text_input('Enter Video URL')
+
+if url:
+    try:
+        # Get video information (formats)
+        video_info = get_video_info(url)
+        formats = video_info.get('formats', [])
+
+        # Prepare list of available formats (e.g., 'best', 'mp4', 'webm')
+        format_options = []
+        for f in formats:
+            # Ensure format has 'format_note' and 'ext' to avoid key errors
+            if 'format_note' in f and 'ext' in f and 'filesize' in f:
+                format_options.append(f"{f['format_note']} - {f['ext']} - {f['filesize']} bytes")
+
+        if format_options:
+            # User selects a format
+            selected_format = st.selectbox('Select Video Quality', format_options)
 
-# Streamlit App
-st.title('Video Downloader')
-st.write("Enter the video link to download it from YouTube, Facebook, Instagram, X (formerly Twitter), or Telegram.")
+            # Convert selected format to yt-dlp's format identifier
+            selected_format_id = formats[format_options.index(selected_format)]['format_id']
 
-# User input
-video_url = st.text_input("Video URL")
-download_button = st.button("Download Video")
+            if st.button('Download'):
+                with st.spinner('Downloading...'):
+                    download_video(url, selected_format_id)
+                st.success('Download complete!')
+        else:
+            st.error("No formats available.")
+    except Exception as e:
+        st.error(f"Error fetching video info: {e}")
 
-if download_button:
-    if video_url:
-        with st.spinner("Downloading..."):
-            result = download_video(video_url, os.path.expanduser("~") + '/Downloads')
-            st.success(result)
-    else:
-        st.error("Please enter a valid video URL.")