| 123456789101112131415161718192021222324252627282930313233343536 |
- import os
- import unicodedata
- if not os.getenv("FTLANG_CACHE"):
- current_file_path = os.path.abspath(__file__)
- current_dir = os.path.dirname(current_file_path)
- root_dir = os.path.dirname(current_dir)
- ftlang_cache_dir = os.path.join(root_dir, 'resources', 'fasttext-langdetect')
- os.environ["FTLANG_CACHE"] = str(ftlang_cache_dir)
- # print(os.getenv("FTLANG_CACHE"))
- from fast_langdetect import detect_language
- def detect_lang(text: str) -> str:
- if len(text) == 0:
- return ""
- try:
- lang_upper = detect_language(text)
- except:
- html_no_ctrl_chars = ''.join([l for l in text if unicodedata.category(l)[0] not in ['C', ]])
- lang_upper = detect_language(html_no_ctrl_chars)
- try:
- lang = lang_upper.lower()
- except:
- lang = ""
- return lang
- if __name__ == '__main__':
- print(os.getenv("FTLANG_CACHE"))
- print(detect_lang("This is a test."))
- print(detect_lang("<html>This is a test</html>"))
- print(detect_lang("这个是中文测试。"))
- print(detect_lang("<html>这个是中文测试。</html>"))
|
