Update app.py

app.py

@@ -3,12 +3,25 @@ import re, unidecode
 from unidecode import unidecode
 import yt_dlp
 import os
+import pydub
+import numpy as np
 
 # no space, punctuation, accent in lower string
 def cleanString(string):
     cleanString = unidecode(string)
     cleanString = re.sub('\W+','_', cleanString)
-    return cleanString.lower()
+    return cleanString.lower
+
+def read_audio(f, normalized=False):
+    """MP3 to numpy array"""
+    a = pydub.AudioSegment.from_mp3(f)
+    y = np.array(a.get_array_of_samples())
+    if a.channels == 2:
+        y = y.reshape((-1, 2))
+    if normalized:
+        return a.frame_rate, np.float32(y) / 2**15
+    else:
+        return a.frame_rate, y
 
 def download_audio(url):
     path_to_folder_audio_mp3 = "./audio/"
@@ -31,12 +44,20 @@ def download_audio(url):
             if cleanString(local_link) == cleanString(filename):
                 os.rename(os.path.join(path_to_folder_audio_mp3, filename),os.path.join(path_to_folder_audio_mp3, new_local_link)) 
 
-    return path_to_folder_audio_mp3 + new_local_link 
+        file_path = path_to_folder_audio_mp3 + new_local_link
+
+        with open(file_path, "r") as f:
+            audio = 
+
+    return file_path, read_audio(file_path)
 
 # Gradio interface
 iface = gr.Interface(fn=download_audio, 
                      inputs=gr.Textbox(label="YouTube Video URL"),
-                     outputs=gr.Audio(label="Output Audio", type="filepath", show_download_button=True, format="mp3"),
+                     outputs=[
+                         gr.File(label="Output Audio File"),
+                         gr.Audio(label="Play Audio", show_download_button=False, format="mp3"),
+                     ],
                      allow_flagging="never"
                      )
 iface.launch()