added youtube song import

app.py

@@ -12,13 +12,74 @@ import gradio as gr
 from scipy.io.wavfile import write
 
 from audiocraft.models import MusicGen
-
+import tempfile
 import os
 from audiocraft.data.audio import audio_write
 
 
 MODEL = None
 
+import yt_dlp as youtube_dl
+from moviepy.editor import VideoFileClip
+
+YT_LENGTH_LIMIT_S = 480  # limit to 1 hour YouTube files
+
+def download_yt_audio(yt_url, filename):
+    info_loader = youtube_dl.YoutubeDL()
+    
+    try:
+        info = info_loader.extract_info(yt_url, download=False)
+    except youtube_dl.utils.DownloadError as err:
+        raise gr.Error(str(err))
+    
+    file_length = info["duration_string"]
+    file_h_m_s = file_length.split(":")
+    file_h_m_s = [int(sub_length) for sub_length in file_h_m_s]
+    
+    if len(file_h_m_s) == 1:
+        file_h_m_s.insert(0, 0)
+    if len(file_h_m_s) == 2:
+        file_h_m_s.insert(0, 0)
+    file_length_s = file_h_m_s[0] * 3600 + file_h_m_s[1] * 60 + file_h_m_s[2]
+    
+    if file_length_s > YT_LENGTH_LIMIT_S:
+        yt_length_limit_hms = time.strftime("%HH:%MM:%SS", time.gmtime(YT_LENGTH_LIMIT_S))
+        file_length_hms = time.strftime("%HH:%MM:%SS", time.gmtime(file_length_s))
+        raise gr.Error(f"Maximum YouTube length is {yt_length_limit_hms}, got {file_length_hms} YouTube video.")
+    
+    ydl_opts = {"outtmpl": filename, "format": "worstvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best"}
+    
+    with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+        try:
+            ydl.download([yt_url])
+        except youtube_dl.utils.ExtractorError as err:
+            raise gr.Error(str(err))
+
+
+def convert_to_mp3(input_path, output_path):
+    try:
+        video_clip = VideoFileClip(input_path)
+        audio_clip = video_clip.audio
+        print("Converting to MP3...")
+        audio_clip.write_audiofile(output_path)
+    except Exception as e:
+        print("Error:", e)
+
+def load_youtube_audio(yt_link):
+    
+    gr.Info("Loading your YouTube link ... ")
+    with tempfile.TemporaryDirectory() as tmpdirname:
+        filepath = os.path.join(tmpdirname, "video.mp4")
+        download_yt_audio(yt_link, filepath)
+
+        mp3_output_path = "video_sound.mp3"
+        convert_to_mp3(filepath, mp3_output_path)
+        print("Conversion complete. MP3 saved at:", mp3_output_path)
+    
+    
+        
+    return mp3_output_path
+
 def split_process(audio, chosen_out_track):
     os.makedirs("out", exist_ok=True)
     write('test.wav', audio[0], audio[1])
@@ -99,6 +160,10 @@ with gr.Blocks(css=css) as demo:
             
         with gr.Column():
             uploaded_sound = gr.Audio(type="numpy", label="Input", source="upload")
+            with.gr.Row():
+                youtube_link = gr.Textbox(show_label=False, placeholder="you can also paste YT link and load it")
+                yt_load_btn = gr.Button("Load YT song")
+                
             with gr.Row():
                 chosen_track = gr.Radio(["vocals", "bass", "drums", "other", "all-in"], label="Track", info="Which track from your audio do you want to mashup ?", value="vocals")
             load_sound_btn = gr.Button('Load your chosen track')
@@ -161,6 +226,7 @@ with gr.Blocks(css=css) as demo:
             inputs=[music_prompt, melody, duration, cfg_coef],
             outputs=[output]
         )
+    yt_load_btn.click(fn=load_youtube_audio, inputs=[youtube_link], outputs=[uploaded_sound], queue=False)
     load_sound_btn.click(split_process, inputs=[uploaded_sound, chosen_track], outputs=[melody], api_name="splt_trck")
     submit.click(predict, inputs=[music_prompt, melody, duration, cfg_coef], outputs=[output])