Update src/main.py

src/main.py

@@ -264,10 +264,12 @@ def song_cover_pipeline(main_vocals, backup_vocals, voice_model, pitch_change, k
         
         song_dir = os.path.join(output_dir, song_id)
         print(f"DEBUG PRINT: song_dir {song_dir}")
-
+        if not os.path.exists(song_dir):
+            os.makedirs(song_dir)
+            
         pitch_change = pitch_change * 12 + pitch_change_all
-        ai_vocals_path = os.path.join(song_dir, f'_{voice_model}_p{pitch_change}_i{index_rate}_fr{filter_radius}_rms{rms_mix_rate}_pro{protect}_{f0_method}{"" if f0_method != "mangio-crepe" else f"_{crepe_hop_length}"}.wav')
-        ai_cover_path = os.path.join(song_dir, f'({voice_model} Ver)_cover.{output_format}')
+        ai_vocals_path = os.path.join(song_dir, f'{os.path.basename(os.path.splitext(main_vocals)[0])}_{voice_model}_p{pitch_change}_i{index_rate}_fr{filter_radius}_rms{rms_mix_rate}_pro{protect}_{f0_method}{"" if f0_method != "mangio-crepe" else f"_{crepe_hop_length}"}.wav')
+        ai_cover_path = os.path.join(song_dir, f'{os.path.basename(os.path.splitext(main_vocals)[0])}({voice_model} Ver)_cover.{output_format}')
 
         print(f"DEBUG PRINT: ai_vocals_path: {ai_vocals_path}, ai_cover_path: {ai_cover_path}")