Fix import issue

video_accent_analyzer.py

@@ -87,6 +87,10 @@ class VideoAccentAnalyzer:
         self.model_loaded = False
         self._load_model()
 
+
+
+
+
     def _load_model(self):
         """Load the accent classification model with error handling"""
         print("🤖 Loading accent classification model...")
@@ -128,8 +132,29 @@ class VideoAccentAnalyzer:
             print(f"❌ Trimming exception: {e}")
             return input_path
 
+    def _get_youtube_cookies(self):
+        """Get YouTube cookies from browser"""
+        import browser_cookie3
+
+        try:
+            # Try Firefox first
+            cookies = browser_cookie3.firefox(domain_name='.youtube.com')
+        except:
+            try:
+                # Try Chrome as fallback
+                cookies = browser_cookie3.chrome(domain_name='.youtube.com')
+            except:
+                print("⚠️ Could not get cookies from browser")
+                return None
+
+        return cookies
+
+
+
+
+
     def download_video(self, url, max_duration=None):
-        """Download video using yt-dlp with improved error handling"""
+        """Download video using yt-dlp with cookie support"""
         is_valid, result = self._validate_url(url)
         if not is_valid:
             print(f"❌ {result}")
@@ -138,23 +163,28 @@ class VideoAccentAnalyzer:
         url = result
         output_path = os.path.join(self.temp_dir, "video.%(ext)s")
 
-        # Enhanced yt-dlp options for better compatibility
+        # Enhanced yt-dlp options
         ydl_opts = {
             'outtmpl': output_path,
-            'format': 'worst[ext=mp4]/worst',  # Use worst quality for faster download and better compatibility
-            'quiet': False,  # Show some output for debugging
+            'format': 'worst[ext=mp4]/worst',
+            'quiet': False,
             'no_warnings': False,
             'socket_timeout': 60,
-            'retries': 5,
-            'fragment_retries': 5,
-            'extract_flat': False,
-            'writeinfojson': False,
-            'writethumbnail': False,
-            'writesubtitles': False,
+            'retries': 5
         }
 
+        # Add cookies for YouTube URLs
+        if 'youtube.com' in url or 'youtu.be' in url:
+            cookies = self._get_youtube_cookies()
+            if cookies:
+                cookie_file = os.path.join(self.temp_dir, 'cookies.txt')
+                with open(cookie_file, 'w') as f:
+                    f.write('# Netscape HTTP Cookie File\n')
+                    for cookie in cookies:
+                        f.write(f'.youtube.com\tTRUE\t/\tFALSE\t{cookie.expires}\t{cookie.name}\t{cookie.value}\n')
+                ydl_opts['cookiefile'] = cookie_file
+
         if max_duration:
-            # More generous time limit for download
             ydl_opts['match_filter'] = lambda info: None if info.get('duration', 0) <= 200000 else "Video too long"
 
         try: