From 6301b7699ce5f6b1d167eb758330190e9bc84fee Mon Sep 17 00:00:00 2001
From: XXXXRT666 <runtongzhong@gmail.com>
Date: Sat, 13 Jul 2024 12:46:40 +0800
Subject: [PATCH] clean path

---
 tools/uvr5/webui.py | 11 ++++-------
 webui.py            |  1 +
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/tools/uvr5/webui.py b/tools/uvr5/webui.py
index a690a686..8066428a 100644
--- a/tools/uvr5/webui.py
+++ b/tools/uvr5/webui.py
@@ -2,6 +2,7 @@ import os
 import traceback,gradio as gr
 import logging
 from tools.i18n.i18n import I18nAuto
+from tools.my_utils import clean_path
 i18n = I18nAuto()
 
 logger = logging.getLogger(__name__)
@@ -26,13 +27,9 @@ is_share=eval(sys.argv[4])
 def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format0):
     infos = []
     try:
-        inp_root = inp_root.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
-        save_root_vocal = (
-            save_root_vocal.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
-        )
-        save_root_ins = (
-            save_root_ins.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
-        )
+        inp_root = clean_path(inp_root)
+        save_root_vocal = clean_path(save_root_vocal)
+        save_root_ins = clean_path(save_root_ins)
         is_hp3 = "HP3" in model_name
         if model_name == "onnx_dereverb_By_FoxJoy":
             pre_fun = MDXNetDereverb(15)
diff --git a/webui.py b/webui.py
index a200a747..aee1fa9a 100644
--- a/webui.py
+++ b/webui.py
@@ -199,6 +199,7 @@ def open_asr(asr_inp_dir, asr_opt_dir, asr_model, asr_model_size, asr_lang):
     global p_asr
     if(p_asr==None):
         asr_inp_dir=my_utils.clean_path(asr_inp_dir)
+        asr_opt_dir=my_utils.clean_path(asr_opt_dir)
         cmd = f'"{python_exec}" tools/asr/{asr_dict[asr_model]["path"]}'
         cmd += f' -i "{asr_inp_dir}"'
         cmd += f' -o "{asr_opt_dir}"'