From 237526a7e68c7425172684bd1cfeff922fdad8ad Mon Sep 17 00:00:00 2001
From: KamioRinn <63162909+KamioRinn@users.noreply.github.com>
Date: Wed, 26 Feb 2025 23:37:19 +0800
Subject: [PATCH] Fix invalid path (#2114)

* Fix korean invalid path

* Fix japanese invalid path

* Fix langsegmenter invalid path
---
 .../text/LangSegmenter/langsegmenter.py       |  2 +-
 GPT_SoVITS/text/japanese.py                   | 34 +++++++++++++++++++
 GPT_SoVITS/text/korean.py                     | 14 +++++---
 3 files changed, 44 insertions(+), 6 deletions(-)

diff --git a/GPT_SoVITS/text/LangSegmenter/langsegmenter.py b/GPT_SoVITS/text/LangSegmenter/langsegmenter.py
index 225e3e1..148c013 100644
--- a/GPT_SoVITS/text/LangSegmenter/langsegmenter.py
+++ b/GPT_SoVITS/text/LangSegmenter/langsegmenter.py
@@ -50,7 +50,7 @@ def load_fasttext_model(
             model = fast_langdetect.ft_detect.infer.fasttext.load_model(str(model_path))
         else:
             python_dir = os.getcwd()
-            if (str(model_path)[:len(python_dir)] == python_dir):
+            if (str(model_path)[:len(python_dir)].upper() == python_dir.upper()):
                 model = fast_langdetect.ft_detect.infer.fasttext.load_model(os.path.relpath(model_path, python_dir))
             else:
                 import tempfile
diff --git a/GPT_SoVITS/text/japanese.py b/GPT_SoVITS/text/japanese.py
index 440062a..d815ef4 100644
--- a/GPT_SoVITS/text/japanese.py
+++ b/GPT_SoVITS/text/japanese.py
@@ -5,6 +5,40 @@ import hashlib
 try:
     import pyopenjtalk
     current_file_path = os.path.dirname(__file__)
+
+    # 防止win下无法读取模型
+    if os.name == 'nt':
+        python_dir = os.getcwd()
+        OPEN_JTALK_DICT_DIR = pyopenjtalk.OPEN_JTALK_DICT_DIR.decode("utf-8")
+        if not (re.match(r'^[A-Za-z0-9_/\\:.]*$', OPEN_JTALK_DICT_DIR)):
+            if (OPEN_JTALK_DICT_DIR[:len(python_dir)].upper() == python_dir.upper()):
+                OPEN_JTALK_DICT_DIR = os.path.join(os.path.relpath(OPEN_JTALK_DICT_DIR,python_dir))
+            else:
+                import shutil
+                if not os.path.exists('TEMP'):
+                    os.mkdir('TEMP')
+                if not os.path.exists(os.path.join("TEMP", "ja")):
+                    os.mkdir(os.path.join("TEMP", "ja"))
+                if os.path.exists(os.path.join("TEMP", "ja", "open_jtalk_dic")):
+                    shutil.rmtree(os.path.join("TEMP", "ja", "open_jtalk_dic"))
+                shutil.copytree(pyopenjtalk.OPEN_JTALK_DICT_DIR.decode("utf-8"), os.path.join("TEMP", "ja", "open_jtalk_dic"), )
+                OPEN_JTALK_DICT_DIR = os.path.join("TEMP", "ja", "open_jtalk_dic")
+            pyopenjtalk.OPEN_JTALK_DICT_DIR = OPEN_JTALK_DICT_DIR.encode("utf-8")
+
+        if not (re.match(r'^[A-Za-z0-9_/\\:.]*$', current_file_path)):
+            if (current_file_path[:len(python_dir)].upper() == python_dir.upper()):
+                current_file_path = os.path.join(os.path.relpath(current_file_path,python_dir))
+            else:
+                if not os.path.exists('TEMP'):
+                    os.mkdir('TEMP')
+                if not os.path.exists(os.path.join("TEMP", "ja")):
+                    os.mkdir(os.path.join("TEMP", "ja"))
+                if not os.path.exists(os.path.join("TEMP", "ja", "ja_userdic")):
+                    os.mkdir(os.path.join("TEMP", "ja", "ja_userdic"))
+                    shutil.copyfile(os.path.join(current_file_path, "ja_userdic", "userdict.csv"),os.path.join("TEMP", "ja", "ja_userdic", "userdict.csv"))
+                current_file_path = os.path.join("TEMP", "ja")
+
+
     def get_hash(fp: str) -> str:
         hash_md5 = hashlib.md5()
         with open(fp, "rb") as f:
diff --git a/GPT_SoVITS/text/korean.py b/GPT_SoVITS/text/korean.py
index 33ea6e7..79d89af 100644
--- a/GPT_SoVITS/text/korean.py
+++ b/GPT_SoVITS/text/korean.py
@@ -28,14 +28,18 @@ if os.name == 'nt':
                             if not (re.match(r'^[A-Za-z0-9_/\\:.]*$', installpath)):
                                 import shutil
                                 python_dir = os.getcwd()
-                                if (installpath[:len(python_dir)] == python_dir):
+                                if (installpath[:len(python_dir)].upper() == python_dir.upper()):
                                     dicpath = os.path.join(os.path.relpath(installpath,python_dir),'data','mecabrc')
                                 else:
-                                    if os.path.exists(os.path.join('komecabdata')):
-                                        shutil.rmtree(os.path.join('komecabdata'))
+                                    if not os.path.exists('TEMP'):
+                                        os.mkdir('TEMP')
+                                    if not os.path.exists(os.path.join('TEMP', 'ko')):
+                                        os.mkdir(os.path.join('TEMP', 'ko'))
+                                    if os.path.exists(os.path.join('TEMP', 'ko', 'ko_dict')):
+                                        shutil.rmtree(os.path.join('TEMP', 'ko', 'ko_dict'))
 
-                                    shutil.copytree(os.path.join(installpath, 'data'), 'komecabdata')
-                                    dicpath = os.path.join('komecabdata', 'mecabrc')
+                                    shutil.copytree(os.path.join(installpath, 'data'), os.path.join('TEMP', 'ko', 'ko_dict'))
+                                    dicpath = os.path.join('TEMP', 'ko', 'ko_dict', 'mecabrc')
                             else:
                                 dicpath=os.path.abspath(os.path.join(installpath, 'data/mecabrc'))
                             return dicpath