yuisub

Auto translation of new anime episodes based on Yui-MHCP001 LLM

Install

Make sure you have Python >= 3.9 installed on your system

pip install yuisub

If you wanna use the a2t module, you need to install Whisper first

# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install openai-whisper

Command Line Usage

yuisub can be used from the command line to generate bilingual ASS files. Here's how to use it:

yuisub -h  # Displays help message

Library

yuisub can also be used as a library

import asyncio

from yuisub import SubtitleTranslator

# Using an asynchronous environment
async def main() -> None:
    translator = SubtitleTranslator(
        # if you wanna use audio input
        # torch_device='cuda',
        # whisper_model='medium',

        model='gpt_model_name',
        api_key='your_openai_api_key',
        base_url='api_url',
        bangumi_url='https://bangumi.tv/subject/424883/',
        bangumi_access_token='your_bangumi_token',
    )

    sub_zh, sub_bilingual = await translator.get_subtitles(sub='path/to/sub.srt') # Or audio='path/to/audio.mp3',
    sub_zh.save('path/to/output_zh.ass')
    sub_bilingual.save('path/to/output_bilingual.ass')

asyncio.run(main())

License

This project is licensed under the GPL-3.0 license - see the LICENSE file for details.