Update app.py

app.py

@@ -6,35 +6,56 @@ import gradio as gr
 
 def is_internet_available():
     try:
-        # Try resolving YouTube hostname
         socket.gethostbyname('www.youtube.com')
         return True
     except socket.error:
         return False
 
-def download_video(url):
+def download_media(url, format_choice):
     if not url.strip():
         raise gr.Error("⚠️ Please enter a valid YouTube or TikTok URL.")
     if not is_internet_available():
         raise gr.Error("❌ No internet access detected. Please enable internet in your deployment environment.")
 
     temp_dir = tempfile.mkdtemp()
+    # File template
     output_path = os.path.join(temp_dir, "%(title)s.%(ext)s")
 
-    ydl_opts = {
-        'format': 'best',
-        'outtmpl': output_path,
-        'quiet': True,
-        'no_warnings': True,
-        'noplaylist': True,
-        'socket_timeout': 15,
-    }
+    # yt-dlp options differ based on format
+    if format_choice == "Audio (MP3)":
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'outtmpl': output_path,
+            'quiet': True,
+            'no_warnings': True,
+            'noplaylist': True,
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+                'preferredquality': '192',
+            }],
+            'socket_timeout': 15,
+        }
+    else:  # Video
+        ydl_opts = {
+            'format': 'bestvideo+bestaudio/best',
+            'outtmpl': output_path,
+            'quiet': True,
+            'no_warnings': True,
+            'noplaylist': True,
+            'merge_output_format': 'mp4',
+            'socket_timeout': 15,
+        }
 
     try:
         with YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=True)
             downloaded_file = ydl.prepare_filename(info)
 
+        # If audio, extension is mp3 due to postprocessor
+        if format_choice == "Audio (MP3)":
+            downloaded_file = os.path.splitext(downloaded_file)[0] + ".mp3"
+
         if not os.path.exists(downloaded_file):
             raise gr.Error("❌ Download failed. File not found.")
 
@@ -51,11 +72,14 @@ def download_video(url):
             raise gr.Error(f"❌ Error: {err}")
 
 demo = gr.Interface(
-    fn=download_video,
-    inputs=gr.Textbox(label="🎥 Enter YouTube or TikTok URL"),
-    outputs=gr.File(label="📁 Downloaded Video File"),
-    title="🎬 Video Downloader",
-    description="Download videos from YouTube or TikTok. Paste a valid video link and get the file.",
+    fn=download_media,
+    inputs=[
+        gr.Textbox(label="🎥 Enter YouTube or TikTok URL"),
+        gr.Radio(["Audio (MP3)", "Video (MP4)"], label="Select format", value="Audio (MP3)")
+    ],
+    outputs=gr.File(label="📁 Downloaded File"),
+    title="🎬 Audio/Video Downloader",
+    description="Download audio (MP3) or video (MP4) from YouTube or TikTok by pasting a valid URL.",
     theme="default"
 )