Update app.py

app.py

@@ -1,14 +1,34 @@
-import json
+from TTS.api import TTS
 import gradio as gr
 from gradio import Dropdown
+from scipy.io.wavfile import write
 import os
 import shutil
 import re
 user_choice = ""
-
+MAX_NUMBER_SENTENCES = 10
+file_upload_available = os.environ.get("ALLOW_FILE_UPLOAD")
 tts = TTS("tts_models/multilingual/multi-dataset/bark", gpu=True)
 
 
+def split_process(audio, chosen_out_track):
+    gr.Info("Cleaning your audio sample...")
+    os.makedirs("out", exist_ok=True)
+    write('test.wav', audio[0], audio[1])
+    os.system("python3 -m demucs.separate -n mdx_extra_q -j 4 test.wav -o out")
+    # return "./out/mdx_extra_q/test/vocals.wav","./out/mdx_extra_q/test/bass.wav","./out/mdx_extra_q/test/drums.wav","./out/mdx_extra_q/test/other.wav"
+    if chosen_out_track == "vocals":
+        print("Audio sample cleaned")
+        return "./out/mdx_extra_q/test/vocals.wav"
+    elif chosen_out_track == "bass":
+        return "./out/mdx_extra_q/test/bass.wav"
+    elif chosen_out_track == "drums":
+        return "./out/mdx_extra_q/test/drums.wav"
+    elif chosen_out_track == "other":
+        return "./out/mdx_extra_q/test/other.wav"
+    elif chosen_out_track == "all-in":
+        return "test.wav"
+
 
 def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
     print("""
@@ -127,6 +147,17 @@ span.record-icon > span.dot.svelte-1thnwz {
       transform: rotate(360deg);
   }
 }
+#share-btn-container {
+  display: flex; 
+  padding-left: 0.5rem !important; 
+  padding-right: 0.5rem !important; 
+  background-color: #000000; 
+  justify-content: center; 
+  align-items: center; 
+  border-radius: 9999px !important; 
+  max-width: 15rem;
+  height: 36px;
+}
 """