Commit 5b761b49 authored by w-e-w's avatar w-e-w

correct webpath when webui_dir is not work_dir

parent 6fb2194d
import os import os
import sys import sys
from modules.paths_internal import models_path, script_path, data_path, extensions_dir, extensions_builtin_dir # noqa: F401 from modules.paths_internal import models_path, script_path, data_path, extensions_dir, extensions_builtin_dir, cwd # noqa: F401
import modules.safe # noqa: F401 import modules.safe # noqa: F401
......
...@@ -8,6 +8,7 @@ import shlex ...@@ -8,6 +8,7 @@ import shlex
commandline_args = os.environ.get('COMMANDLINE_ARGS', "") commandline_args = os.environ.get('COMMANDLINE_ARGS', "")
sys.argv += shlex.split(commandline_args) sys.argv += shlex.split(commandline_args)
cwd = os.getcwd()
modules_path = os.path.dirname(os.path.realpath(__file__)) modules_path = os.path.dirname(os.path.realpath(__file__))
script_path = os.path.dirname(modules_path) script_path = os.path.dirname(modules_path)
......
...@@ -2,12 +2,12 @@ import os ...@@ -2,12 +2,12 @@ import os
import gradio as gr import gradio as gr
from modules import localization, shared, scripts from modules import localization, shared, scripts
from modules.paths import script_path, data_path from modules.paths import script_path, data_path, cwd
def webpath(fn): def webpath(fn):
if fn.startswith(script_path): if fn.startswith(cwd):
web_path = os.path.relpath(fn, script_path).replace('\\', '/') web_path = os.path.relpath(fn, cwd)
else: else:
web_path = os.path.abspath(fn) web_path = os.path.abspath(fn)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment