Update app.py

app.py

@@ -156,15 +156,17 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
 
             if last_mp3:
                 print("Last .mp3 file:", last_mp3)
-                return last_mp3
+                instrumental_mp3_path = "./output/vocoder/stems/instrumental.mp3"
+                vocal_mp3_path = "./output/vocoder/stems/vocal.mp3"
+                return last_mp3, instrumental_mp3_path, vocal_mp3_path
             else:
-                return None
+                return None, None, None
         else:
             print("Output folder is empty.")
-            return None
+            raise gr.Error(f"Error occurred: Output folder is empty.")
     except subprocess.CalledProcessError as e:
         print(f"Error occurred: {e}")
-        return None
+        raise gr.Error(f"Error occurred: {e}")
     finally:
         # Clean up temporary files
         os.remove(genre_txt_path)
@@ -225,6 +227,8 @@ with gr.Blocks() as demo:
                 
                 submit_btn = gr.Button("Submit")
                 music_out = gr.Audio(label="Audio Result")
+                instrumental = gr.Audio(label="Intrumental")
+                vocal = gr.Audio(label="Vocal")
 
         gr.Examples(
             examples = [
@@ -267,6 +271,6 @@ Living out my dreams with this mic and a deal"""
     submit_btn.click(
         fn = infer, 
         inputs = [genre_txt, lyrics_txt, num_segments, max_new_tokens],
-        outputs = [music_out]
+        outputs = [music_out, instrumental, vocal]
     )
 demo.queue().launch(show_api=False, show_error=True)