Update app.py

app.py

@@ -45,11 +45,11 @@ logging.info(f"Using device: {device}")
 def download_audio(url, method_choice):
     """
     Downloads audio from a given URL using the specified method.
-
+    
     Args:
         url (str): The URL of the audio.
         method_choice (str): The method to use for downloading audio.
-
+    
     Returns:
         tuple: (path to the downloaded audio file, is_temp_file), or (None, False) if failed.
     """
@@ -58,20 +58,28 @@ def download_audio(url, method_choice):
     try:
         if 'youtube.com' in parsed_url.netloc or 'youtu.be' in parsed_url.netloc:
             audio_file = download_youtube_audio(url, method_choice)
+            if not audio_file:
+                error_msg = f"Failed to download audio from {url} using method {method_choice}. Ensure yt-dlp is up to date."
+                logging.error(error_msg)
+                return None, False
         elif parsed_url.scheme == 'rtsp':
             audio_file = download_rtsp_audio(url)
+            if not audio_file:
+                error_msg = f"Failed to download RTSP audio from {url}"
+                logging.error(error_msg)
+                return None, False
         else:
             audio_file = download_direct_audio(url, method_choice)
-        
-        if not audio_file or not os.path.exists(audio_file):
-            error_msg = f"Failed to download audio from {url} using method {method_choice}"
-            logging.error(error_msg)
-            return None, False
+            if not audio_file:
+                error_msg = f"Failed to download audio from {url} using method {method_choice}"
+                logging.error(error_msg)
+                return None, False
         return audio_file, True
     except Exception as e:
         error_msg = f"Error downloading audio from {url} using method {method_choice}: {str(e)}"
         logging.error(error_msg)
         return None, False
+        
 
 def download_youtube_audio(url, method_choice):
     """
@@ -96,13 +104,13 @@ def download_youtube_audio(url, method_choice):
         logging.error(f"Error downloading using {method_choice}: {str(e)}")
         return None
 
-def yt_dlp_method(url):
+ef yt_dlp_method(url):
     """
     Downloads YouTube audio using yt-dlp and saves it to a temporary file.
-
+    
     Args:
         url (str): The YouTube URL.
-
+    
     Returns:
         str: Path to the downloaded audio file, or None if failed.
     """
@@ -118,20 +126,51 @@ def yt_dlp_method(url):
             'preferredquality': '192',
         }],
         'quiet': False,
-        'no_warnings': True,
-        'logger': logging.getLogger(),  # Capture yt-dlp logs
+        'no_warnings': False,
+        'logger': MyLogger(),  # Use a custom logger to capture yt-dlp logs
+        'progress_hooks': [my_hook],  # Hook to capture download progress and errors
     }
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=True)
+            if 'entries' in info:
+                # Can be a playlist or a list of videos
+                info = info['entries'][0]
             output_file = ydl.prepare_filename(info)
             output_file = os.path.splitext(output_file)[0] + '.mp3'
-            logging.info(f"Downloaded YouTube audio: {output_file}")
-            return output_file
+            if os.path.exists(output_file):
+                logging.info(f"Downloaded YouTube audio: {output_file}")
+                return output_file
+            else:
+                error_msg = "yt-dlp did not produce an output file."
+                logging.error(error_msg)
+                return None
     except Exception as e:
         logging.error(f"yt-dlp failed to download audio: {str(e)}")
         return None
 
+class MyLogger(object):
+    """
+    Custom logger for yt-dlp to capture logs and errors.
+    """
+    def debug(self, msg):
+        logging.debug(msg)
+    def info(self, msg):
+        logging.info(msg)
+    def warning(self, msg):
+        logging.warning(msg)
+    def error(self, msg):
+        logging.error(msg)
+
+def my_hook(d):
+    """
+    Hook function to capture yt-dlp download progress and errors.
+    """
+    if d['status'] == 'finished':
+        logging.info('Download finished, now converting...')
+    elif d['status'] == 'error':
+        logging.error(f"Download error: {d['filename']}")
+
 def pytube_method(url):
     """
     Downloads audio from a YouTube URL using pytube and saves it to a temporary file.
@@ -460,7 +499,7 @@ def transcribe_audio(input_source, pipeline_type, model_id, dtype, batch_size, d
             # Input source is a URL
             audio_path, is_temp_file = download_audio(input_source, download_method)
             if not audio_path:
-                error_msg = f"Error downloading audio from {input_source} using method {download_method}"
+                error_msg = f"Error downloading audio from {input_source} using method {download_method}. Check logs for details."
                 logging.error(error_msg)
                 yield verbose_messages + error_msg, "", None
                 return