Update app.py

app.py

@@ -1,15 +1,13 @@
-import os
+print("please wait...")
+
 import gradio as gr
 import yt_dlp
 
 def download_media(url, download_video):
-    download_dir = 'downloads'
-    os.makedirs(download_dir, exist_ok=True)
-    
     if download_video:
         ydl_opts = {
             'format': 'bestvideo+bestaudio/best',
-            'outtmpl': f'{download_dir}/%(title)s.%(ext)s',
+            'outtmpl': 'downloads/%(title)s.%(ext)s',
         }
     else:
         ydl_opts = {
@@ -19,13 +17,13 @@ def download_media(url, download_video):
                 'preferredcodec': 'mp3',
                 'preferredquality': '192',
             }],
-            'outtmpl': f'{download_dir}/%(title)s.%(ext)s',
+            'outtmpl': 'downloads/%(title)s.%(ext)s',
         }
 
     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:
@@ -33,16 +31,16 @@ def download_media(url, download_video):
 
     return output_file
 
-def handle_download(url, download_video):
-    output_file = download_media(url, download_video)
+def get_output_component(download_video):
     if download_video:
-        return gr.update(value=output_file, visible=True), gr.update(visible=False)
+        return gr.File(label="Downloaded Media")
     else:
-        return gr.update(visible=False), gr.update(value=output_file, visible=True)
+        return gr.Audio(label="Downloaded Media")
 
+# 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 from 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 form youtube url </div>")
     
     with gr.Row():
         url_input = gr.Textbox(label="YouTube URL")
@@ -54,12 +52,18 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange"
         output_audio = gr.Audio(label="Downloaded Media", visible=False)
         output_file = gr.File(label="Downloaded Media", visible=False)
 
+    def handle_download(url, download_video):
+        output_file = download_media(url, download_video)
+        if download_video:
+            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://www.linkedin.com/in/muhammad-haseeb-ahmed-1954b5230/'>Muhammad Haseeb Ahmed</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()