Update app.py

app.py

@@ -97,7 +97,7 @@ def wipe_npz_file(folder_path):
     print("YO")
     
 def split_process(audio, chosen_out_track):
-    gr.Info("Cleaning your audio sample")
+    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")
@@ -123,7 +123,11 @@ def update_selection(selected_state: gr.SelectData):
 
     
 def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
-
+    print"""
+—————
+NEW INFERENCE:
+———————
+    """
     if clean_audio is True :
         print("We want to clean audio sample")
         # Extract the file name without the extension
@@ -184,12 +188,17 @@ def infer(prompt, input_wav_file, clean_audio, hidden_numpy_audio):
     # Print the contents
     for item in contents:
         print(item)  
-
+    gr.Info("Preparing final waveform video ...")
     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"""
+—————
+NEW INFERENCE:
+———————
+    """
     print(f"USING VOICE LIBRARY: {c_name}")
     # Split the text into sentences based on common punctuation marks
     sentences = re.split(r'(?<=[.!?])\s+', prompt)
@@ -200,11 +209,13 @@ def infer_from_c(prompt, c_name):
     # Join the selected sentences back into a single string
     limited_prompt = ' '.join(first_nb_sentences)
     
+    gr.Info(f"Generating audio from prompt with {c_name} ;)")
     tts.tts_to_file(text=limited_prompt,
                 file_path="output.wav",
                 voice_dir="examples/library/",
                 speaker=f"{c_name}")
-
+    
+    gr.Info("Preparing final waveform video ...")
     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)