Update app.py

app.py

@@ -4,32 +4,33 @@ import gradio as gr
 import yt_dlp
 
 def download_media(url, download_video):
-    if download_video:
-        ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',
-            'outtmpl': 'downloads/%(title)s.%(ext)s',
-        }
-    else:
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
-            'outtmpl': 'downloads/%(title)s.%(ext)s',
-        }
+    ydl_opts = {
+        'format': 'bestvideo+bestaudio/best' if download_video else 'bestaudio/best',
+        'outtmpl': 'downloads/%(title)s.%(ext)s',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192',
+        }] if not download_video else [],
+        'noplaylist': True,
+        'quiet': True,
+        'progress_hooks': [progress_hook],
+    }
 
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info_dict = ydl.extract_info(url, download=True)
         file_title = ydl.prepare_filename(info_dict)
         
-        if download_video:
-            output_file = file_title
-        else:
-            output_file = file_title.rsplit('.', 1)[0] + '.mp3'
+        if not download_video:
+            file_title = file_title.rsplit('.', 1)[0] + '.mp3'
+
+    return file_title
 
-    return output_file
+def progress_hook(d):
+    if d['status'] == 'downloading':
+        print(f"Downloading: {d['_percent_str']} of {d['_total_bytes_str']}")
+    elif d['status'] == 'finished':
+        print("Download complete, now converting...")
 
 def get_output_component(download_video):
     if download_video:
@@ -39,11 +40,11 @@ def get_output_component(download_video):
 
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange")) as demo:
-    gr.Markdown(f"# <div style='text-align: center;'> YOUTUBE Downloader</div>")
-    gr.Markdown(f"## <div style='text-align: center;'> download mp3/mp4 form youtube url </div>")
+    gr.Markdown(f"# <div style='text-align: center;'>YOUTUBE Downloader</div>")
+    gr.Markdown(f"## <div style='text-align: center;'>Download MP3/MP4 from YouTube URL</div>")
     
     with gr.Row():
-        url_input = gr.Textbox(label="YouTube URL")
+        url_input = gr.Textbox(label="YouTube URL", placeholder="Enter YouTube video URL here")
     with gr.Row():
         download_video_checkbox = gr.Checkbox(label="Download Video", value=False)
     with gr.Row():    
@@ -58,12 +59,14 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange"
             return gr.update(value=output_file, visible=True), gr.update(visible=False)
         else:
             return gr.update(visible=False), gr.update(value=output_file, visible=True)
-    gr.Markdown(f"### <div style='text-align: center;'>made with ❤ by <a href='https://huggingface.co/Gradio-Blocks'>Gradio-Blocks-Party-Member</a></div>")
-    
+
     download_button.click(
         handle_download, 
         inputs=[url_input, download_video_checkbox], 
         outputs=[output_file, output_audio]
     )
 
+    gr.Markdown(f"### <div style='text-align: center;'>Made with ❤ by <a href='https://huggingface.co/Gradio-Blocks'>Gradio-Blocks-Party-Member</a></div>")
+
 demo.launch()
+