Update app.py

app.py

@@ -1,10 +1,7 @@
-
-import gradio as gr
 import yt_dlp
-import os 
-
+import gradio as gr
 
-def download_youtube_wav(url, custom_name):
+def download_youtube_wav(url, custom_name, custom_directory):
     ydl_opts = {
         'format': 'bestaudio/best',
         'postprocessors': [{
@@ -12,20 +9,21 @@ def download_youtube_wav(url, custom_name):
             'preferredcodec': 'wav',
             'preferredquality': '192',
         }],
-        'outtmpl': custom_name + '.%(ext)s',
+        'outtmpl': f'{custom_directory}/{custom_name}.%(ext)s',
     }
 
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([url])
-    
-    return f'File "{custom_name}.wav" telah di-download!'
+
+    return f'Download complete! File saved as {custom_name}.wav in {custom_directory}'
 
 iface = gr.Interface(
     fn=download_youtube_wav,
-    inputs=['text', 'text'],
-    outputs='text',
-    title='YouTube to WAV Downloader',
-    description='Enter the YouTube URL and custom name for the WAV file you want to download.'
+    inputs=['text', 'text', 'text'],
+    outputs='Audio',
+    title='YouTube WAV Downloader',
+    description='Enter the YouTube URL, custom filename, and directory to download the video as a WAV file.'
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()