Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import yt_dlp
 import os
 from pathlib import Path
+import random
+import time
 
 # 페이지 설정
 st.set_page_config(page_title="Simple YouTube Downloader", page_icon="📺")
@@ -11,15 +13,35 @@ st.title("Simple YouTube Downloader 📺")
 output_dir = Path("downloads")
 output_dir.mkdir(exist_ok=True)
 
-def reset_form():
-    st.session_state.url = ""
-    
+def reset_input():
+    # session_state 대신 빈 문자열 반환
+    return ""
+
 def download_video(url, progress_bar, status_text):
     try:
+        # User Agent 목록
+        user_agents = [
+            '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/605.1.15 (KHTML, like Gecko) Version/17.2.1 Safari/605.1.15',
+            'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:122.0) Gecko/20100101 Firefox/122.0'
+        ]
+        
         ydl_opts = {
-            'format': 'best',  # 최고 품질의 통합 포맷
+            'format': 'best',
             'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
-            'quiet': True
+            'quiet': True,
+            'no_warnings': True,
+            # 봇 감지 우회 설정
+            'user_agent': random.choice(user_agents),
+            'referer': 'https://www.youtube.com/',
+            'http_chunk_size': random.randint(10000000, 15000000),
+            'retries': 3,
+            'sleep_interval': 1,
+            'max_sleep_interval': 5,
+            'headers': {
+                'Accept-Language': 'en-US,en;q=0.9',
+                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
+            }
         }
         
         def progress_hook(d):
@@ -36,17 +58,36 @@ def download_video(url, progress_bar, status_text):
 
         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
+        # 재시도 로직 추가
+        for attempt in range(3):
+            try:
+                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                    info = ydl.extract_info(url, download=True)
+                    filename = ydl.prepare_filename(info)
+                    return filename
+            except yt_dlp.utils.ExtractorError as e:
+                if "Sign in to confirm you're not a bot" in str(e):
+                    status_text.text(f"봇 감지 우회 중... ({attempt + 1}/3)")
+                    time.sleep(random.uniform(3, 5))
+                    continue
+                raise e
+            except Exception as e:
+                if attempt < 2:
+                    status_text.text(f"재시도 중... ({attempt + 1}/3)")
+                    time.sleep(2)
+                    continue
+                raise e
 
     except Exception as e:
         st.error(f"오류 발생: {str(e)}")
         return None
 
 # 메인 UI
-video_url = st.text_input("YouTube URL 입력:", key="url", 
+if 'video_url' not in st.session_state:
+    st.session_state.video_url = ""
+
+video_url = st.text_input("YouTube URL 입력:", 
+                         value=st.session_state.video_url,
                          placeholder="https://www.youtube.com/watch?v=...")
 
 col1, col2 = st.columns(2)
@@ -72,115 +113,5 @@ with col1:
 
 with col2:
     if st.button("초기화"):
-        reset_form()
-
-
-# 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)
-
-# # Input field for YouTube video URL
-# video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
-
-# # Function to download video
-# def download_video(url):
-#     try:
-        
-#         ydl_opts = {
-#             'format': 'bestvideo+bestaudio/best',
-#             'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
-#             'merge_output_format': 'webm',
-#             'quiet': False,  # Change to False to see more detailed error output
-#             'no_warnings': 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/',
-#             'cookiefile': 'youtube.com_cookies.txt',  # Ensure this file path is correct
-#             'socket_timeout': 30,
-#             'http_chunk_size': 10485760,  # 10MB
-#             'retries': 10,  # Increase retry attempts
-#             'headers': {
-#                 'Accept-Language': 'en-US,en;q=0.9',
-#                 'Accept-Encoding': 'gzip, deflate, br',
-#                 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',
-#                 'Upgrade-Insecure-Requests': '1',
-#             }
-#         }
-
-#         # 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 & Troubleshooting"):
-#     st.markdown("""
-#     **If you're experiencing download issues:**
-    
-#     1. Cookie Authentication Method:
-#        - Install a browser extension like "Get cookies.txt"
-#        - Visit YouTube and ensure you're logged in
-#        - Export cookies to 'youtube.com_cookies.txt'
-#        - Place the file in the same directory as this app
-    
-#     2. Alternative Solutions:
-#        - Try different videos
-#        - Check if the video is public/not age-restricted
-#        - Try again later if YouTube is blocking requests
-#     """)
-
+        st.session_state.video_url = ""
+        st.rerun()