mirror of
https://github.com/MCV-Software/TWBlue.git
synced 2025-07-18 06:06:06 -04:00
change: Improved translation services' settings. Closes #632
This commit is contained in:
@@ -3,12 +3,12 @@ import config
|
||||
from deepl import Translator
|
||||
|
||||
def translate(text: str, target_language: str) -> str:
|
||||
key = config.app["translator"]["translator_api_key"]
|
||||
key = config.app["translator"]["deepl_api_key"]
|
||||
t = Translator(key)
|
||||
return t.translate_text(text, target_lang=target_language).text
|
||||
|
||||
def languages():
|
||||
key = config.app["translator"]["translator_api_key"]
|
||||
key = config.app["translator"]["deepl_api_key"]
|
||||
t = Translator(key)
|
||||
langs = t.get_target_languages()
|
||||
return langs
|
@@ -17,9 +17,9 @@ class TranslatorController(object):
|
||||
self.response = False
|
||||
self.dialog = translateDialog()
|
||||
pub.subscribe(self.on_engine_changed, "translator.engine_changed")
|
||||
if config.app["translator"]["engine"] == "libretranslate":
|
||||
if config.app["translator"]["engine"] == "LibreTranslate":
|
||||
self.dialog.engine_select.SetSelection(0)
|
||||
elif config.app["translator"]["engine"] == "deepl":
|
||||
elif config.app["translator"]["engine"] == "DeepL":
|
||||
self.dialog.engine_select.SetSelection(1)
|
||||
threading.Thread(target=self.load_languages).start()
|
||||
if self.dialog.ShowModal() == wx.ID_OK:
|
||||
@@ -35,27 +35,24 @@ class TranslatorController(object):
|
||||
self.dialog.set_languages(self.languages)
|
||||
|
||||
def on_engine_changed(self, engine):
|
||||
if engine == "LibreTranslate":
|
||||
config.app["translator"]["engine"] = engine.lower()
|
||||
elif engine == "DeepL":
|
||||
config.app["translator"]["engine"] = engine.lower()
|
||||
config.app["translator"]["engine"] = engine
|
||||
config.app.write()
|
||||
threading.Thread(target=self.load_languages).start()
|
||||
|
||||
def translate(self):
|
||||
log.debug("Received translation request for language %s, text=%s" % (self.target_language, self.text))
|
||||
if config.app["translator"].get("engine") == "libretranslate":
|
||||
translator = libre_translate.CustomLibreTranslateAPI(config.app["translator"]["translator_api_url"])
|
||||
if config.app["translator"].get("engine") == "LibreTranslate":
|
||||
translator = libre_translate.CustomLibreTranslateAPI(config.app["translator"]["lt_api_url"], config.app["translator"]["lt_api_key"])
|
||||
vars = dict(q=self.text, target=self.target_language)
|
||||
return translator.translate(**vars)
|
||||
elif config.app["translator"]["engine"] == "deepl" and config.app["translator"]["translator_api_key"] != "":
|
||||
elif config.app["translator"]["engine"] == "DeepL" and config.app["translator"]["deepl_api_key"] != "":
|
||||
return deep_l.translate(text=self.text, target_language=self.target_language)
|
||||
|
||||
def get_languages(self):
|
||||
languages = {}
|
||||
if config.app["translator"].get("engine") == "libretranslate":
|
||||
translator = libre_translate.CustomLibreTranslateAPI(config.app["translator"]["translator_api_url"])
|
||||
if config.app["translator"].get("engine") == "LibreTranslate":
|
||||
translator = libre_translate.CustomLibreTranslateAPI(config.app["translator"]["lt_api_url"], config.app["translator"]["lt_api_key"])
|
||||
languages = {l.get("code"): l.get("name") for l in translator.languages()}
|
||||
elif config.app["translator"]["engine"] == "deepl" and config.app["translator"]["translator_api_key"] != "":
|
||||
elif config.app["translator"]["engine"] == "DeepL" and config.app["translator"]["deepl_api_key"] != "":
|
||||
languages = {language.code: language.name for language in deep_l.languages()}
|
||||
return dict(sorted(languages.items(), key=lambda x: x[1]))
|
||||
|
Reference in New Issue
Block a user