variables.py

import os
import sys
from pathlib import Path


class Variables:
    def __init__(self):
        self.sample_rate = None
        self.duration = None
        self.frequency_left = None
        self.frequency_right = None
        self.video_name = ""
        self.volume = None
        self.frequency_transition = None
        self.output_file = ""

    def name_video(self):
        self.video_name = input("Enter the name of the video file (including the file extension): ")
        if not Path(self.video_name).exists():
            raise SystemExit("The file does not exist in this directory. Please ensure the file is in this directory.")
        elif not self.video_name.endswith(".mp4"):
            raise SystemExit("The file is not an MP4 file. Please ensure the file is an MP4 file.")
        return self.video_name

    @staticmethod
    def command_exists(command):
        try:
            os.system(f"where {command}")
            return True
        except FileNotFoundError:
            return False

    @staticmethod
    def is_ffmpeg_installed():
        try:
            ffmpeg = os.system("where ffmpeg")
            if ffmpeg == 0:
                return "ffmpeg"
        except FileNotFoundError:
            return False

    def interface(self):
        try:
            self.sample_rate = int(input("Sample Rate (e.g., 44100 for CD-quality audio): "))
            self.duration = int(input("Duration (in seconds): "))
            self.frequency_left = int(input("Frequency for the left ear (in Hz): "))
            self.frequency_right = int(input("Frequency for the right ear (in Hz): "))
            self.volume = input("Volume (e.g., 1.0 for 100%, 0.5 for 50%): ")
            transition_input = input("Enter frequency transitions for left ear separated by commas (optional, press Enter to skip): ")
            self.frequency_transition = {
                "left": [int(f) for f in transition_input.split(",")] if transition_input else [self.frequency_left],
                "right": [int(f) for f in input("Enter frequency transitions for right ear separated by commas (optional, press Enter to skip): ").split(",")] if transition_input else [self.frequency_right],
            }
            self.output_file = input("Enter the desired name for the output file (including the .mp4 extension): ")
            if not self.output_file.endswith(".mp4"):
                raise SystemExit("The output file must have a .mp4 extension.")
        except ValueError:
            print("Error: Sample rate, duration, and frequencies must be integers.")
            sys.exit()

        if not all([self.sample_rate, self.duration, self.frequency_left, self.frequency_right, self.output_file]):
            print("Error: All values must be set and non-zero.")
            sys.exit()

        return self.sample_rate, self.duration, self.frequency_left, self.frequency_right, self.volume, self.frequency_transition, self.video_name, self.output_file