Update app.py

app.py

@@ -1,109 +1,74 @@
 import gradio as gr
 import yt_dlp
 import os
-import re
-from pathlib import Path
 
-def get_browser_cookies():
-    """Get cookies from browser - tries Chrome first, then Firefox, then Edge"""
-    browsers = ['chrome', 'firefox', 'edge']
-    for browser in browsers:
-        try:
-            return browser
-        except Exception:
-            continue
-    return None
+def download_video_as_mp3(video_url, output_dir="downloads"):
+    """
+    Downloads a YouTube video as an MP3 file using yt-dlp.
 
-def sanitize_filename(title):
-    """Remove invalid characters from filename"""
-    return re.sub(r'[<>:"/\\|?*]', '', title)
+    :param video_url: The URL of the YouTube video.
+    :param output_dir: Directory where the MP3 file will be saved.
+    :return: A tuple containing the filepath of the downloaded MP3 and a status message.
+             Returns (None, error_message) if an error occurred.
+    """
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
 
-def download_audio(url, output_dir="downloads"):
-    try:
-        # Create output directory if it doesn't exist
-        if not os.path.exists(output_dir):
-            os.makedirs(output_dir)
-        
-        # Get browser cookies
-        browser = get_browser_cookies()
-        if not browser:
-            return {
-                "status": "error",
-                "message": "No supported browser found. Please ensure Chrome, Firefox, or Edge is installed."
-            }
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192',
+        }],
+        'outtmpl': os.path.join(output_dir, '%(title)s.%(ext)s'),
+        'quiet': True, # Set to True for cleaner Gradio output, remove if you want yt-dlp's progress in terminal
+        'noplaylist': True,
+    }
 
-        # Configure yt-dlp options
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
-            'outtmpl': os.path.join(output_dir, '%(title)s.%(ext)s'),
-            'cookiesfrombrowser': (browser,),  # Add browser cookies
-            'verbose': True
-        }
-        
-        # Download with yt-dlp
+    try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            # Get video info first
-            info = ydl.extract_info(url, download=False)
-            title = info.get('title', 'video')
-            duration = info.get('duration', 0)
-            author = info.get('uploader', 'Unknown')
-            
-            # Download the video
-            ydl.download([url])
-            
-            # Construct the output filepath
-            safe_title = sanitize_filename(title)
-            output_file = os.path.join(output_dir, f"{safe_title}.mp3")
-            
-            return {
-                "status": "success",
-                "message": f"Successfully downloaded: {safe_title}",
-                "title": title,
-                "author": author,
-                "duration": f"{duration // 60}:{duration % 60:02d}",
-                "file_path": output_file
-            }
-    
+            info_dict = ydl.extract_info(video_url, download=True)
+            video_title = info_dict.get('title', 'video')
+            output_filepath = os.path.join(output_dir, f"{video_title}.mp3")
+        status_message = "Download complete!"
+        return output_filepath, status_message
+
     except Exception as e:
-        return {
-            "status": "error",
-            "message": f"Error: {str(e)}",
-            "title": "",
-            "author": "",
-            "duration": "",
-            "file_path": ""
-        }
+        error_message = f"An error occurred: {e}"
+        return None, error_message
+
+
+def gradio_interface(video_url):
+    """
+    Gradio interface function to download YouTube video as MP3.
+
+    :param video_url: YouTube video URL from Gradio Textbox.
+    :return: A tuple containing the filepath of the downloaded MP3 (for Gradio File output) and status message (for Gradio Textbox).
+    """
+    if not video_url:
+        return None, "Please enter a YouTube video URL."
+
+    output_directory = "downloads" # Default output directory
+
+    filepath, status_message = download_video_as_mp3(video_url, output_directory)
 
-def process_url(url):
-    if not url.strip():
-        return "Please enter a YouTube URL"
-    
-    result = download_audio(url)
-    
-    if result["status"] == "error":
-        return result["message"]
-    
-    return f"""Download successful!
-Title: {result['title']}
-Author: {result['author']}
-Duration: {result['duration']}
-Saved to: {result['file_path']}"""
+    return filepath, status_message
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=process_url,
-    inputs=gr.Textbox(label="YouTube URL", placeholder="Enter YouTube video URL here..."),
-    outputs=gr.Textbox(label="Status"),
-    title="YouTube to MP3 Downloader",
-    description="Enter a YouTube URL to download the audio as MP3. Make sure you're logged into YouTube in Chrome, Firefox, or Edge.",
-    examples=[["https://www.youtube.com/watch?v=dQw4w9WgXcQ"]],
-    theme=gr.themes.Base()
-)
 
 if __name__ == "__main__":
+    iface = gr.Interface(
+        fn=gradio_interface,
+        inputs=gr.Textbox(label="YouTube Video URL", placeholder="Enter YouTube URL here"),
+        outputs=[
+            gr.File(label="Downloaded MP3 File"),
+            gr.Textbox(label="Status"),
+        ],
+        title="YouTube to MP3 Downloader",
+        description="Enter a YouTube video URL and click 'Submit' to download it as MP3. The file will be saved in the 'downloads' directory.",
+        examples=[
+            ["https://www.youtube.com/watch?v=dQw4w9WgXcQ"],
+            ["https://www.youtube.com/watch?v=your_video_id"],
+        ]
+    )
     iface.launch()