Update app.py

app.py

@@ -2,50 +2,158 @@ import streamlit as st
 import yt_dlp
 import os
 import tempfile
+from pathlib import Path
+
+# Set page config
+st.set_page_config(page_title="YouTube Video Downloader", page_icon="📺")
 
 # Set the title of the app
-st.title("YouTube Video Downloader")
+st.title("YouTube Video Downloader 📺")
+
+# Create output directory if it doesn't exist
+output_dir = Path("downloads")
+output_dir.mkdir(exist_ok=True)
 
 # Input field for YouTube video URL
-video_url = st.text_input("Enter YouTube Video URL:")
+video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
+
+# Format selection
+format_option = st.selectbox(
+    "Select Format:",
+    options=[
+        "Best Quality (Video + Audio)",
+        "Video Only (Best Quality)",
+        "Audio Only (Best Quality)",
+        "720p",
+        "480p",
+        "360p"
+    ]
+)
+
+# Function to get format based on selection
+def get_format_options(selection):
+    if selection == "Best Quality (Video + Audio)":
+        return {
+            'format': 'bestvideo+bestaudio/best',
+            'merge_output_format': 'mp4'
+        }
+    elif selection == "Video Only (Best Quality)":
+        return {
+            'format': 'bestvideo/best',
+            'merge_output_format': 'mp4'
+        }
+    elif selection == "Audio Only (Best Quality)":
+        return {
+            'format': 'bestaudio/best',
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+            }]
+        }
+    elif selection == "720p":
+        return {
+            'format': 'bestvideo[height<=720]+bestaudio/best[height<=720]',
+            'merge_output_format': 'mp4'
+        }
+    elif selection == "480p":
+        return {
+            'format': 'bestvideo[height<=480]+bestaudio/best[height<=480]',
+            'merge_output_format': 'mp4'
+        }
+    elif selection == "360p":
+        return {
+            'format': 'bestvideo[height<=360]+bestaudio/best[height<=360]',
+            'merge_output_format': 'mp4'
+        }
 
 # Function to download video
-def download_video(url):
-    # Create a temporary directory to store the downloaded video
-    with tempfile.TemporaryDirectory() as tmpdirname:
-        # Define download options
+def download_video(url, format_selection):
+    try:
+        # Get format options
+        format_opts = get_format_options(format_selection)
+        
+        # Base options
         ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',
-            'outtmpl': os.path.join(tmpdirname, '%(title)s.%(ext)s'),
-            'merge_output_format': 'webm',
-            'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.93 Safari/537.36',
+            'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
+            'quiet': True,
+            'no_warnings': True,
+            'extract_flat': False,
+            # Add cookies and user agent to bypass some restrictions
+            'cookiefile': 'cookies.txt',  # If you have a cookies file
+            'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
         }
+        
+        # Update options with format-specific settings
+        ydl_opts.update(format_opts)
 
+        # Progress placeholder
+        progress_bar = st.progress(0)
+        status_text = st.empty()
+        
+        # Custom progress hook
+        def progress_hook(d):
+            if d['status'] == 'downloading':
+                try:
+                    progress = d['downloaded_bytes'] / d['total_bytes']
+                    progress_bar.progress(progress)
+                    status_text.text(f"Downloading: {progress:.1%}")
+                except:
+                    # For cases where total_bytes is unknown
+                    status_text.text("Downloading... (size unknown)")
+            elif d['status'] == 'finished':
+                progress_bar.progress(1.0)
+                status_text.text("Processing...")
 
-        # Download the video using yt-dlp
+        ydl_opts['progress_hooks'] = [progress_hook]
+        
+        # Download the video
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            # Use a progress hook to update the progress bar
-            def progress_hook(d):
-                if d['status'] == 'downloading':
-                    st.progress(d['downloaded_bytes'] / d['total_bytes'])
-                elif d['status'] == 'finished':
-                    st.success("Download completed!")
-
-            ydl.params['progress_hooks'] = [progress_hook]
+            # First extract info to get the title
+            info = ydl.extract_info(url, download=False)
+            filename = ydl.prepare_filename(info)
+            
+            # Then download
             ydl.download([url])
-        
-        # Return the path of the downloaded video
-        return os.path.join(tmpdirname, f"{ydl.prepare_filename(ydl.extract_info(url))}")
+            
+            return filename
+
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")
+        return None
 
 # Download button
-if st.button("Download Video"):
+if st.button("Download"):
     if video_url:
-        # Display progress bar
-        st.write("Downloading video...")
-        downloaded_file_path = download_video(video_url)
-
-        # Provide a download link
-        st.write("Download completed! Click below to download your video:")
-        st.markdown(f"[Download Video]({downloaded_file_path})", unsafe_allow_html=True)
+        try:
+            with st.spinner("Preparing download..."):
+                downloaded_file_path = download_video(video_url, format_option)
+                
+            if downloaded_file_path and os.path.exists(downloaded_file_path):
+                # Read the file and provide it for download
+                with open(downloaded_file_path, 'rb') as file:
+                    st.download_button(
+                        label="Click here to download",
+                        data=file,
+                        file_name=os.path.basename(downloaded_file_path),
+                        mime="application/octet-stream"
+                    )
+                st.success("✅ Download ready!")
+            else:
+                st.error("❌ Download failed. Please try again.")
+        except Exception as e:
+            st.error(f"❌ An error occurred: {str(e)}")
     else:
-        st.error("Please enter a valid YouTube URL.")
+        st.warning("⚠️ Please enter a valid YouTube URL.")
+
+# Add some helpful information
+with st.expander("ℹ️ Help"):
+    st.markdown("""
+    **How to use:**
+    1. Paste a YouTube video URL in the input field
+    2. Select your preferred format
+    3. Click the Download button
+    4. Wait for the download to complete
+    5. Click the download button that appears
+    
+    **Note:** Some videos might be restricted due to YouTube's policies.
+    """)