add uvr job

inst.py

@@ -5,6 +5,7 @@ from datetime import datetime
 from pathlib import Path
 from time import sleep
 
+import requests
 from tqdm import tqdm
 
 from args import mdx23c_8kfft_instvoc_hq_process_data, htdemucs_ft_process_data, uvr_mdx_net_voc_ft_process_data
@@ -100,6 +101,19 @@ def ensemble(stem_outputs, stem_save_path, format=WAV):
     save_format(stem_save_path, format, '320k')
 
 
+def uvr_job(song_id, platform='netease'):
+    audio_dir = os.getcwd()
+    audio_file = f'{song_id}.m4a' if platform == 'youtube' else f'{song_id}.mp3'
+    audio_path = os.path.join(audio_dir, audio_file)
+
+    url = f"http://or.luotao.net/api/download_song?song_id={song_id}&platform={platform}"
+    r = requests.get(url, allow_redirects=True)
+    open(audio_path, 'wb').write(r.content)
+
+    instrumental_path, vocals_path = run_ensemble_models(audio_file, audio_dir)
+    return instrumental_path
+
+
 # /Users/taoluo/Downloads/test/kimk_audio_MDX23C-8KFFT-InstVoc_HQ_(Instrumental).WAV
 #
 if __name__ == '__main__':