added Infos and prints

app.py

@@ -97,11 +97,13 @@ def wipe_npz_file(folder_path):
     print("YO")
     
 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"
@@ -128,9 +130,11 @@ def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
         new_name = os.path.splitext(os.path.basename(input_wav_file))[0]
         print(f"FILE BASENAME is: {new_name}")
         if os.path.exists(os.path.join("bark_voices", f"{new_name}_cleaned")):
+            print("This file has already been cleaned")
             check_name = os.path.join("bark_voices", f"{new_name}_cleaned")
             source_path = os.path.join(check_name, f"{new_name}_cleaned.wav")
         else:
+            print("This file is new, we need to clean and store it")
             source_path = split_process(hidden_numpy_audio, "vocals")
         
             # Rename the file
@@ -138,6 +142,7 @@ def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
             os.rename(source_path, new_path)
             source_path = new_path
     else :
+        print("We do NOT want to clean audio sample")
         # Path to your WAV file
         source_path = input_wav_file
 
@@ -156,6 +161,8 @@ def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
     # Move the WAV file to the new directory
     shutil.move(source_path, os.path.join(destination_path, f"{file_name}.wav"))
 
+    # —————
+    
     # Split the text into sentences based on common punctuation marks
     sentences = re.split(r'(?<=[.!?])\s+', prompt)
     
@@ -164,7 +171,8 @@ def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
     
     # Join the selected sentences back into a single string
     limited_prompt = ' '.join(first_nb_sentences)
-    
+
+    gr.Info("Generating audio from prompt")
     tts.tts_to_file(text=limited_prompt,
                 file_path="output.wav",
                 voice_dir="bark_voices/",
@@ -178,11 +186,11 @@ def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
         print(item)  
 
     tts_video = gr.make_waveform(audio="output.wav")
-    
+    print("FINISHED")
     return "output.wav", tts_video, gr.update(value=f"bark_voices/{file_name}/{contents[1]}", visible=True), gr.Group.update(visible=True), destination_path
 
 def infer_from_c(prompt, c_name):
-
+    print(f"USING VOICE LIBRARY: {c_name}")
     # Split the text into sentences based on common punctuation marks
     sentences = re.split(r'(?<=[.!?])\s+', prompt)
     
@@ -198,7 +206,7 @@ def infer_from_c(prompt, c_name):
                 speaker=f"{c_name}")
 
     tts_video = gr.make_waveform(audio="output.wav")
-    
+    print("FINISHED")
     return "output.wav", tts_video, gr.update(value=f"examples/library/{c_name}/{c_name}.npz", visible=True), gr.Group.update(visible=True)