+rtsp

app.py

@@ -56,6 +56,9 @@ def download_audio(url, method_choice):
         if 'youtube.com' in parsed_url.netloc or 'youtu.be' in parsed_url.netloc:
             # Use YouTube download methods
             audio_file = download_youtube_audio(url, method_choice)
+        elif parsed_url.scheme == 'rtsp':
+            # Use RTSP download methods
+            audio_file = download_rtsp_audio(url)
         else:
             # Use direct download methods
             audio_file = download_direct_audio(url, method_choice)
@@ -65,6 +68,27 @@ def download_audio(url, method_choice):
     except Exception as e:
         logging.error(f"Error downloading audio: {str(e)}")
         return f"Error: {str(e)}", False
+
+def download_rtsp_audio(url):
+    """
+    Downloads audio from an RTSP URL using FFmpeg.
+
+    Args:
+        url (str): The RTSP URL.
+
+    Returns:
+        str: Path to the downloaded audio file, or None if failed.
+    """
+    logging.info("Using FFmpeg to download RTSP stream")
+    output_file = tempfile.mktemp(suffix='.mp3')
+    command = ['ffmpeg', '-i', url, '-acodec', 'libmp3lame', '-ab', '192k', '-y', output_file]
+    try:
+        subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        logging.info(f"Downloaded RTSP audio to: {output_file}")
+        return output_file
+    except Exception as e:
+        logging.error(f"Error downloading RTSP audio: {str(e)}")
+        return None
         
 def download_youtube_audio(url, method_choice):
     """
@@ -113,38 +137,36 @@ def youtube_dl_method(url):
         logging.error(f"Error in youtube_dl_method: {str(e)}")
         return None
 
-def yt_dlp_method(url):
+def yt_dlp_direct_method(url):
     """
-    Downloads audio using yt-dlp.
+    Downloads audio using yt-dlp (supports various protocols and sites).
 
     Args:
-        url (str): The YouTube URL.
+        url (str): The URL of the audio or webpage containing audio.
 
     Returns:
         str: Path to the downloaded audio file, or None if failed.
     """
-    logging.info("Using yt-dlp method")
+    logging.info("Using yt-dlp direct method")
+    output_file = tempfile.mktemp(suffix='.mp3')
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'outtmpl': output_file,
+        'quiet': True,
+        'no_warnings': True,
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192',
+        }],
+    }
     try:
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
-            'outtmpl': '%(id)s.%(ext)s',
-            'quiet': True,
-            'no_warnings': True,
-        }
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=True)
-            output_file = ydl.prepare_filename(info)
-            if output_file.endswith('.webm') or output_file.endswith('.mp4'):
-                output_file = output_file.rsplit('.', 1)[0] + '.mp3'
-            logging.info(f"Downloaded YouTube audio: {output_file}")
-            return output_file
+            ydl.download([url])
+        logging.info(f"Downloaded audio to: {output_file}")
+        return output_file
     except Exception as e:
-        logging.error(f"Error in yt_dlp_method: {str(e)}")
+        logging.error(f"Error in yt_dlp_direct_method: {str(e)}")
         return None
         
 def pytube_method(url):
@@ -252,11 +274,11 @@ def requests_method(url):
         
 def download_direct_audio(url, method_choice):
     """
-    Downloads audio from a direct URL using the specified method.
+    Downloads audio from a direct URL or podcast URL using the specified method.
 
     Args:
         url (str): The direct URL of the audio file.
-        method_choice (str): The method to use for downloading ('wget', 'requests').
+        method_choice (str): The method to use for downloading ('wget', 'requests', 'yt-dlp', 'ffmpeg', 'aria2').
 
     Returns:
         str: Path to the downloaded audio file, or None if failed.
@@ -265,6 +287,9 @@ def download_direct_audio(url, method_choice):
     methods = {
         'wget': wget_method,
         'requests': requests_method,
+        'yt-dlp': yt_dlp_direct_method,
+        'ffmpeg': ffmpeg_method,
+        'aria2': aria2_method,
     }
     method = methods.get(method_choice)
     if method is None: