From 973109a0c2b06492cb8cf545d1c7344bea9362bb Mon Sep 17 00:00:00 2001 From: Ming <58206790+Ming-Zhou0201@users.noreply.github.com> Date: Mon, 29 Jul 2024 20:15:39 +0800 Subject: [PATCH] update webui.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 路径不再需要写入site-packages下的.pth文件中,而是使用os.environ['PYTHONPATH']的方式写入,避免其他项目使用同一个python环境因路径问题产生报错 --- webui.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/webui.py b/webui.py index aee1fa9a..71d12465 100644 --- a/webui.py +++ b/webui.py @@ -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