Update run.py

run.py

@@ -1,44 +1,32 @@
 import gradio as gr
+import yt_dlp
 import os
 
+def downloader(video_url, audio_format, audio_name):
+    save_path = os.path.join(audio_name)
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': audio_format,
+        }],
+        'outtmpl': save_path,
+    }
 
-def get_video_title(url):
-    result = subprocess.run(["yt-dlp", "--get-title", url], capture_output=True, text=True)
-    if result.returncode == 0:
-        return result.stdout.strip()
-    else:
-        return "Unknown Video"
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([video_url])
+    return save_path
 
-def fetch(url, custom_name, ext):
-    title = get_video_title(url)
-    max_length = 50
-    truncated_title = title[:max_length].strip()
+with gr.Blocks() as demo:
+    gr.Markdown("# YouTube Audio Downloader")
     
-    filename = f"{custom_name}.{ext}" if custom_name else f"{truncated_title}.{ext}"
-    opts = {
-        "wav": ["-f", "ba", "-x", "--audio-format", "wav"],
-        "mp4": ["-f", "bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best"],
-    }[ext]
-    command = ["yt-dlp"] + opts + [url, "-o", filename]
-    subprocess.run(command)
-
-    return filename
-
-def play_audio(output_audio):
-    audio = AudioSegment.from_file(output_audio)
-    audio.export("output.wav", format="wav")
-    gr.Interface(fn=None, live=False, outputs="audio").play("output.wav")
-
-app = gr.Interface(
-    theme='Hev832/EasyAndCool',
-    fn=play_audio,
-    inputs=[
-        gr.Textbox(label="YouTube video address", placeholder="Paste video link here..."),
-        gr.Textbox(label="File name", placeholder="Defaults to video title"),
-        gr.Dropdown(value="wav", label="Format")
-    ],
-    outputs=None,
-    description="<div style='font-size:30px; text-align:center;'>YouTube Audio downloader</div>"
-)
+    video_url = gr.Textbox(label="YouTube video link")
+    audio_name = gr.Textbox(label="Audio name of YouTube audio")
+    audio_format = gr.Radio(["wav", "flac", "mp3"], label="Select the output format")
+    
+    output = gr.Textbox(label="Output")
+    
+    download_button = gr.Button("Download")
+    download_button.click(downloader, inputs=[video_url, audio_format, audio_name], outputs=output)
 
-app.launch(debug=True, share=True)
+demo.launch()