Update app.py

app.py

@@ -1,9 +1,53 @@
 import gradio as gr
 import os
 import bark
+from bark import generate_audio, preload_models, SAMPLE_RATE
+import time
+import scipy
+import noisereduce as nr
+
+########################
+# Voice cloning functions
+
+# make sure to only use CPU
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
+os.environ["SUNO_USE_SMALL_MODELS"] = "1"
+
+preload_models()
+
+
+def generate_cloned_voice_audio(text_prompt, history_prompt):
+
+
+    # keep track of duration 
+    t0 = time.time()
+
+    # generate cloned voice audio
+    audio_array = generate_audio(
+        text_prompt,
+        history_prompt = history_prompt
+    )
+
+    # keep track of duration 
+    generation_duration_s = time.time() - t0
+    audio_duration_s = audio_array.shape[0] / SAMPLE_RATE
+    print(f"took {generation_duration_s:.0f}s to generate {audio_duration_s:.0f}s of audio")
+
+    # reduce noise
+    reduced_noise_audio_array = nr.reduce_noise(y=audio_array, sr=SAMPLE_RATE)
+    
+    # write to file
+    audio_output_path = "output_audio.wav"
+    noisereduced_audio_output_path = "output_noisereduced_audio.wav"
+    scipy.io.wavfile.write(audio_output_path, rate=SAMPLE_RATE, data=audio_array)
+    scipy.io.wavfile.write(noisereduced_audio_output_path, rate=SAMPLE_RATE, data=reduced_noise_audio_array)
+
+    return audio_output_path
+
+########################
 
 def greet(name):
-    if os.paht.isfile("pm_voice.npz"):
+    if os.path.isfile("pm_voice.npz"):
         preffix = "Found the voice file"
     else:
         preffix = "Voice file not found"