Update run.py

run.py

@@ -1,9 +1,8 @@
 import gradio as gr
 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()
@@ -12,22 +11,23 @@ def get_video_title(url):
 
 def fetch(url, custom_name, ext):
     title = get_video_title(url)
-    #  
-    max_length = 50  # 
+    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"],
-
+        "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',
@@ -37,10 +37,8 @@ app = gr.Interface(
         gr.Textbox(label="File name", placeholder="Defaults to video title"),
         gr.Dropdown(value="wav", label="Format")
     ],
-    outputs=[
-        gr.Audio(label="otputs"),
+    outputs=None,
     description="<div style='font-size:30px; text-align:center;'>YouTube Audio Player</div>"
-
 )
 
-app.launch(debug=True, share=True)
+app.launch(debug=True, share=True)