fixaudio.py

import os
import sys
import subprocess
from pathlib import Path
import uuid

def get_user_confirmation(prompt, default="y"):
    valid = {"yes": True, "y": True, "ye": True, "no": False, "n": False}
    if default == "y": prompt += " [Y/n] "
    elif default == "n": prompt += " [y/N] "
    else: raise ValueError(f"Invalid default value: '{default}'")

    while True:
        choice = input(prompt).lower()
        if choice == '': return valid[default]
        elif choice in valid: return valid[choice]
        else: print("Please respond with 'yes' or 'no' (or 'y' or 'n').")

def check_audio_properties(file_path):
    try:
        sample_rate_result = subprocess.run(
            ['ffprobe', '-v', 'error', '-select_streams', 'a:0', '-show_entries', 'stream=sample_rate,channels', '-of', 'csv=p=0', str(file_path)],
            stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, check=True
        )
        rate, channels = sample_rate_result.stdout.strip().split(',')
        return int(rate), int(channels)
    except (ValueError, subprocess.CalledProcessError):
        return None, None

def convert_audio(input_file, output_file):
    try:
        temp_output = output_file.with_stem(f"{output_file.stem}_{uuid.uuid4().hex[:8]}_temp") if input_file == output_file else output_file

        command = [
            'ffmpeg', '-y', '-i', str(input_file), '-acodec', 'pcm_s16le',
            '-ar', '44100', '-ac', '1', str(temp_output)
        ]
        subprocess.run(command, check=True, stderr=subprocess.PIPE, text=True)

        if input_file == output_file:
            os.remove(input_file)
            os.rename(temp_output, output_file)
        return True

    except Exception as e:
        print(f"❌ Error processing {input_file.name}: {e}")
        return False

def process_directory():
    current_dir = Path.cwd()
    print(f"🔍 Processing audio files in: {current_dir}\n")

    if not get_user_confirmation("Do you want to start the conversion process?"):
        print("Process aborted by user.")
        return

    replace_originals = get_user_confirmation("\nReplace original files with new mono 44100Hz WAV files? If no, both will be kept.", default="n")
    print(f"\n{'Replacing' if replace_originals else 'Keeping'} original files.\n")

    total_files, successful, failed, skipped = 0, 0, 0, 0
    audio_extensions = ('.mp3', '.m4a', '.flac', '.ogg', '.aac', '.wav', '.wma', '.opus')

    for file_path in current_dir.rglob('*'):
        if file_path.suffix.lower() in audio_extensions:
            total_files += 1
            output_path = file_path.with_suffix('.wav')

            print(f"\n🔄 Processing: {file_path.relative_to(current_dir)}")

            if file_path.stat().st_size == 0:
                print(f"⚠️ Skipping {file_path.name} (empty file)")
                failed += 1
                continue

            sample_rate, channels = check_audio_properties(file_path)
            if sample_rate == 44100 and channels == 1:
                print(f"✅ {file_path.name} is already mono 44100Hz. Skipping.")
                skipped += 1
                continue
            elif sample_rate == 44100 and channels > 1:
                print(f"🔄 {file_path.name} is 44100Hz but has {channels} channels. Converting to mono.")
            elif sample_rate != 44100 and channels == 1:
                print(f"🔄 {file_path.name} is mono but at {sample_rate}Hz. Converting to 44100Hz.")
            else:
                print(f"🔄 {file_path.name} is {channels}-channel {sample_rate}Hz. Converting to mono 44100Hz.")

            if convert_audio(file_path, output_path):
                if replace_originals and (file_path.suffix.lower() != '.wav' or not file_path.samefile(output_path)):
                    os.remove(file_path)
                    print(f"✅ Converted and replaced with {output_path.name}")
                else:
                    print(f"✅ Converted. Kept both {file_path.name} and {output_path.name}")
                successful += 1
            else:
                print(f"❌ Conversion failed for {file_path.name}")
                failed += 1

    print(f"\n📊 Conversion Summary:")
    print(f"   Total audio files found: {total_files}")
    print(f"   Successfully converted:  {successful}")
    print(f"   Already correct format:  {skipped}")
    print(f"   Failed conversions:      {failed}")

if __name__ == "__main__":
    if sys.platform == 'win32':
        sys.stdout.reconfigure(encoding='utf-8')

    try:
        process_directory()
    except KeyboardInterrupt:
        print("\n\n❗ Process interrupted by user. Exiting gracefully.")
        sys.exit(0)