refine uvr job

inst.py

@@ -22,6 +22,11 @@ from separate import (
 
 
 def run_ensemble_models(audio_path, export_path, format=WAV, clean=True):
+    vocals_final_path = Path(export_path) / f"{Path(audio_path).stem}.vocal.{format.lower()}"
+    instrumental_final_path = Path(export_path) / f"{Path(audio_path).stem}.instrumental.{format.lower()}"
+    if os.path.isfile(instrumental_final_path) and os.path.isfile(vocals_final_path):
+        return instrumental_final_path, vocals_final_path
+
     start = datetime.now()
     process_datas = [mdx23c_8kfft_instvoc_hq_process_data, uvr_mdx_net_voc_ft_process_data,
                      htdemucs_ft_process_data]
@@ -77,9 +82,6 @@ def run_ensemble_models(audio_path, export_path, format=WAV, clean=True):
         vocals_export_paths.append(str(vocals_path))
 
     # merge each model outputs
-    vocals_final_path = Path(export_path) / f"{Path(audio_path).stem}.vocal.{format.lower()}"
-    instrumental_final_path = Path(export_path) / f"{Path(audio_path).stem}.instrumental.{format.lower()}"
-
     ensemble(vocals_export_paths, vocals_final_path)
     ensemble(instrumental_export_paths, instrumental_final_path)
 
@@ -106,9 +108,10 @@ def uvr_job(song_id, platform='netease'):
     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)
+    if not os.path.isfile(audio_path):
+        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

requirements.txt

@@ -1,7 +1,7 @@
 altgraph==0.17.3
 audioread==3.0.0
 certifi==2022.12.07
-cffi==1.15.1
+cffi==1.16.0
 cryptography==3.4.6
 einops==0.6.0
 future==0.18.3