youtubetoaudioconverter.py

from pytube import YouTube
import requests
from moviepy.editor import *
import os
import tkinter as tk
from tkinter import filedialog

def downloadYtClip(current_url: str):
    """
    Takes in the current youtube link you want to download and returns the path where you downloaded the file
    
    Args:
    current_url (str), the youtube link passed in

    Returns:
        the directory path where the video is downloaded (assuming the download goes successful)
    """
    req = requests.head(current_url, allow_redirects=True).url
    if "youtube.com" in req:
       try:
            video_to_download = YouTube(current_url)
            print("We have found your video")
       except:
            print("Video Failed, please try again!")
            return None 
       currentStream = video_to_download.streams.filter(file_extension='mp4').first()
       
       try:
           downloaded_file_path = currentStream.download()
           print(downloaded_file_path)
           return downloaded_file_path
       except:
           print("Video failed to download")
    else:
        print("Error youtube link is invalid")
        return None
    return

def convertToMp3(clip: str, outputPath: str):
    """
    Converts the downloaded clip to MP3 format and saves it to the specified output path.
    
    Args:
    clip (str): The path to the downloaded video clip
    outputPath (str): The path to save the converted MP3 file
    """
    try:
        downloadedClip = VideoFileClip(clip)
        print("Video FPS:", downloadedClip.fps)
        audioClip = downloadedClip.audio
        audioClip.write_audiofile(outputPath, codec='libmp3lame')
        audioClip.close()
        downloadedClip.close()
    except Exception as error:
        print("Error in convertToMp3:", error)
        if 'downloadedClip' in locals():
            downloadedClip.close()
        if 'audioClip' in locals():
            audioClip.close()
    
def convertToWav(clip:str, outputPath:str):
    """
    Takes in the current path where the downloaded clip is in and the output directory path where you want to store your wav file
    Converts to wav format then stores it in the path
    """
    try:
        downloadedClip = VideoFileClip(clip)
        #Grab the audio of that downloaded video file clip
        audioClip  = downloadedClip.audio
        #Writes the audio file
        audioClip.write_audiofile(outputPath, codec='libwavlame')
        audioClip.close()
        downloadedClip.close()
    # Raise exception error
    except Exception as error:
        print("Looks like you got an error", error)


def main():
    #Instantiate Tkinter object
    currentFrame = tk.Tk()
    currentFrame.title("Youtube to mp3 audio converter")
    currentFrame.geometry('400x200')
    lbl = tk.Label(currentFrame, text = "Enter the youtube video you want to convert.\n NOTE it has to be a youtube url.\n", foreground="red")
    userInputtxt = tk.Text(currentFrame, height=2, width=30)
    lbl.pack() 
    userInputtxt.pack()
    #Functions to handle whether the file that wants to be converted is in mp3 format or wav format
    #Asks input forom the tkinter
    def handleTkinterMp3Input():
        songInput = userInputtxt.get(1.0, "end-1c")
        downloadedYtClip = downloadYtClip(songInput)
        if downloadedYtClip != None:
            currentPath = filedialog.askdirectory()
            if not os.path.exists(currentPath):
                os.makedirs(currentPath)
            mp3FilePath = os.path.join(currentPath, os.path.basename(downloadedYtClip).replace(".mp4", ".mp3"))
            convertToMp3(downloadedYtClip, mp3FilePath)
    
    def handleTkinterWavInput():
        songInput = userInputtxt.get(1.0, "end-1c")
        downloadedYtClip = downloadYtClip(songInput)
        if downloadedYtClip != None:
            currentPath = filedialog.askdirectory()
            if not os.path.exists(currentPath):
                os.makedirs(currentPath)
            mp3FilePath = os.path.join(currentPath, os.path.basename(downloadedYtClip).replace(".mp4", ".wav"))
            convertToMp3(downloadedYtClip, mp3FilePath)

    convert_to_mp3_button = tk.Button(currentFrame, text="Convert to Mp3", command=handleTkinterMp3Input)
    convert_to_mp3_button.place(x=500, y=90)
    handleTkinterWavInputButton = tk.Button(currentFrame, text="Convert to Wav", command=handleTkinterWavInput)
    handleTkinterWavInputButton.pack()
    handleTkinterWavInputButton.place(x=800, y=90)
    currentFrame.mainloop()
    convert_to_mp3_button.quit()
    handleTkinterWavInputButton.quit()
    currentFrame.quit()  


if __name__ in "__main__":
    main()