初始化提交

webui.sh

@@ -0,0 +1,78 @@
+#!/usr/bin/env sh
+
+# If you could not download the model from the official site, you can use the mirror site.
+# Just remove the comment of the following line .
+# 如果你无法从官方网站下载模型，你可以使用镜像网站。
+# 只需要移除下面一行的注释即可。
+
+# export HF_ENDPOINT=https://hf-mirror.com
+
+CURRENT_DIR=$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)
+export PYTHONPATH="$CURRENT_DIR${PYTHONPATH:+:$PYTHONPATH}"
+
+# 0.0.0.0 只能表示“监听所有网卡”，不适合作为浏览器访问地址。
+# macOS/Linux 下浏览器打开 http://0.0.0.0:8501 可能会经过代理或网关，
+# 最终出现 502。默认绑定并打开 127.0.0.1，与 Windows 启动脚本保持一致。
+MPT_WEBUI_HOST="${MPT_WEBUI_HOST:-127.0.0.1}"
+MPT_WEBUI_PORT="${MPT_WEBUI_PORT:-8501}"
+
+if [ -x "$CURRENT_DIR/.venv/bin/python" ]; then
+  PORT_CHECK_CMD="$CURRENT_DIR/.venv/bin/python"
+  set -- "$CURRENT_DIR/.venv/bin/python" -m streamlit
+elif command -v uv >/dev/null 2>&1; then
+  PORT_CHECK_CMD="uv run python"
+  set -- uv run streamlit
+elif command -v streamlit >/dev/null 2>&1; then
+  echo "***** Warning: using streamlit from PATH. If dependencies fail, run 'uv sync --frozen' first. *****"
+  PORT_CHECK_CMD="python3"
+  set -- streamlit
+else
+  echo "***** Neither project Python, uv, nor streamlit was found. Please install dependencies first. *****"
+  exit 1
+fi
+
+find_available_port() {
+  WEBUI_HOST="$MPT_WEBUI_HOST" WEBUI_PORT="$MPT_WEBUI_PORT" "$@" - <<'PY' 2>/dev/null
+import os
+import socket
+import sys
+
+host = os.environ.get("WEBUI_HOST", "127.0.0.1")
+preferred = int(os.environ.get("WEBUI_PORT", "8501"))
+candidates = [preferred] + [port for port in range(8502, 8600) if port != preferred]
+
+for port in candidates:
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
+        try:
+            sock.bind((host, port))
+        except OSError:
+            continue
+        print(port)
+        sys.exit(0)
+
+sys.exit(1)
+PY
+}
+
+# 用 Python 做端口探测，避免依赖 lsof/nc 在不同 macOS/Linux 发行版上的差异。
+# shellcheck disable=SC2086
+SELECTED_WEBUI_PORT=$(find_available_port $PORT_CHECK_CMD)
+
+if [ -z "$SELECTED_WEBUI_PORT" ]; then
+  echo "***** No available WebUI port found in 8501-8599 for $MPT_WEBUI_HOST. *****"
+  exit 1
+fi
+
+if [ "$SELECTED_WEBUI_PORT" != "$MPT_WEBUI_PORT" ]; then
+  echo "***** Port $MPT_WEBUI_PORT is unavailable, using $SELECTED_WEBUI_PORT instead. *****"
+fi
+
+MPT_WEBUI_PORT="$SELECTED_WEBUI_PORT"
+
+echo "***** WebUI address: http://$MPT_WEBUI_HOST:$MPT_WEBUI_PORT *****"
+"$@" run "$CURRENT_DIR/webui/Main.py" \
+  --server.address="$MPT_WEBUI_HOST" \
+  --server.port="$MPT_WEBUI_PORT" \
+  --browser.serverAddress="$MPT_WEBUI_HOST" \
+  --browser.gatherUsageStats=False \
+  --server.enableCORS=True