From dc194db0dda8cbf9e5d454c2468f6f56f9db0070 Mon Sep 17 00:00:00 2001 From: Piero Toffanin Date: Wed, 8 Nov 2023 21:55:55 -0500 Subject: [PATCH] Refactor translate_file endpoint --- libretranslate/app.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/libretranslate/app.py b/libretranslate/app.py index beb5bb0..d4e4354 100644 --- a/libretranslate/app.py +++ b/libretranslate/app.py @@ -733,12 +733,10 @@ def create_app(args): if os.path.splitext(file.filename)[1] not in frontend_argos_supported_files_format: abort(400, description=_("Invalid request: file format not supported")) - source_langs = [source_lang] - src_langs = [next(iter([l for l in languages if l.code == source_lang]), None) for source_lang in source_langs] + src_lang = next(iter([l for l in languages if l.code == source_lang]), None) - for idx, lang in enumerate(src_langs): - if lang is None: - abort(400, description=_("%(lang)s is not supported", lang=source_langs[idx])) + if src_lang is None: + 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) @@ -751,7 +749,7 @@ def create_app(args): file.save(filepath) - translated_file_path = argostranslatefiles.translate_file(src_langs[0].get_translation(tgt_lang), filepath) + translated_file_path = argostranslatefiles.translate_file(src_lang.get_translation(tgt_lang), filepath) translated_filename = os.path.basename(translated_file_path) return jsonify(