Update app.py

app.py

@@ -216,9 +216,9 @@ import streamlit as st
 import yt_dlp
 import os
 from pathlib import Path
-import json
 from http.cookiejar import MozillaCookieJar
-from datetime import datetime
+import random
+import time
 
 # Constants
 COOKIES_FILE = 'youtube.com_cookies.txt'
@@ -231,79 +231,99 @@ st.title("YouTube Video Downloader 📺")
 # Create output directory if it doesn't exist
 OUTPUT_DIR.mkdir(exist_ok=True)
 
+def get_random_user_agent():
+    """랜덤 User-Agent 생성"""
+    user_agents = [
+        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36',
+        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+        'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36',
+        'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36',
+        'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:122.0) Gecko/20100101 Firefox/122.0'
+    ]
+    return random.choice(user_agents)
+
 def validate_cookies():
-    """쿠키 파일 유효성 검사"""
+    """쿠키 파일 유효성 검사 및 갱신"""
     if not os.path.exists(COOKIES_FILE):
         return False, "Cookie file not found"
     
     try:
         cookie_jar = MozillaCookieJar(COOKIES_FILE)
         cookie_jar.load()
-        
-        # Check for essential YouTube cookies
-        essential_cookies = ['CONSENT', 'LOGIN_INFO', 'PREF']
-        found_cookies = [cookie.name for cookie in cookie_jar]
-        
-        missing_cookies = [cookie for cookie in essential_cookies if cookie not in found_cookies]
-        if missing_cookies:
-            return False, f"Missing essential cookies: {', '.join(missing_cookies)}"
-            
-        return True, "Cookies valid"
-    except Exception as e:
-        return False, f"Cookie validation error: {str(e)}"
-
-def get_video_info(url):
-    """동영상 정보 미리보기 추출"""
-    ydl_opts = {
-        'quiet': True,
-        'no_warnings': True,
-        'extract_flat': True,
-        'cookiefile': COOKIES_FILE,
-    }
-    
-    try:
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            return ydl.extract_info(url, download=False)
+        return True, "Cookies loaded successfully"
     except Exception as e:
-        return None
+        return False, f"Cookie error: {str(e)}"
+
+def download_with_retry(url, ydl_opts, max_retries=3, delay=5):
+    """재시도 로직이 포함된 다운로드 함수"""
+    for attempt in range(max_retries):
+        try:
+            # 매 시도마다 새로운 User-Agent 사용
+            ydl_opts['headers']['User-Agent'] = get_random_user_agent()
+            
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info = ydl.extract_info(url, download=True)
+                filename = ydl.prepare_filename(info)
+                return filename, info
+                
+        except yt_dlp.utils.DownloadError as e:
+            if "Sign in to confirm you're not a bot" in str(e):
+                if attempt < max_retries - 1:
+                    st.warning(f"Bot detection encountered. Retrying in {delay} seconds... (Attempt {attempt + 1}/{max_retries})")
+                    time.sleep(delay)
+                    delay *= 2  # 지수 백오프
+                    continue
+            raise e
+        except Exception as e:
+            if attempt < max_retries - 1:
+                st.warning(f"Error occurred. Retrying... (Attempt {attempt + 1}/{max_retries})")
+                time.sleep(delay)
+                continue
+            raise e
 
 def download_video(url):
     try:
         ydl_opts = {
-            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',  # Prefer MP4
+            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
             'outtmpl': str(OUTPUT_DIR / '%(title)s.%(ext)s'),
             'merge_output_format': 'mp4',
             
-            # Cookie and authentication settings
+            # 쿠키 및 인증 설정
             'cookiefile': COOKIES_FILE,
-            'cookiesfrombrowser': ('chrome',),  # Also try to get cookies from Chrome
+            'cookiesfrombrowser': ('chrome',),
             
-            # Enhanced options
+            # 향상된 옵션
             'quiet': True,
             'no_warnings': True,
             'extract_flat': False,
             'socket_timeout': 30,
             'retries': 10,
             'fragment_retries': 10,
-            'file_access_retries': 10,
             
-            # 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',
-                'Accept-Encoding': 'gzip,deflate',
-                'DNT': '1',
+                'User-Agent': get_random_user_agent(),
+                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
+                'Accept-Language': 'en-US,en;q=0.5',
+                'Accept-Encoding': 'gzip, deflate, br',
+                'Connection': 'keep-alive',
                 'Upgrade-Insecure-Requests': '1',
+                'Sec-Fetch-Dest': 'document',
+                'Sec-Fetch-Mode': 'navigate',
+                'Sec-Fetch-Site': 'none',
+                'Sec-Fetch-User': '?1',
+                'Cache-Control': 'max-age=0',
             },
             
-            # Additional options
-            'age_limit': None,  # Try to bypass age restriction
+            # 추가 옵션
+            'age_limit': None,
             'geo_bypass': True,
             'geo_bypass_country': 'US',
+            'sleep_interval': 2,  # 요청 사이 대기 시간
+            'max_sleep_interval': 5,
         }
         
