extractor.py

import os
import moviepy
from tkinter import Tk, filedialog, simpledialog, messagebox


if not os.path.exists("input"):
    os.makedirs("input")
if not os.path.exists("output"):
    os.makedirs("output")

def extract_audio_from_video(video_path, audio_path):
    try:
        video = moviepy.VideoFileClip(video_path)
        audio = video.audio
        audio.write_audiofile(audio_path)
        return True
    except Exception as e:
        print(f"An error occurred: {e}")
        return False

def main():
    root = Tk()
    root.withdraw()
    
    video_path = filedialog.askopenfilename(title="Select the video file", filetypes=[("Video Files", "*.mp4;*.avi;*.mkv;*.flv;*.mov")])
    if not video_path:
        return

    dest_path = os.path.join("input", os.path.basename(video_path))
    with open(video_path, "rb") as fsrc:
        with open(dest_path, "wb") as fdst:
            fdst.write(fsrc.read())

    audio_format = simpledialog.askstring("Output Format", "Enter desired audio format (e.g. mp3, wav):", initialvalue="mp3")

    output_filename = os.path.splitext(os.path.basename(dest_path))[0] + f".{audio_format}"
    audio_path = os.path.join("output", output_filename)
    
    success = extract_audio_from_video(dest_path, audio_path)

    if success:
        messagebox.showinfo("Success", f"Audio saved successfully at: {audio_path}")
    else:
        messagebox.showerror("Error", "Failed to extract audio. Please check the video file.")

if __name__ == "__main__":
    main()