Update run.py

run.py

@@ -1,21 +1,33 @@
 import gradio as gr
-import yt_dlp
+import os
+import subprocess
 
+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"
+
+def fetch(url, custom_name, ext):
+ title = get_video_title(url)
+ # 
+ max_length = 50 # 
+ truncated_title = title[:max_length].strip()
+ 
+ filename = f"{custom_name}.{ext}" if custom_name else f"{truncated_title}.{ext}"
+ opts = {
+ "mp3": ["-f", "ba", "-x", "--audio-format", "mp3"],
+ "wav": ["-f", "ba", "-x", "--audio-format", "wav"],
+
+ }[ext]
+ command = ["yt-dlp"] + opts + [url, "-o", filename]
+ subprocess.run(command)
+
+ return filename
 
 
-def download_audio(url):
- ydl_opts = {
- 'format': 'bestaudio/best',
- 'postprocessors': [{
- 'key': 'FFmpegExtractAudio',
- 'preferredcodec': 'wav',
- 'preferredquality': '192',
- }],
- 'outtmpl': '%(title)s.%(ext)s',
- }
- with yt_dlp.YoutubeDL(ydl_opts) as ydl:
- ydl.download([url])
- return 
 
 with gr.Blocks() as demo: 
  used_letters_var = gr.State([])
@@ -25,13 +37,14 @@ with gr.Blocks() as demo:
  with gr.Row() as row:
  with gr.Column():
  url = gr.Textbox(label="URL INPUT")
+ custom_name = gr.Textbox(label="Custom Name (defalut if you want!)")
  outputs = gr.Audio(label="outputs")
  with gr.Column():
  btn = gr.Button("download!")
  
 
  btn.click(
- download_audio, 
+ fetch, 
  [url],
  [outputs]
  )