🐛 Try to fix the language detect issue

This commit is contained in:
LittleSheep 2024-07-28 22:01:24 +08:00
parent 27994733dd
commit 7ae1f8021b

View File

@ -6,12 +6,20 @@ import (
"github.com/pemistahl/lingua-go"
)
func DetectLanguage(content string) string {
return "unknown"
var detector lingua.LanguageDetector
detector := lingua.NewLanguageDetectorBuilder().
FromLanguages(lingua.AllLanguages()...).
func CreateLanguageDetector() lingua.LanguageDetector {
return lingua.NewLanguageDetectorBuilder().
FromAllLanguages().
WithLowAccuracyMode().
Build()
}
func DetectLanguage(content string) string {
if detector == nil {
detector = CreateLanguageDetector()
}
if lang, ok := detector.DetectLanguageOf(content); ok {
return strings.ToLower(lang.String())
}