Update run.py

run.py

@@ -1,15 +1,16 @@
 import gradio as gr
-import yt_dlp
 import os
-import spaces
+import yt_dlp
 
-@spaces.GPU
-def downloader(video_url, audio_format, audio_name):
+def downloader(video_url, audio_format, audio_name=None):
     # Ensure the directory exists
     os.makedirs('audios', exist_ok=True)
     
     # Use a temporary placeholder for the output file
-    temp_output_path = f"audios/{audio_name}.%(ext)s"
+    if audio_name:
+        temp_output_path = f"audios/{audio_name}.%(ext)s"
+    else:
+        temp_output_path = f"audios/%(title)s.%(ext)s"
     
     ydl_opts = {
         'format': 'bestaudio/best',
@@ -23,13 +24,26 @@ def downloader(video_url, audio_format, audio_name):
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([video_url])
     
-    # Find the downloaded file and rename it
-    temp_file = temp_output_path.replace('%(ext)s', audio_format)
-    final_output_path = f"audios/{audio_name}.{audio_format}"
-    os.rename(temp_file, final_output_path)
+    # Find the downloaded file and rename it if audio_name is provided
+    if audio_name:
+        temp_file = temp_output_path.replace('%(ext)s', audio_format)
+        final_output_path = f"audios/{audio_name}.{audio_format}"
+        os.rename(temp_file, final_output_path)
+    else:
+        final_output_path = None
+        # Since the file is already named by title, find it by the pattern
+        for file in os.listdir('audios'):
+            if file.endswith(f".{audio_format}"):
+                final_output_path = os.path.join('audios', file)
+                break
     
     return final_output_path
 
+# Example usage:
+# downloader('https://www.youtube.com/watch?v=dQw4w9WgXcQ', 'mp3', 'my_audio')
+# downloader('https://www.youtube.com/watch?v=dQw4w9WgXcQ', 'mp3')
+
+
 
     
 
@@ -71,7 +85,8 @@ with gr.Blocks() as demo:
     gr.Markdown("# YouTube Downloader 2.0")
     with gr.Row():
         video_url = gr.Textbox(placeholder="https://youtu.be/yQBGdXGCUbA?si=7avvqH-6OOkGWqFm", label="YouTube video link")
-        audio_name = gr.Textbox(placeholder="killshot", label="Audio name of YouTube audio")
+    with gr.Row():
+        audio_name = gr.Textbox(placeholder="killshot", label="Audio name of YouTube audio", info="this is optional")
         audio_format = gr.Radio(["wav", "flac", "mp3"], label="Select the output format", value=" wav")
     with gr.Row():
         output = gr.Audio(label="Output")