use rapid api

app.py

@@ -6,31 +6,61 @@ import json
 import subprocess
 import os
 import tempfile  # Import tempfile
+from pydub import AudioSegment  # Import AudioSegment
 
 # Define the FastAPI URL
 API_URL = "http://astarwiz.com:9998"
+rapid_key = os.environ.get("RAPID_API_KEY")
 
-def download_youtube_audio(url: str, output_dir: Optional[str] = None) -> str:
+def download_youtube_audio(metadata, output_dir: Optional[str] = None) -> str:
+    video_id = metadata['id']
+    
     if output_dir is None:
         output_dir = tempfile.gettempdir()
 
-    ydl_opts = {
-        'format': 'bestaudio/best',
-        'extractaudio': True,  # Download only audio
-        'audioformat': 'mp3',  # Convert to mp3
-        'outtmpl': os.path.join(output_dir, '%(title)s.%(ext)s'),  # Output filename
-        'noplaylist': True,  # Only download single video, not playlist
+    output_filename = os.path.join(output_dir, f"{video_id}.mp3")
+    
+    if os.path.exists(output_filename):
+        return output_filename  # Return if the file already exists
+    
+    url = "https://youtube86.p.rapidapi.com/api/youtube/links"
+    headers = {
+        'Content-Type': 'application/json',
+        'x-rapidapi-host': 'youtube86.p.rapidapi.com',
+        'x-rapidapi-key': rapid_key  # Replace <key> with your actual API key
     }
-
-    try:
-        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
-            info_dict = ydl.extract_info(url, download=True)  # Download audio
-            audio_file = os.path.join(output_dir, f"{info_dict['title']}.mp3")  # Construct the file path
-
-        print(f"Successfully downloaded: {audio_file}")
-        return audio_file
-    except Exception as e:
-        raise Exception(f"Error downloading YouTube audio: {str(e)}")
+    data = {
+        "url": metadata['url']
+    }
+    
+    response = requests.post(url, headers=headers, json=data)
+    print('Fetched audio links')
+    
+    if response.status_code == 200:
+        result = response.json()
+        for url in result[0]['urls']:
+            if url.get('isBundle'):
+                audio_url = url['url']
+                extension = url['extension']
+                audio_response = requests.get(audio_url)
+                
+                if audio_response.status_code == 200:
+                    temp_filename = os.path.join(output_dir, f"{video_id}.{extension}")
+                    with open(temp_filename, 'wb') as audio_file:
+                        audio_file.write(audio_response.content)
+                    
+                    # Convert to MP3 and downsample to 16000 Hz
+                    audio = AudioSegment.from_file(temp_filename, format=extension)
+                    audio = audio.set_frame_rate(16000)
+                    audio.export(output_filename, format="mp3", parameters=["-ar", "16000"])
+                    
+                    os.remove(temp_filename)  # Remove the temporary file
+                    return output_filename  # Return the final MP3 filename
+        
+        return None  # Return None if no successful download occurs
+    else:
+        print("Error:", response.status_code, response.text)
+        return None  # Return None on failure
 
 def run_asr(audio_file, youtube_url):
     temp_file = None

requirements.txt

@@ -1 +1 @@
-youtube_dl
+pydub