Update webui.py

路径不需要写入site-packages下的.pth文件中,而是使用os.environ['PYTHONPATH']的方式写入,避免其他项目使用同一个python环境因路径问题产生报错
This commit is contained in:
Ming 2024-08-01 22:38:06 +08:00 committed by GitHub
parent 4c8b761220
commit 9c0168e0b8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -32,15 +32,12 @@ os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
os.environ["all_proxy"] = ""
for site_packages_root in site_packages_roots:
if os.path.exists(site_packages_root):
try:
with open("%s/users.pth" % (site_packages_root), "w") as f:
f.write(
"%s\n%s/tools\n%s/tools/damo_asr\n%s/GPT_SoVITS\n%s/tools/uvr5"
% (now_dir, now_dir, now_dir, now_dir, now_dir)
)
break
except PermissionError:
pass
for name in ["tools","tools/damo_asr","GPT_SoVITS","tools/uvr5"]:
if "PYTHONPATH" in os.environ:
os.environ["PYTHONPATH"] = os.path.join(now_dir, name)+ os.pathsep + os.environ["PYTHONPATH"]
else:
os.environ["PYTHONPATH"] = os.path.join(now_dir, name)+ os.pathsep
sys.path = os.environ['PYTHONPATH'].split(os.pathsep) + sys.path
from tools import my_utils
import traceback
import shutil
@ -923,4 +920,4 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
share=is_share,
server_port=webui_port_main,
quiet=True,
)
)