From 2499c6c37997a1bdb91651fbdb8783bcc9aeef2a Mon Sep 17 00:00:00 2001
From: Kyle Harrity <hpkyh47@gmail.com>
Date: Fri, 25 Mar 2022 02:19:56 -0400
Subject: [PATCH] include language detected, confidence for /translate calls
with auto detect
---
app/app.py | 49 +++++++++++++++++++++++++++++++++----------------
1 file changed, 33 insertions(+), 16 deletions(-)
diff --git a/app/app.py b/app/app.py
index 7b42c98..3979c2e 100644
--- a/app/app.py
+++ b/app/app.py
@@ -446,18 +446,18 @@ def create_app(args):
else:
# Unable to accurately detect languages for short texts
candidate_langs = overall_candidates
- source_langs.append(candidate_langs[0]["language"])
+ source_langs.append(candidate_langs[0])
if args.debug:
print(text_to_check, candidate_langs)
print("Auto detected: %s" % candidate_langs[0]["language"])
else:
if batch:
- source_langs = [source_lang for text in q]
+ source_langs = [ {"confidence": 100.0, "language": source_lang} for text in q]
else:
- source_langs = [source_lang]
+ source_langs = [ {"confidence": 100.0, "language": source_lang} ]
- src_langs = [next(iter([l for l in languages if l.code == source_lang]), None) for source_lang in source_langs]
+ src_langs = [next(iter([l for l in languages if l.code == source_lang["language"]]), None) for source_lang in source_langs]
for idx, lang in enumerate(src_langs):
if lang is None:
@@ -483,26 +483,43 @@ def create_app(args):
if text_format == "html":
translated_text = str(translate_html(translator, text))
else:
- translated_text = translator.translate(transliterate(text, target_lang=source_langs[idx]))
+ translated_text = translator.translate(transliterate(text, target_lang=source_langs[idx]["language"]))
results.append(unescape(translated_text))
- return jsonify(
- {
- "translatedText": results
- }
- )
+ if source_lang == "auto":
+ return jsonify(
+ {
+ "translatedText": results,
+ "detectedLanguage": source_langs
+ }
+ )
+ else:
+ return jsonify(
+ {
+ "translatedText": results
+ }
+ )
else:
translator = src_langs[0].get_translation(tgt_lang)
if text_format == "html":
translated_text = str(translate_html(translator, q))
else:
- translated_text = translator.translate(transliterate(q, target_lang=source_langs[0]))
- return jsonify(
- {
- "translatedText": unescape(translated_text)
- }
- )
+ translated_text = translator.translate(transliterate(q, target_lang=source_langs[0]["language"]))
+
+ if source_lang == "auto":
+ return jsonify(
+ {
+ "translatedText": unescape(translated_text),
+ "detectedLanguage": source_langs[0]
+ }
+ )
+ else:
+ return jsonify(
+ {
+ "translatedText": unescape(translated_text)
+ }
+ )
except Exception as e:
abort(500, description="Cannot translate text: %s" % str(e))
--
GitLab