Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import yt_dlp
 import os
-import tempfile
 from pathlib import Path
 
 # Set page config
@@ -22,7 +21,6 @@ format_option = st.selectbox(
     "Select Format:",
     options=[
         "Best Quality (Video + Audio)",
-        "Video Only (Best Quality)",
         "Audio Only (Best Quality)",
         "720p",
         "480p",
@@ -30,16 +28,10 @@ format_option = st.selectbox(
     ]
 )
 
-# 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',
+            'format': 'best',  # Changed from bestvideo+bestaudio to best
             'merge_output_format': 'mp4'
         }
     elif selection == "Audio Only (Best Quality)":
@@ -52,35 +44,36 @@ def get_format_options(selection):
         }
     elif selection == "720p":
         return {
-            'format': 'bestvideo[height<=720]+bestaudio/best[height<=720]',
+            'format': 'best[height<=720]',  # Simplified format selection
             'merge_output_format': 'mp4'
         }
     elif selection == "480p":
         return {
-            'format': 'bestvideo[height<=480]+bestaudio/best[height<=480]',
+            'format': 'best[height<=480]',
             'merge_output_format': 'mp4'
         }
     elif selection == "360p":
         return {
-            'format': 'bestvideo[height<=360]+bestaudio/best[height<=360]',
+            'format': 'best[height<=360]',
             'merge_output_format': 'mp4'
         }
 
-# Function to download video
 def download_video(url, format_selection):
     try:
-        # Get format options
         format_opts = get_format_options(format_selection)
         
-        # Base options
         ydl_opts = {
             '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',
+            # Enhanced options to bypass restrictions
+            'nocheckcertificate': True,
+            'ignoreerrors': True,
+            'no_color': True,
+            'noprogress': True,
+            'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+            'referer': 'https://www.youtube.com/',
+            'http-chunk-size': '10485760',  # 10MB chunks for better reliability
         }
         
         # Update options with format-specific settings
@@ -90,7 +83,6 @@ def download_video(url, format_selection):
         progress_bar = st.progress(0)
         status_text = st.empty()
         
-        # Custom progress hook
         def progress_hook(d):
             if d['status'] == 'downloading':
                 try:
@@ -98,7 +90,6 @@ def download_video(url, format_selection):
                     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)
@@ -106,13 +97,9 @@ def download_video(url, format_selection):
 
         ydl_opts['progress_hooks'] = [progress_hook]
         
-        # Download the video
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            # 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 filename
@@ -129,7 +116,6 @@ if st.button("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",
@@ -145,7 +131,7 @@ if st.button("Download"):
     else:
         st.warning("⚠️ Please enter a valid YouTube URL.")
 
-# Add some helpful information
+# Add help section
 with st.expander("ℹ️ Help"):
     st.markdown("""
     **How to use:**