Update app.py

app.py

@@ -1,3 +1,202 @@
+# 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
@@ -16,77 +215,24 @@ 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
-        
+# Function to download video
+def download_video(url):
     try:
-        format_opts = get_format_options(format_selection)
-        
         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,
-            '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
+            'cookiefile': 'youtube.com_cookies.txt',  # 쿠키 파일 사용
+            'socket_timeout': 30,
+            'http_chunk_size': 10485760,  # 10MB
         }
-        
-        # Update options with format-specific settings
-        ydl_opts.update(format_opts)
 
-        # Progress elements
+        # Progress placeholder
         progress_bar = st.progress(0)
         status_text = st.empty()
         
@@ -104,36 +250,13 @@ def download_video(url, format_selection):
 
         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
+            info = ydl.extract_info(url, download=True)
+            filename = ydl.prepare_filename(info)
+            return filename
 
     except Exception as e:
-        st.error(f"An error occurred during setup: {str(e)}")
+        st.error(f"An error occurred: {str(e)}")
         return None
 
 # Download button
@@ -141,55 +264,37 @@ if st.button("Download"):
     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):
-                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)}")
+                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 with a different video or format.")
+                st.error("❌ Download failed. Please try again.")
         except Exception as e:
-            st.error(f"❌ An unexpected error occurred: {str(e)}")
+            st.error(f"❌ An error occurred: {str(e)}")
     else:
         st.warning("⚠️ Please enter a valid YouTube URL.")
 
-# Add help section
-with st.expander("ℹ️ Help"):
+# Help section
+with st.expander("ℹ️ Help & Troubleshooting"):
     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
+    **If you're experiencing download issues:**
     
-    **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()
+    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
+    """)