Update app.py

app.py

@@ -1,22 +1,53 @@
 import gradio as gr
+import unidecode
+from unidecode import unidecode
 import yt_dlp
+import os
+
+# no space, punctuation, accent in lower string
+def cleanString(string):
+    cleanString = unidecode(string)
+    cleanString = re.sub('\W+','_', cleanString)
+    return cleanString.lower()
 
 def download_audio(url):
+    path_to_folder_audio_mp3 = "./audio/"
     ydl_opts = {
+        # 'format': 'bestaudio/best',
         'format': 'm4a/bestaudio/best',
+        # 'outtmpl': '%(id)s.%(ext)s',
+        'outtmpl': f'{path_to_folder_audio_mp3}%(title)s',
         'postprocessors': [{
             'key': 'FFmpegExtractAudio',
             'preferredcodec': 'mp3',
+            # 'preferredquality': '192',
         }]
     }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info_dict = ydl.extract_info(url, download=True)
-        video_url = info_dict['url']
-    return video_url
+        video_title = info_dict['title']
+        # ydl.download([video_link])
+
+        # Rename the audio file
+        local_link = video_title + ".mp3"
+        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)) 
+
+    return path_to_folder_audio_mp3 + new_local_link 
+
+# def video_to_audio(url):
+#     video_url = download_video(url)
+#     video = mp.VideoFileClip(video_url)
+#     audio = video.audio
+#     audio_file = "output_audio.mp3"
+#     audio.write_audiofile(audio_file)
+#     return audio_file
 
 iface = gr.Interface(fn=download_audio, 
                      inputs=gr.Textbox(label="YouTube Video URL"),
-                     outputs=gr.Audio(label="Output Audio", type="filepath", format="mp3"),
+                     outputs=gr.File(label="Output Audio"),
                      allow_flagging="never"
                      )
-iface.launch()
+iface.launch(debug=True)