ytmp3.py

import os
import logging
from yt_dlp import YoutubeDL

DOWNLOAD_FOLDER = os.path.join(os.getcwd(), "downloads")
os.makedirs(DOWNLOAD_FOLDER, exist_ok=True)

errors = []  # List to capture errors

def setup_logging(hidden, log_file):
    """
    Set up logging based on the provided options.

    :param hidden: If True, suppress all output except errors.
    :param log_file: Path to the log file (if provided).
    """
    log_level = logging.ERROR if hidden else logging.DEBUG
    logging.basicConfig(
        level=log_level,
        format="%(asctime)s - %(levelname)s - %(message)s",
        handlers=[logging.StreamHandler()]
    )
    if log_file:
        file_handler = logging.FileHandler(log_file, encoding="utf-8")
        file_handler.setFormatter(logging.Formatter("%(asctime)s - %(levelname)s - %(message)s"))
        logging.getLogger().addHandler(file_handler)

def download_audio(video_url):
    """
    Download audio as MP3 from a given YouTube URL.

    :param video_url: The URL of the YouTube video.
    """
    ydl_opts = {
        "format": "bestaudio/best",
        "retries": 3,
        "verbose": False,
        "outtmpl": os.path.join(DOWNLOAD_FOLDER, "%(title)s.%(ext)s"),
        "postprocessors": [
            {
                "key": "FFmpegExtractAudio",
                "preferredcodec": "mp3",
                "preferredquality": "192",
            }
        ],
    }

    with YoutubeDL(ydl_opts) as ydl:
        try:
            ydl.download([video_url])
            logging.info(f"Downloaded: {video_url}")
        except Exception as e:
            logging.error(f"Failed to download {video_url}: {e}")
            errors.append(f"Failed to download {video_url}: {e}")  # Capture the error

def process_video(link):
    """
    Process a single video link and download its audio.

    :param link: The original YouTube link.
    """
    if "music.youtube.com" in link:
        link = link.replace("music.youtube.com", "www.youtube.com")

    logging.info(f"Downloading the original video: {link}")
    download_audio(link)

def download_from_file(link_file):
    """
    Download audio from a file containing YouTube links.

    :param link_file: Path to the file containing the YouTube links (one per line).
    """
    if not os.path.exists(link_file):
        logging.error(f"File '{link_file}' does not exist.")
        errors.append(f"File '{link_file}' does not exist.")
        return

    with open(link_file, "r", encoding="utf-8") as file:
        links = [line.strip() for line in file if line.strip()]

    if not links:
        logging.error("No links found in the file.")
        errors.append("No links found in the file.")
        return

    for link in links:
        process_video(link)

def inline_download(video_url):
    """
    Allow inline downloading of a single YouTube link.

    :param video_url: The URL of the YouTube video.
    """
    logging.info(f"Downloading audio for the provided link: {video_url}")
    process_video(video_url)

def done_menu():
    """
    Display a menu after the download is complete, offering the user to either return to the main menu or exit.
    This will also show errors that occurred during the process.
    """
    global errors  # Use the global errors list to display them
    while True:
        if errors:
            print("\nErrors that occurred during the download process:")
            for error in errors:
                print(f"- {error}")

        print("\nDone! What would you like to do next?")
        print("1. Return to the main menu")
        print("2. Exit the program")

        choice = input("Enter the number of your choice: ")

        if choice == '1':
            return True  # Return to the main menu by returning to the main loop
        elif choice == '2':
            print("Exiting the downloader. Goodbye!")
            return False  # Exit the program
        else:
            print("Invalid choice. Please try again.")

def main():
    """
    CLI-like interface for the user to choose how to download content.
    """
    print("Welcome to the YouTube Audio Downloader!")
    
    while True:
        print("\nPlease choose an option:")
        print("1. Download from a file containing YouTube links.")
        print("2. Download a single YouTube video by providing the URL.")
        print("3. Exit.")

        choice = input("Enter the number of your choice: ")

        if choice == '1':
            file_path = input("Enter the path to the file containing YouTube links: ")
            logging_level = input("Enable hidden logging (only errors)? (y/n): ").strip().lower()
            hidden = logging_level == 'y'
            setup_logging(hidden, None)
            download_from_file(file_path)
            if not done_menu():  # If the user chooses to exit
                break
        elif choice == '2':
            video_url = input("Enter the YouTube video URL: ")
            logging_level = input("Enable hidden logging (only errors)? (y/n): ").strip().lower()
            hidden = logging_level == 'y'
            setup_logging(hidden, None)
            inline_download(video_url)
            if not done_menu():  # If the user chooses to exit
                break
        elif choice == '3':
            print("Exiting the downloader. Goodbye!")
            break
        else:
            print("Invalid choice. Please try again.")

if __name__ == "__main__":
    main()