Update app.py

app.py

@@ -392,6 +392,7 @@ class AudioProcessor:
 class VideoProcessor:
     def __init__(self):
         self.supported_formats = ['.mp4', '.avi', '.mov', '.mkv']
+        self.cookie_file_path = "cookies.txt"  # Chemin du fichier cookies
         self.ydl_opts = {
             'format': 'bestaudio/best',
             'postprocessors': [{
@@ -403,7 +404,15 @@ class VideoProcessor:
         }
     
     
+    def load_cookies(self):
+        """Charge les cookies depuis Hugging Face et les enregistre localement."""
+        dataset = load_dataset("Adjoumani/YoutubeCookiesDataset")
+        cookies = dataset["train"]["cookies"][0]
+        with open(self.cookie_file_path, "w") as f:
+            f.write(cookies)
+        print(f"Cookies enregistrés dans {self.cookie_file_path}")
 
+        
     def extract_video_id(self, url: str) -> str:
         try:
             parsed_url = urlparse(url)
@@ -427,7 +436,7 @@ class VideoProcessor:
             ydl.download([url])
         return 'temp_audio.mp3'  """
 
-    def download_youtube_audio(self, url: str) -> str:
+    """def download_youtube_audio(self, url: str) -> str:
         try:
             # Fichier cookies
             cookie_file_path = "cookies.txt"
@@ -453,6 +462,25 @@ class VideoProcessor:
             if not os.path.exists(audio_path):
                 raise FileNotFoundError(f"Le fichier {audio_path} n'a pas été généré.")
     
+            return audio_path
+        except Exception as e:
+            raise RuntimeError(f"Erreur lors du téléchargement : {str(e)}")"""
+
+    def download_youtube_audio(self, url: str) -> str:
+        try:
+            # Ajoutez le fichier cookies dans les options
+            ydl_opts = self.ydl_opts.copy()
+            ydl_opts['cookiefile'] = self.cookie_file_path
+
+            # Téléchargement
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                ydl.download([url])
+
+            # Vérifier si le fichier audio existe
+            audio_path = 'temp_audio.mp3'
+            if not os.path.exists(audio_path):
+                raise FileNotFoundError(f"Le fichier {audio_path} n'a pas été généré.")
+
             return audio_path
         except Exception as e:
             raise RuntimeError(f"Erreur lors du téléchargement : {str(e)}")
@@ -843,6 +871,7 @@ def handle_youtube_input():
                 if transcription:
                     process_and_display_results(None, None, transcription)
                 else:
+                    video_processor.load_cookies()
                     audio_path = video_processor.download_youtube_audio(youtube_url)
                     process_and_display_results(audio_path)