Update app.py

app.py

@@ -1,304 +1,3 @@
-# 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=...")
-
-# # Format selection
-# format_option = st.selectbox(
-#     "Select Format:",
-#     options=[
-#         "Best Quality (Video + Audio)",
-#         "Audio Only (Best Quality)",
-#         "720p",
-#         "480p",
-#         "360p"
-#     ]
-# )
-
-# def get_format_options(selection):
-#     base_options = {
-#         'format': 'best',
-#         'merge_output_format': 'mp4',
-#     }
-    
-#     if selection == "Best Quality (Video + Audio)":
-#         return base_options
-#     elif selection == "Audio Only (Best Quality)":
-#         return {
-#             'format': 'bestaudio/best',
-#             'postprocessors': [{
-#                 'key': 'FFmpegExtractAudio',
-#                 'preferredcodec': 'mp3',
-#             }]
-#         }
-#     elif selection == "720p":
-#         base_options['format'] = 'best[height<=720]'
-#         return base_options
-#     elif selection == "480p":
-#         base_options['format'] = 'best[height<=480]'
-#         return base_options
-#     elif selection == "360p":
-#         base_options['format'] = 'best[height<=360]'
-#         return base_options
-    
-#     return base_options
-
-# def validate_url(url):
-#     if not url:
-#         return False
-#     return 'youtube.com' in url or 'youtu.be' in url
-
-# def download_video(url, format_selection):
-#     if not validate_url(url):
-#         st.error("Please enter a valid YouTube URL")
-#         return None
-        
-#     try:
-#         format_opts = get_format_options(format_selection)
-        
-#         ydl_opts = {
-#             'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
-#             'quiet': True,
-#             'no_warnings': True,
-#             'nocheckcertificate': True,
-#             'ignoreerrors': False,  # Changed to False to catch errors
-#             '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',
-#             'extract_flat': True,  # Only extract video metadata first
-#         }
-        
-#         # Update options with format-specific settings
-#         ydl_opts.update(format_opts)
-
-#         # Progress elements
-#         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]
-        
-#         # First, try to extract info without downloading
-#         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-#             try:
-#                 info = ydl.extract_info(url, download=False)
-#                 if info is None:
-#                     st.error("Could not fetch video information. The video might be private or age-restricted.")
-#                     return None
-                
-#                 # If info extraction successful, proceed with download
-#                 ydl_opts['extract_flat'] = False  # Now enable full extraction
-#                 with yt_dlp.YoutubeDL(ydl_opts) as ydl_download:
-#                     filename = ydl_download.prepare_filename(info)
-#                     ydl_download.download([url])
-                    
-#                     # Check if file exists and has size > 0
-#                     if os.path.exists(filename) and os.path.getsize(filename) > 0:
-#                         return filename
-#                     else:
-#                         st.error("Download completed but file appears to be empty or missing.")
-#                         return None
-                        
-#             except yt_dlp.utils.DownloadError as e:
-#                 st.error(f"Download Error: {str(e)}")
-#                 return None
-#             except Exception as e:
-#                 st.error(f"An unexpected error occurred: {str(e)}")
-#                 return None
-
-#     except Exception as e:
-#         st.error(f"An error occurred during setup: {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, format_option)
-                
-#             if downloaded_file_path and os.path.exists(downloaded_file_path):
-#                 try:
-#                     with open(downloaded_file_path, 'rb') as file:
-#                         file_data = file.read()
-#                         if len(file_data) > 0:
-#                             st.download_button(
-#                                 label="Click here to download",
-#                                 data=file_data,
-#                                 file_name=os.path.basename(downloaded_file_path),
-#                                 mime="application/octet-stream"
-#                             )
-#                             st.success("✅ Download ready!")
-#                         else:
-#                             st.error("❌ Downloaded file is empty.")
-#                 except IOError as e:
-#                     st.error(f"❌ Error reading file: {str(e)}")
-#             else:
-#                 st.error("❌ Download failed. Please try again with a different video or format.")
-#         except Exception as e:
-#             st.error(f"❌ An unexpected error occurred: {str(e)}")
-#     else:
-#         st.warning("⚠️ Please enter a valid YouTube URL.")
-
-# # Add help section
-# with st.expander("ℹ️ Help"):
-#     st.markdown("""
-#     **How to use:**
-#     1. Paste a YouTube video URL in the input field
-#     2. Select your preferred format
-#     3. Click the Download button
-#     4. Wait for the download to complete
-#     5. Click the download button that appears
-    
-#     **Troubleshooting:**
-#     - If download fails, try a different format
-#     - Some videos might be restricted or private
-#     - Make sure the URL is from YouTube
-#     """)
-
-# # Clean up old files
-# def cleanup_old_files():
-#     try:
-#         for file in output_dir.glob('*'):
-#             if file.is_file():
-#                 file.unlink()
-#     except Exception as e:
-#         st.warning(f"Could not clean up old files: {str(e)}")
-
-# # Run cleanup when the app starts
-# cleanup_old_files()
-
-
-
-
-# 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',
-#             # Add these options to mimic browser behavior
-#             '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/',
-#             'cookiefile': 'youtube.com_cookies.txt',  # 쿠키 파일 사용
-#             'socket_timeout': 30,
-#             'http_chunk_size': 10485760,  # 10MB
-#         }
-
-#         # 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
-#     """)
-
 
 import streamlit as st
 import yt_dlp
