mirror of
https://github.com/LibreTranslate/LibreTranslate.git
synced 2024-11-21 23:41:01 +00:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
from functools import lru_cache
|
|
|
|
import linguars
|
|
from lexilang.detector import detect as lldetect
|
|
|
|
|
|
class Language:
|
|
def __init__(self, code, confidence):
|
|
self.code = code
|
|
self.confidence = float(confidence)
|
|
|
|
def __str__(self):
|
|
return (f"code: {self.code:<9} confidence: {self.confidence:>5.1f} ")
|
|
|
|
@lru_cache(maxsize=None)
|
|
def load_detector(langcodes = ()):
|
|
languages = []
|
|
for lc in langcodes:
|
|
try:
|
|
languages.append(linguars.Language.from_iso_code_639_1(lc))
|
|
except Exception:
|
|
print(f"{lc} is not supported by lingua")
|
|
pass # Not supported
|
|
|
|
return linguars.LanguageDetector(languages=languages)
|
|
|
|
|
|
class Detector:
|
|
def __init__(self, langcodes = ()):
|
|
self.langcodes = langcodes
|
|
self.detector = load_detector(langcodes)
|
|
|
|
def detect(self, text):
|
|
if len(text) < 18:
|
|
code, conf = lldetect(text, self.langcodes)
|
|
if conf > 0:
|
|
return [Language(code, round(conf * 100))]
|
|
|
|
top_3_choices = self.detector.confidence(text)[:3]
|
|
if top_3_choices[0][1] == 0:
|
|
return [Language("en", 0)]
|
|
return [Language(lang.iso_code_639_1, round(conf * 100)) for lang, conf in top_3_choices]
|
|
|