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