import os
import subprocess

# 递归查找所有mp4文件
def find_mp4_files(root_dir):
    mp4_files = []
    for root, _, files in os.walk(root_dir):
        for file in files:
            if file.lower().endswith('.mp4'):
                mp4_files.append(os.path.join(root, file))
    return mp4_files

# 放大音频音量2倍
def amplify_audio(mp4_path):
    dir_name, base_name = os.path.split(mp4_path)
    name, ext = os.path.splitext(base_name)
    output_path = os.path.join(dir_name, f"{name}_amplified{ext}")
    # ffmpeg命令：音量放大2倍
    cmd = [
        'ffmpeg', '-y', '-i', mp4_path,
        '-vcodec', 'copy',
        '-af', 'volume=2.0',
        output_path
    ]
    print(f"正在处理: {mp4_path}")
    subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    print(f"输出文件: {output_path}")

def main():
    root_dir = os.path.abspath('.')
    mp4_files = find_mp4_files(root_dir)
    if not mp4_files:
        print("未找到mp4文件。")
        return
    for mp4 in mp4_files:
        amplify_audio(mp4)
    print("全部处理完成！")

if __name__ == "__main__":
    main()