add Youtube

link_processor.py

@@ -1,11 +1,9 @@
 from pathlib import Path
 import yt_dlp
-import tempfile
 import streamlit as st
-from video_processor import VideoProcessor
-from audio_processor import AudioProcessor
 import re
 import os
+import tempfile
 
 
 class LinkDownloader:
@@ -15,9 +13,7 @@ class LinkDownloader:
 
     def sanitize_filename(self, title):
         """Sanitize filename by removing special characters and limiting length"""
-        # Replace special characters with underscore
         clean_name = re.sub(r'[^a-zA-Z0-9.]', '_', title)
-        # Limit length
         if len(clean_name) > 50:
             clean_name = clean_name[:50]
         return clean_name
@@ -25,44 +21,55 @@ class LinkDownloader:
     def download_from_url(self, url):
         """Download video from URL using yt-dlp"""
         try:
-            # Create a temporary filename template
             temp_filename = 'downloaded_video.%(ext)s'
             temp_filepath = str(self.output_dir / temp_filename)
 
-            # Configure yt-dlp options
+            # Configure yt-dlp options with more robust authentication handling
             ydl_opts = {
                 'format': 'best',
                 'outtmpl': temp_filepath,
-                'quiet': False,  # Enable output for debugging
+                'quiet': False,
                 'no_warnings': False,
                 'extract_audio': False,
+                'ignoreerrors': True,  # Continue on download errors
+                'no_check_certificate': True,  # Skip HTTPS certificate validation
+                'username': None,  # Will be filled by user if needed
+                'password': None,  # Will be filled by user if needed
             }
 
-            st.info("Starting video download...")
-            print(f"Download directory: {self.output_dir}")
-            print(f"Download template: {temp_filepath}")
+            # First try without authentication
+            try:
+                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                    info = ydl.extract_info(url, download=False)
+                    if info:
+                        ydl.download([url])
+            except yt_dlp.utils.DownloadError as e:
+                if "Sign in" in str(e):
+                    # If authentication is required, prompt for credentials
+                    st.warning("Authentication required. Please enter your YouTube credentials.")
+                    username = st.text_input("YouTube Email", key="yt_email")
+                    password = st.text_input("YouTube Password", type="password", key="yt_pass")
 
-            # Download the video
-            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                # Extract info first
-                info = ydl.extract_info(url, download=False)
-                # Get the actual extension
-                ext = info.get('ext', 'mp4')
+                    if username and password:
+                        ydl_opts['username'] = username
+                        ydl_opts['password'] = password
 
-                # Download the video
-                ydl.download([url])
+                        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                            info = ydl.extract_info(url, download=True)
+                else:
+                    raise e
 
-                # Construct the actual file path
-                actual_filepath = str(self.output_dir / f"downloaded_video.{ext}")
-                print(f"Expected downloaded file: {actual_filepath}")
+            # Get the actual extension and filepath
+            ext = info.get('ext', 'mp4')
+            actual_filepath = str(self.output_dir / f"downloaded_video.{ext}")
 
-                # Verify file exists
-                if not os.path.exists(actual_filepath):
-                    print(f"Files in directory: {os.listdir(self.output_dir)}")
-                    raise FileNotFoundError(f"Downloaded file not found at {actual_filepath}")
+            # Verify file exists
+            if not os.path.exists(actual_filepath):
+                print(f"Files in directory: {os.listdir(self.output_dir)}")
+                raise FileNotFoundError(f"Downloaded file not found at {actual_filepath}")
 
-                print(f"File size: {os.path.getsize(actual_filepath)} bytes")
-                return actual_filepath
+            print(f"File size: {os.path.getsize(actual_filepath)} bytes")
+            return actual_filepath
 
         except Exception as e:
             st.error(f"Error during download: {str(e)}")

requirements.txt

@@ -1,10 +1,10 @@
-numpy==1.23.5
-pandas==1.5.3
-streamlit==1.29.0
-torch==2.1.0
-torchaudio==2.1.0
+numpy==1.26.3
+pandas==2.2.0
+streamlit==1.30.0
+torch==2.1.2
+torchaudio==2.1.2
 demucs==4.0.1
 soundfile==0.12.1
 diffq==0.2.4
 ffmpeg-python==0.2.0
-yt-dlp==2024.12.23
+yt-dlp==2024.1.7