Update app.py

app.py

@@ -113,9 +113,12 @@ import streamlit as st
 import yt_dlp
 import os
 from pathlib import Path
+import time
+from datetime import datetime
+import json
 
 # Set page config
-st.set_page_config(page_title="YouTube Video Downloader", page_icon="📺")
+st.set_page_config(page_title="YouTube Video Downloader", page_icon="📺", layout="wide")
 
 # Set the title of the app
 st.title("YouTube Video Downloader 📺")
@@ -124,98 +127,181 @@ 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)
+# Authentication settings
+AUTH_CONFIG = {
+    'COOKIES_FILE': 'youtube.com_cookies.txt',
+    'TOKEN_FILE': 'auth_token.json'
+}
 
-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):
+def load_auth_token():
     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
+        if os.path.exists(AUTH_CONFIG['TOKEN_FILE']):
+            with open(AUTH_CONFIG['TOKEN_FILE'], 'r') as f:
+                data = json.load(f)
+                if datetime.fromisoformat(data['expires']) > datetime.now():
+                    return data['token']
+    except Exception:
+        pass
+    return None
+
+def save_auth_token(token):
+    with open(AUTH_CONFIG['TOKEN_FILE'], 'w') as f:
+        json.dump({
+            'token': token,
+            'expires': (datetime.now().isoformat())
+        }, f)
+
+def setup_ydl_options(has_cookies):
+    opts = {
+        'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
+        'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
+        'merge_output_format': 'mp4',
+        'quiet': True,
+        'no_warnings': True,
+        'extract_flat': False,
+        '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,
+        'retries': 3,
+        'file_access_retries': 3,
+        'fragment_retries': 3,
+        'skip_unavailable_fragments': True,
+        'abort_on_unavailable_fragment': False,
+    }
+    
+    # Add cookies file if available
+    if has_cookies:
+        opts['cookiefile'] = AUTH_CONFIG['COOKIES_FILE']
+    
+    # Add authentication token if available
+    token = load_auth_token()
+    if token:
+        opts['headers'] = {
+            'Authorization': f'Bearer {token}'
         }
+    
+    return opts
 
-        # 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:
+def download_video(url, progress_callback):
+    max_retries = 3
+    retry_count = 0
+    
+    while retry_count < max_retries:
         try:
-            with st.spinner("Preparing download..."):
-                downloaded_file_path = download_video(video_url)
+            has_cookies = os.path.exists(AUTH_CONFIG['COOKIES_FILE'])
+            ydl_opts = setup_ydl_options(has_cookies)
+            
+            # Add progress hooks
+            def progress_hook(d):
+                if d['status'] == 'downloading':
+                    try:
+                        progress = d['downloaded_bytes'] / d['total_bytes']
+                        progress_callback(progress, f"Downloading: {progress:.1%}")
+                    except:
+                        progress_callback(0, "Downloading... (size unknown)")
+                elif d['status'] == 'finished':
+                    progress_callback(1.0, "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, None
                 
-            if downloaded_file_path and os.path.exists(downloaded_file_path):
-                with open(downloaded_file_path, 'rb') as file:
+        except yt_dlp.utils.ExtractorError as e:
+            if "Sign in to confirm you're not a bot" in str(e):
+                retry_count += 1
+                if retry_count < max_retries:
+                    time.sleep(5)  # Wait before retry
+                    continue
+                return None, "Bot detection triggered. Please try again later or use authentication."
+            return None, f"Video extraction failed: {str(e)}"
+            
+        except Exception as e:
+            return None, f"Download failed: {str(e)}"
+    
+    return None, "Maximum retries reached. Please try again later."
+
+# Main UI
+col1, col2 = st.columns([2, 1])
+
+with col1:
+    video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
+    
+    if st.button("Download", type="primary"):
+        if video_url:
+            progress_bar = st.progress(0)
+            status_text = st.empty()
+            
+            def update_progress(progress, status):
+                progress_bar.progress(progress)
+                status_text.text(status)
+            
+            downloaded_file, error = download_video(video_url, update_progress)
+            
+            if downloaded_file and os.path.exists(downloaded_file):
+                with open(downloaded_file, 'rb') as file:
                     st.download_button(
-                        label="Click here to download",
+                        label="⬇️ Download Video",
                         data=file,
-                        file_name=os.path.basename(downloaded_file_path),
-                        mime="application/octet-stream"
+                        file_name=os.path.basename(downloaded_file),
+                        mime="video/mp4"
                     )
                 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.error(f"❌ {error}")
+        else:
+            st.warning("⚠️ Please enter a valid YouTube URL.")
+
+with col2:
+    with st.expander("⚙️ Settings & Information"):
+        st.markdown("""
+        **Authentication Status:**
+        """)
+        
+        if os.path.exists(AUTH_CONFIG['COOKIES_FILE']):
+            st.success("✅ Cookie authentication available")
+        else:
+            st.warning("""
+            ⚠️ No cookie authentication configured
+            
+            To enable cookie authentication:
+            1. Export cookies from your browser
+            2. Save as 'youtube.com_cookies.txt' in the app directory
+            """)
+        
+        st.markdown("""
+        **Download Options:**
+        - Videos are downloaded in best available quality
+        - Format: MP4 (when available)
+        - Automatic quality selection
+        """)
+
+with st.expander("❓ Help & Troubleshooting"):
     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
+    **Common Issues & Solutions:**
+    
+    1. **Bot Detection:**
+       - The app implements automatic retries
+       - Use cookie authentication for better success rate
+       - Wait a few minutes between attempts
+    
+    2. **Download Fails:**
+       - Verify the video is public
+       - Check your internet connection
+       - Try a different video
+       - Clear browser cache and cookies
+    
+    3. **Quality Issues:**
+       - The app automatically selects the best available quality
+       - Some videos may have quality restrictions
+       - Premium content may require authentication
     
-    **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
+    **Need More Help?**
+    - Check if the video is available in your region
+    - Verify YouTube's terms of service
+    - Consider using cookie authentication
     """)