main.py

import yaml

import translate_tool, audio_tool, whisper_tool
if __name__ == '__main__':
    with open('config.yaml', encoding='utf-8') as f:
        config = yaml.load(f.read(), Loader=yaml.FullLoader)

    print("audio extract begin")
    audio_tool.audio_extract(config['input'], config['output'])
    print("audio extract success")

    print("whisper begin")
    whisper_tool.do_whisper(config['output'], config['srt_path'], config['from'], config['hf_model_path'],
                            config['device'])
    print("whisper success")

    print("translate begin")
    translate_tool.do_translate(config['srt_path'], config['srt_translate_path'], config['from'], config['to'],
                                config['translate_threads'])
    print("translate success")

    print("convert subtitle begin")
    audio_tool.convert_subtitle(config['srt_path'], config['ass_path'])
    print("add subtitle begin")
    audio_tool.add_subtitle(config['input'],config['out_sb'],config['ass_path'])

    print("success")