Update app.py

app.py

@@ -9,22 +9,15 @@ import numpy as np
 # no space, punctuation, accent in lower string
 def cleanString(string):
     cleanString = unidecode(string)
-    cleanString = re.sub('\W+','_', cleanString)
+    # cleanString = re.sub('\W+','_', cleanString)
+    cleanString = re.sub(r'[^\w\s]','',cleanString)
+    cleanString = cleanString.replace(" ", "_")
     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
-
+# from YouTube url to audio file path and sample rate + numpy array
 def download_audio(url):
-    path_to_folder_audio_mp3 = "./audio/"
+
+    path_to_folder_audio_mp3 = "./"
     ydl_opts = {
         'format': 'm4a/bestaudio/best',
         'outtmpl': f'{path_to_folder_audio_mp3}%(title)s',
@@ -33,6 +26,7 @@ def download_audio(url):
             'preferredcodec': 'mp3',
         }]
     }
+
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info_dict = ydl.extract_info(url, download=True)
         video_title = info_dict['title']
@@ -42,21 +36,22 @@ def download_audio(url):
         new_local_link = cleanString(video_title) + ".mp3"
         for filename in os.listdir(path_to_folder_audio_mp3):
             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)) 
+                os.rename(os.path.join(path_to_folder_audio_mp3, filename),os.path.join(path_to_folder_audio_mp3, new_local_link))
 
+        # get audio file path
         file_path = path_to_folder_audio_mp3 + new_local_link
 
-    return file_path, read_audio(file_path)
-
-# Gradio interface
-iface = gr.Interface(fn=download_audio, 
-                     inputs=gr.Textbox(label="YouTube Video URL"),
-                     outputs=[
-                         gr.File(label="Output Audio File"),
-                         gr.Audio(label="Play Audio", show_download_button=False, format="mp3"),
-                     ],
-                     allow_flagging="never",
-                     title="YouTube Video to Audio (mp3)",
-                     description="Just paste any YouTube video url and get its corresponding audio file in mp3.",
-                     )
-iface.launch(debug=True)
+    return file_path, file_path
+
+with gr.Blocks() as demo:
+    gr.Markdown("<h1><center>Free YouTube URL Video-to-Audio</center></h1>")
+    gr.Markdown("<center>Enter the link of any YouTube video to generate its mp3 audio file.</center>")
+    
+    input_text_url = gr.Textbox(placeholder='Youtube video URL', label='YouTube URL')
+    result_button_audio = gr.Button('Get Audio File')
+    output_audio_file = gr.File(label='mp3 audio file')
+    output_audio_play = gr.Audio(type="filepath", label="Listen to audio")
+    
+    result_button_audio.click(download_audio, inputs = input_text_url, outputs = [output_audio_file, output_audio_play])
+
+demo.queue().launch(debug = False)