Add get_hf_hub and make_safe_directory_name

Author: wannaphong

pythainlp/corpus/__init__.py

@@ -23,6 +23,7 @@
     "get_corpus_default_db",
     "get_corpus_path",
     "get_path_folder_corpus",
+    "get_hf_hub",
     "path_pythainlp_corpus",
     "provinces",
     "remove",
@@ -41,6 +42,7 @@
     "thai_wikipedia_titles",
     "thai_words",
     "thai_wsd_dict",
+    "make_safe_directory_name",
 ]
 
 import os
@@ -98,6 +100,8 @@ def corpus_db_path() -> str:
     get_corpus_default_db,
     get_corpus_path,
     get_path_folder_corpus,
+    make_safe_directory_name,
+    get_hf_hub,
     path_pythainlp_corpus,
     remove,
 )  # these imports must come before other pythainlp.corpus.* imports

pythainlp/corpus/core.py

@@ -8,6 +8,7 @@
 
 import json
 import os
+import re
 from typing import Union
 
 from pythainlp import __version__
@@ -614,3 +615,52 @@ def remove(name: str) -> bool:
 
 def get_path_folder_corpus(name, version, *path):
     return os.path.join(get_corpus_path(name, version), *path)
+
+
+def make_safe_directory_name(name:str) -> str:
+    """
+    Make safe directory name
+
+    :param str name: directory name
+    :return: safe directory name
+    :rtype: str
+    """
+    # Replace invalid characters with an underscore
+    safe_name = re.sub(r'[<>:"/\\|?*]', '_', name)
+    # Remove leading/trailing spaces or periods (especially important for Windows)
+    safe_name = safe_name.strip(' .')
+    # Prevent names that are reserved on Windows
+    reserved_names = ['CON', 'PRN', 'AUX', 'NUL', 'COM1', 'COM2', 'COM3', 'COM4', 'COM5', 'COM6', 'COM7', 'COM8', 'COM9', 'LPT1', 'LPT2', 'LPT3', 'LPT4', 'LPT5', 'LPT6', 'LPT7', 'LPT8', 'LPT9']
+    if safe_name.upper() in reserved_names:
+        safe_name = f"_{safe_name}" # Prepend underscore to avoid conflict
+    return safe_name
+
+
+def get_hf_hub(repo_id:str, filename: str=None) -> str:
+    """
+    HuggingFace Hub in pythainlp-home
+
+    :param str repo_id: repo_id
+    :param str filename: filename
+    :return: path
+    :rtype: str
+    """
+    if _CHECK_MODE == "1":
+        print("PyThaiNLP is read-only mode. It can't download.")
+        return False
+    from huggingface_hub import hf_hub_download, snapshot_download
+    hf_root = get_full_data_path("hf_models")
+    name_dir = make_safe_directory_name(repo_id)
+    root_project = os.path.join(hf_root, name_dir)
+    if filename!=None:
+        output_path = hf_hub_download(
+            repo_id=repo_id,
+            filename=filename,
+            local_dir=root_project
+        )
+    else:
+        output_path = snapshot_download(
+            repo_id=repo_id,
+            local_dir=root_project
+        )
+    return output_path