-        # Progress tracking
+        # 진행 상황 표시
         progress_bar = st.progress(0)
         status_text = st.empty()
         
@@ -314,69 +334,45 @@ def download_video(url):
                     progress_bar.progress(progress)
                     status_text.text(f"Downloading: {progress:.1%} | Speed: {d.get('speed_str', 'N/A')} | ETA: {d.get('eta_str', 'N/A')}")
                 except:
-                    status_text.text(f"Downloading... Size: {d.get('total_bytes_str', 'unknown')} | Speed: {d.get('speed_str', 'N/A')}")
+                    status_text.text(f"Downloading... Speed: {d.get('speed_str', 'N/A')}")
             elif d['status'] == 'finished':
                 progress_bar.progress(1.0)
-                status_text.text("Processing downloaded files...")
+                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, info
+        return download_with_retry(url, ydl_opts)
             
-    except yt_dlp.utils.DownloadError as e:
-        if "Sign in to confirm your age" in str(e):
-            raise Exception("This video requires age verification. Please ensure you're logged in through the cookies file.")
-        elif "The uploader has not made this video available in your country" in str(e):
-            raise Exception("This video is not available in your country. Try using a VPN.")
-        else:
-            raise Exception(f"Download error: {str(e)}")
     except Exception as e:
-        raise Exception(f"An unexpected error occurred: {str(e)}")
+        raise Exception(f"Download error: {str(e)}")
 
-# Check cookie status at startup
+# 쿠키 상태 확인
 cookie_valid, cookie_message = validate_cookies()
 if not cookie_valid:
     st.warning(f"⚠️ Cookie Issue: {cookie_message}")
 else:
     st.success("✅ Cookies loaded successfully")
 
-# Video URL input
+# 비디오 URL 입력
 video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
 
-# Preview button
-if video_url and st.button("Preview Video Info"):
-    with st.spinner("Fetching video information..."):
-        info = get_video_info(video_url)
-        if info:
-            st.write("**Video Information:**")
-            st.write(f"Title: {info.get('title', 'N/A')}")
-            st.write(f"Duration: {info.get('duration_string', 'N/A')}")
-            st.write(f"View Count: {info.get('view_count', 'N/A'):,}")
-            st.write(f"Upload Date: {info.get('upload_date', 'N/A')}")
-
-# Download button
+# 다운로드 버튼
 if st.button("Download Video"):
     if not video_url:
         st.warning("⚠️ Please enter a valid YouTube URL.")
     elif not cookie_valid:
-        st.error("❌ Cannot proceed without valid cookies. Please check your cookie file.")
+        st.error("❌ Cannot proceed without valid cookies.")
     else:
         try:
             with st.spinner("Preparing download..."):
                 downloaded_file_path, video_info = download_video(video_url)
                 
             if downloaded_file_path and os.path.exists(downloaded_file_path):
-                # Add file info
                 file_size = os.path.getsize(downloaded_file_path)
                 st.info(f"""
                 **Download Complete!**
                 - File: {os.path.basename(downloaded_file_path)}
                 - Size: {file_size / (1024*1024):.1f} MB
-                - Format: {video_info.get('format', 'N/A')}
-                - Resolution: {video_info.get('resolution', 'N/A')}
                 """)
                 
                 with open(downloaded_file_path, 'rb') as file:
@@ -391,25 +387,23 @@ if st.button("Download Video"):
         except Exception as e:
             st.error(f"❌ {str(e)}")
 
-# Enhanced help section
-with st.expander("ℹ️ Help & Cookie Information"):
+# Help 섹션
+with st.expander("ℹ️ Troubleshooting Guide"):
     st.markdown("""
-    **Cookie File Status:**
-    - Location: `youtube.com_cookies.txt`
-    - Valid: {'✅' if cookie_valid else '❌'}
-    - Message: {cookie_message}
-    
-    **Common Issues:**
-    1. Age-Restricted Videos:
-       - Ensure you're logged into YouTube when exporting cookies
-       - The cookie file must contain valid authentication tokens
-    
-    2. Region-Restricted Content:
-       - The app attempts to bypass region restrictions automatically
-       - Consider using a VPN if download fails
+    **If you encounter 'Sign in to confirm you're not a bot' error:**
     
-    3. Cookie-Related Problems:
-       - Use a browser extension (e.g., "Get cookies.txt") to export cookies
-       - Make sure you're logged into YouTube before exporting
-       - Cookie file must be in Mozilla/Netscape format
+    1. Cookie Solution:
+       - Make sure you're logged into YouTube in your browser
+       - Install 'Get cookies.txt' or similar extension
+       - Export fresh cookies and replace the existing cookie file
+       
+    2. General Tips:
+       - Wait a few minutes between download attempts
+       - Try using a different video URL
+       - Clear your browser cache before exporting cookies
+       
+    3. Advanced Tips:
+       - The app automatically retries downloads with different settings
+       - Each retry uses a different User-Agent to avoid detection
+       - Built-in delays help avoid triggering YouTube's protection
     """)