1.update genre file write 2.update mix track logic for vocal and instrument

YuE/inference/infer.py

@@ -454,6 +454,8 @@ vocoder_stems_dir = os.path.join(vocoder_output_dir, 'stems')
 vocoder_mix_dir = os.path.join(vocoder_output_dir, 'mix')
 os.makedirs(vocoder_mix_dir, exist_ok=True)
 os.makedirs(vocoder_stems_dir, exist_ok=True)
+vocal_output = None
+instrumental_output = None
 for npy in stage2_result:
     if '_itrack' in npy:
         # Process instrumental
@@ -477,10 +479,13 @@ for npy in stage2_result:
         )
 # mix tracks
 try:
-    mix_output = instrumental_output + vocal_output
-    vocoder_mix = os.path.join(vocoder_mix_dir, os.path.basename(recons_mix))
-    save_audio(mix_output, vocoder_mix, 44100, args.rescale)
-    print(f"Created mix: {vocoder_mix}")
+    if vocal_output is not None and instrumental_output is not None:
+        mix_output = instrumental_output + vocal_output
+        vocoder_mix = os.path.join(vocoder_mix_dir, os.path.basename(recons_mix))
+        save_audio(mix_output, vocoder_mix, 44100, args.rescale)
+        print(f"Created mix: {vocoder_mix}")
+    else:
+        print("Error: Missing vocal or instrumental track for mixing")
 except RuntimeError as e:
     print(e)
     print(f"mix {vocoder_mix} failed! inst: {instrumental_output.shape}, vocal: {vocal_output.shape}")

app.py

@@ -190,7 +190,8 @@ def generate_music_spaces(lyrics: str, genre: str, mood: str, progress=gr.Progre
         
         # Create temporary files
         with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as genre_file:
-            genre_file.write(f"instrumental,{genre},{mood},male vocals")
+            # genre_file.write(f"instrumental,{genre},{mood},male vocals")
+            genre_file.write(f"{mood} female vocal {genre} vocal instrumental")
             genre_file_path = genre_file.name
         
         # Convert lyrics format for YuE compatibility