feat: add youtbue url download

app.py

@@ -12,6 +12,8 @@ from pathlib import Path
 import spaces
 from pydub.exceptions import CouldntEncodeError
 from transformers import pipeline
+import requests
+from bs4 import BeautifulSoup
 
 # Initialize text generation model
 model = pipeline('text-generation', model='EleutherAI/gpt-neo-125M')
@@ -90,6 +92,58 @@ def standardize_title(input_title):
 
     return standardized_output.strip()
 
+def download_youtube_audio(youtube_url: str, output_dir: str = './download', delete_existing: bool = True, simulate: bool = False) -> str:
+    """
+    Downloads audio from a YouTube URL and saves it as an MP3 file with specified yt-dlp options.
+    
+    Args:
+        youtube_url (str): URL of the YouTube video.
+        output_dir (str): Directory to save the downloaded audio file.
+        delete_existing (bool): If True, deletes any existing file with the same name.
+        simulate (bool): If True, simulates the download without actually downloading.
+
+    Returns:
+        str: Path to the downloaded audio file.
+    """
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+
+    r = requests.get(youtube_url)
+    soup = BeautifulSoup(r.text)
+
+    link = soup.find_all(name="title")[0]
+    title = str(link)
+    title = title.replace("<title>","")
+    title = title.replace("</title>","")
+        
+    audio_file = os.path.join(output_dir, title)
+
+    # Remove existing file if requested
+    if delete_existing and os.path.exists(audio_file + '.mp3'):
+        os.remove(audio_file + '.mp3')
+        
+    # Prepare yt-dlp command
+    command = [
+        'yt-dlp', '-x', youtube_url, '-f', 'bestaudio',
+        '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
+        '--extractor-retries', '10', '--force-overwrites', '--username', 'oauth2', '--password', '', '-v'
+    ]
+    
+    if simulate:
+        command.append('-s')
+
+    # Run the download command
+    process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
+
+    for line in iter(process.stdout.readline, ''):
+        print(line, end='')  # Print download progress
+        
+    process.stdout.close()
+    process.wait()
+    
+    return audio_file + '.mp3'
+
+
 def handle_file_upload(file):
     """
     Handle file upload, standardize the filename, change extension to .wav, and copy it to the input folder.
@@ -302,7 +356,7 @@ def delete_folders_and_files(input_dir):
     print("Cleanup completed.")
 
 @spaces.GPU(duration=160) 
-def process_audio(uploaded_file):
+def process_audio(uploaded_file, link):
     """
     Main function to process the uploaded audio file.
     
@@ -319,6 +373,9 @@ def process_audio(uploaded_file):
             input_path, formatted_title = handle_file_upload(uploaded_file)
             if input_path is None:
                 raise ValueError("File upload failed.")
+        elif link:
+            new_file = download_youtube_audio(link)
+            input_path, formatted_title = handle_file_upload(new_file)
         else:
             raise ValueError("Please upload a WAV file.")
 
@@ -363,6 +420,7 @@ def process_audio(uploaded_file):
 with gr.Blocks() as demo:
     gr.Markdown("# Music Player and Processor")
 
+    youtube_url = gr.Textbox(label="URL Youtube Song")
     file_upload = gr.File(label="Upload MP3 file", file_types=[".mp3"])
     process_button = gr.Button("Process Audio")
     log_output = gr.Textbox(label="Processing Log", interactive=False)
@@ -370,7 +428,7 @@ with gr.Blocks() as demo:
 
     process_button.click(
         fn=process_audio,
-        inputs=file_upload,
+        inputs=[file_upload, youtube_url],
         outputs=[log_output, processed_audio_output],
         show_progress=True
     )