Update app.py

app.py

@@ -1,57 +1,58 @@
 import gradio as gr
-import re, unidecode
-from unidecode import unidecode
+import re
+import unidecode
 import yt_dlp
 import os
-import pydub
-import numpy as np
 
-# no space, punctuation, accent in lower string
+# Function to clean the string for file naming
 def cleanString(string):
-    cleanString = unidecode(string)
-    # cleanString = re.sub('\W+','_', cleanString)
-    cleanString = re.sub(r'[^\w\s]','',cleanString)
-    cleanString = cleanString.replace(" ", "_")
-    return cleanString.lower()
+  clean_string = unidecode.unidecode(string)
+  clean_string = re.sub(r'[^\w\s]', '', clean_string)
+  clean_string = clean_string.replace(" ", "_")
+  return clean_string.lower()
 
-# from YouTube url to audio file path and sample rate + numpy array
+# Function to download audio from YouTube
 def download_audio(url):
-
-    path_to_folder_audio_mp3 = "./"
-    ydl_opts = {
-        'format': 'm4a/bestaudio/best',
-        'outtmpl': f'{path_to_folder_audio_mp3}%(title)s',
-        'postprocessors': [{
-            'key': 'FFmpegExtractAudio',
-            'preferredcodec': 'mp3',
-        }]
-    }
-
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info_dict = ydl.extract_info(url, download=True)
-        video_title = info_dict['title']
-
-        # 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))
-
-        # get audio file path
-        file_path = path_to_folder_audio_mp3 + new_local_link
-
-    return file_path, file_path
-
+  path_to_folder_audio_mp3 = "./"
+  ydl_opts = {
+      'format': 'bestaudio/best',
+      'outtmpl': f'{path_to_folder_audio_mp3}%(title)s.%(ext)s',
+      'postprocessors': [{
+          'key': 'FFmpegExtractAudio',
+          'preferredcodec': 'mp3',
+          'preferredquality': '192',
+      }],
+      'noplaylist': True,  # Avoid downloading playlists
+  }
+
+  with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+      info_dict = ydl.extract_info(url, download=True)
+      video_title = info_dict['title']
+
+      # Clean the title for the filename
+      new_local_link = cleanString(video_title) + ".mp3"
+      original_file_path = os.path.join(path_to_folder_audio_mp3, f"{video_title}.mp3")
+
+      # Rename the audio file if it exists
+      if os.path.exists(original_file_path):
+          os.rename(original_file_path, os.path.join(path_to_folder_audio_mp3, new_local_link))
+
+      # Get the final file path
+      file_path = os.path.join(path_to_folder_audio_mp3, new_local_link)
+
+  return file_path, file_path
+
+# Gradio interface
 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)
+  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])
+
+# Launch the Gradio app
+demo.queue().launch(debug=True)