Update app.py

app.py

@@ -300,6 +300,116 @@
 #     """)
 
 
+# import streamlit as st
+# import yt_dlp
+# import os
+# 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 📺")
+
+# # Create output directory if it doesn't exist
+# output_dir = Path("downloads")
+# output_dir.mkdir(exist_ok=True)
+
+# # Check if cookies file exists
+# COOKIES_FILE = 'youtube.com_cookies.txt'
+# has_cookies = os.path.exists(COOKIES_FILE)
+
+# if has_cookies:
+#     st.success("✅ Cookie file detected")
+# else:
+#     st.warning("⚠️ No cookie file found - Some videos might be restricted")
+
+# # Input field for YouTube video URL
+# video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
+
+# def download_video(url):
+#     try:
+#         ydl_opts = {
+#             'format': 'bestvideo+bestaudio/best',
+#             'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
+#             'merge_output_format': 'webm',
+#             'quiet': True,
+#             'no_warnings': 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
+#         }
+
+#         # Add cookies file if available
+#         if has_cookies:
+#             ydl_opts['cookiefile'] = COOKIES_FILE
+
+#         # Progress placeholder
+#         progress_bar = st.progress(0)
+#         status_text = st.empty()
+        
+#         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:
+#                     status_text.text("Downloading... (size unknown)")
+#             elif d['status'] == 'finished':
+#                 progress_bar.progress(1.0)
+#                 status_text.text("Processing...")
+
+#         ydl_opts['progress_hooks'] = [progress_hook]
+        
+#         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+#             info = ydl.extract_info(url, download=True)
+#             filename = ydl.prepare_filename(info)
+#             return filename
+
+#     except Exception as e:
+#         st.error(f"An error occurred: {str(e)}")
+#         return None
+
+# # Download button
+# if st.button("Download"):
+#     if video_url:
+#         try:
+#             with st.spinner("Preparing download..."):
+#                 downloaded_file_path = download_video(video_url)
+                
+#             if downloaded_file_path and os.path.exists(downloaded_file_path):
+#                 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.warning("⚠️ Please enter a valid YouTube URL.")
+
+# # Help section
+# with st.expander("ℹ️ Help & Information"):
+#     st.markdown("""
+#     **About Cookie Authentication:**
+#     - This app uses cookie authentication to bypass YouTube's bot detection
+#     - Cookies help the app behave like a logged-in browser
+#     - No personal data is stored or transmitted
+    
+#     **If downloads fail:**
+#     1. Check if the video is public
+#     2. Verify the URL is correct
+#     3. Try a different video
+#     4. Try again later if YouTube is blocking requests
+#     """)
+
+
 import streamlit as st
 import yt_dlp
 import os
@@ -315,36 +425,69 @@ st.title("YouTube Video Downloader 📺")
 output_dir = Path("downloads")
 output_dir.mkdir(exist_ok=True)
 
-# Check if cookies file exists
-COOKIES_FILE = 'youtube.com_cookies.txt'
-has_cookies = os.path.exists(COOKIES_FILE)
-
-if has_cookies:
-    st.success("✅ Cookie file detected")
-else:
-    st.warning("⚠️ No cookie file found - Some videos might be restricted")
-
 # Input field for YouTube video URL
 video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
 
-def download_video(url):
+# Format selection
+format_option = st.selectbox(
+    "Select Quality",
+    ["Best Quality", "720p", "480p", "360p", "Audio Only"]
+)
+
+def get_format_options(format_choice):
+    if format_choice == "Best Quality":
+        return {
+            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
+            'merge_output_format': 'mp4'
+        }
+    elif format_choice == "720p":
+        return {
+            'format': 'bestvideo[height<=720][ext=mp4]+bestaudio[ext=m4a]/best[height<=720][ext=mp4]/best',
+            'merge_output_format': 'mp4'
+        }
+    elif format_choice == "480p":
+        return {
+            'format': 'bestvideo[height<=480][ext=mp4]+bestaudio[ext=m4a]/best[height<=480][ext=mp4]/best',
+            'merge_output_format': 'mp4'
+        }
+    elif format_choice == "360p":
+        return {
+            'format': 'bestvideo[height<=360][ext=mp4]+bestaudio[ext=m4a]/best[height<=360][ext=mp4]/best',
+            'merge_output_format': 'mp4'
+        }
+    else:  # Audio Only
+        return {
+            'format': 'bestaudio[ext=m4a]/best',
+            'merge_output_format': 'm4a'
+        }
+
+def download_video(url, format_choice):
     try:
