From 9c0168e0b890f402c4700a6bcef0bb6d00959f54 Mon Sep 17 00:00:00 2001
From: Ming <58206790+Ming-Zhou0201@users.noreply.github.com>
Date: Thu, 1 Aug 2024 22:38:06 +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 | 17 +++++++----------
 1 file changed, 7 insertions(+), 10 deletions(-)

diff --git a/webui.py b/webui.py
index 711b503a..f0660560 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
@@ -923,4 +920,4 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
         share=is_share,
         server_port=webui_port_main,
         quiet=True,
-    )
\ No newline at end of file
+    )