@@ -315,35 +14,26 @@ 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=...")
 
+# 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',
+            # Add these options to mimic browser behavior
             '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
+            'cookiefile': 'youtube.com_cookies.txt',  # 쿠키 파일 사용
+            'socket_timeout': 30,
+            'http_chunk_size': 10485760,  # 10MB
         }
 
-        # Add cookies file if available
-        if has_cookies:
-            ydl_opts['cookiefile'] = COOKIES_FILE
-
         # Progress placeholder
         progress_bar = st.progress(0)
         status_text = st.empty()
@@ -395,18 +85,20 @@ if st.button("Download"):
         st.warning("⚠️ Please enter a valid YouTube URL.")
 
 # Help section
-with st.expander("ℹ️ Help & Information"):
+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
+    **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
     
-    **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
+    2. Alternative Solutions:
+       - Try different videos
+       - Check if the video is public/not age-restricted
+       - Try again later if YouTube is blocking requests
     """)
 
 
@@ -425,69 +117,36 @@ with st.expander("ℹ️ Help & Information"):
 # 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=...")
+# # Check if cookies file exists
+# COOKIES_FILE = 'youtube.com_cookies.txt'
+# has_cookies = os.path.exists(COOKIES_FILE)
 
-# # Format selection
-# format_option = st.selectbox(
-#     "Select Quality",
-#     ["Best Quality", "720p", "480p", "360p", "Audio Only"]
-# )
+# if has_cookies:
+#     st.success("✅ Cookie file detected")
+# else:
+#     st.warning("⚠️ No cookie file found - Some videos might be restricted")
 
-# 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'
-#         }
+# # 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_choice):
+# def download_video(url):
 #     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,
-#             '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',
-#             }
+#             '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 format-specific options
-#         ydl_opts.update(get_format_options(format_choice))
-        
-#         # Progress tracking
+
+#         # Add cookies file if available
+#         if has_cookies:
+#             ydl_opts['cookiefile'] = COOKIES_FILE
+
+#         # Progress placeholder
 #         progress_bar = st.progress(0)
 #         status_text = st.empty()
         
@@ -498,34 +157,20 @@ with st.expander("ℹ️ Help & Information"):
 #                     progress_bar.progress(progress)
 #                     status_text.text(f"Downloading: {progress:.1%}")
 #                 except:
-#                     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)")
+#                     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:
-#             # 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
+#             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)}")
+#         st.error(f"An error occurred: {str(e)}")
 #         return None
 
 # # Download button
@@ -533,28 +178,19 @@ with st.expander("ℹ️ Help & Information"):
 #     if video_url:
 #         try:
 #             with st.spinner("Preparing download..."):
-#                 downloaded_file_path = download_video(video_url, format_option)
+#                 downloaded_file_path = download_video(video_url)
                 
 #             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=f"Download ({file_size:.1f} MB)",
+#                         label="Click here to download",
 #                         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 a different video or quality setting.")
+#                 st.error("❌ Download failed. Please try again.")
 #         except Exception as e:
 #             st.error(f"❌ An error occurred: {str(e)}")
 #     else:
@@ -563,17 +199,14 @@ with st.expander("ℹ️ Help & Information"):
 # # Help section
 # with st.expander("ℹ️ Help & Information"):
 #     st.markdown("""
-#     **Features:**
-#     - Multiple quality options available
-#     - Supports both video and audio-only downloads
-#     - Automatic format selection based on availability
+#     **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. 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.
-#     """)
+#     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
+#     """)