From db9f92a1c3f5f1f9b961dc69c5177c4ccf6364c2 Mon Sep 17 00:00:00 2001 From: Ng Ky <15950171+ng-ky@users.noreply.github.com> Date: Sun, 27 Oct 2024 20:35:11 +0800 Subject: [PATCH] Add support for "auto" when translating files --- libretranslate/app.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libretranslate/app.py b/libretranslate/app.py index b210a94..cfedea8 100644 --- a/libretranslate/app.py +++ b/libretranslate/app.py @@ -830,7 +830,7 @@ def create_app(args): src_lang = next(iter([l for l in languages if l.code == source_lang]), None) - if src_lang is None: + if src_lang is None and source_lang != "auto": abort(400, description=_("%(lang)s is not supported", lang=source_lang)) tgt_lang = next(iter([l for l in languages if l.code == target_lang]), None) @@ -852,6 +852,11 @@ def create_app(args): if char_limit > 0: request.req_cost = max(1, int(os.path.getsize(filepath) / char_limit)) + if source_lang == "auto": + src_texts = argostranslatefiles.get_texts(filepath) + candidate_langs = detect_languages(src_texts) + src_lang = candidate_langs[0] + translated_file_path = argostranslatefiles.translate_file(src_lang.get_translation(tgt_lang), filepath) translated_filename = os.path.basename(translated_file_path)