+        # Base options
         ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',
             'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
-            'merge_output_format': 'webm',
             'quiet': True,
             'no_warnings': 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
+            'extract_flat': False,
+            'no_color': True,
+            # Advanced options to help avoid detection
+            'socket_timeout': 30,
+            'retries': 3,
+            'ignoreerrors': True,
+            'nocheckcertificate': True,
+            # Modern browser headers
+            'headers': {
+                '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',
+                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
+                'Accept-Language': 'en-us,en;q=0.5',
+                'Sec-Fetch-Mode': 'navigate',
+            }
         }
-
-        # Add cookies file if available
-        if has_cookies:
-            ydl_opts['cookiefile'] = COOKIES_FILE
-
-        # Progress placeholder
+        
+        # Add format-specific options
+        ydl_opts.update(get_format_options(format_choice))
+        
+        # Progress tracking
         progress_bar = st.progress(0)
         status_text = st.empty()
         
@@ -355,20 +498,34 @@ def download_video(url):
                     progress_bar.progress(progress)
                     status_text.text(f"Downloading: {progress:.1%}")
                 except:
-                    status_text.text("Downloading... (size unknown)")
+                    progress = -1
+                    if 'downloaded_bytes' in d:
+                        status_text.text(f"Downloading... ({d['downloaded_bytes'] / 1024 / 1024:.1f} MB)")
+                    else:
+                        status_text.text("Downloading... (size unknown)")
             elif d['status'] == 'finished':
                 progress_bar.progress(1.0)
                 status_text.text("Processing...")
-
+                
         ydl_opts['progress_hooks'] = [progress_hook]
         
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=True)
-            filename = ydl.prepare_filename(info)
-            return filename
-
+            # First extract info to verify video is accessible
+            info = ydl.extract_info(url, download=False)
+            if info:
+                # Then proceed with download
+                info = ydl.extract_info(url, download=True)
+                filename = ydl.prepare_filename(info)
+                return filename
+            
+    except yt_dlp.utils.DownloadError as e:
+        if "Sign in to confirm you're not a bot" in str(e):
+            st.error("❌ YouTube is requesting verification. Please try again later or try a different video.")
+        else:
+            st.error(f"❌ Download error: {str(e)}")
+        return None
     except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
+        st.error(f"❌ An error occurred: {str(e)}")
         return None
 
 # Download button
@@ -376,19 +533,28 @@ if st.button("Download"):
     if video_url:
         try:
             with st.spinner("Preparing download..."):
-                downloaded_file_path = download_video(video_url)
+                downloaded_file_path = download_video(video_url, format_option)
                 
             if downloaded_file_path and os.path.exists(downloaded_file_path):
+                # Get file size
+                file_size = os.path.getsize(downloaded_file_path) / (1024 * 1024)  # Convert to MB
+                
                 with open(downloaded_file_path, 'rb') as file:
                     st.download_button(
-                        label="Click here to download",
+                        label=f"Download ({file_size:.1f} MB)",
                         data=file,
                         file_name=os.path.basename(downloaded_file_path),
                         mime="application/octet-stream"
                     )
                 st.success("✅ Download ready!")
+                
+                # Clean up downloaded file after successful download
+                try:
+                    os.remove(downloaded_file_path)
+                except:
+                    pass
             else:
-                st.error("❌ Download failed. Please try again.")
+                st.error("❌ Download failed. Please try a different video or quality setting.")
         except Exception as e:
             st.error(f"❌ An error occurred: {str(e)}")
     else:
@@ -397,14 +563,17 @@ if st.button("Download"):
 # Help section
 with st.expander("ℹ️ Help & Information"):
     st.markdown("""
-    **About Cookie Authentication:**
-    - This app uses cookie authentication to bypass YouTube's bot detection
-    - Cookies help the app behave like a logged-in browser
-    - No personal data is stored or transmitted
+    **Features:**
+    - Multiple quality options available
+    - Supports both video and audio-only downloads
+    - Automatic format selection based on availability
     
     **If downloads fail:**
-    1. Check if the video is public
-    2. Verify the URL is correct
-    3. Try a different video
-    4. Try again later if YouTube is blocking requests
+    1. Try a lower quality setting
+    2. Verify the video is publicly available
+    3. Check your internet connection
+    4. Wait a few minutes and try again
+    
+    **Note:**
+    This app respects YouTube's terms of service and is intended for downloading videos that are freely available and permitted to be downloaded.
     """)