From fd4c42b42edfbbe96131f4d878b440f7c2ee277e Mon Sep 17 00:00:00 2001
From: EntropyYue <xiaopa233@qq.com>
Date: Sat, 7 Jun 2025 23:14:21 +0800
Subject: [PATCH] feat: add support for WEBUI_SECRET_KEY_FILE

---
 backend/start.sh          | 6 +++++-
 backend/start_windows.bat | 4 ++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/backend/start.sh b/backend/start.sh
index 84d5ec895..9e106760c 100755
--- a/backend/start.sh
+++ b/backend/start.sh
@@ -14,7 +14,11 @@ if [[ "${WEB_LOADER_ENGINE,,}" == "playwright" ]]; then
     python -c "import nltk; nltk.download('punkt_tab')"
 fi
 
-KEY_FILE=.webui_secret_key
+if [ -n "${WEBUI_SECRET_KEY_FILE}" ]; then
+    KEY_FILE="${WEBUI_SECRET_KEY_FILE}"
+else
+    KEY_FILE=".webui_secret_key"
+fi
 
 PORT="${PORT:-8080}"
 HOST="${HOST:-0.0.0.0}"
diff --git a/backend/start_windows.bat b/backend/start_windows.bat
index 8d9aae3ac..e38fdb2aa 100644
--- a/backend/start_windows.bat
+++ b/backend/start_windows.bat
@@ -18,6 +18,10 @@ IF /I "%WEB_LOADER_ENGINE%" == "playwright" (
 )
 
 SET "KEY_FILE=.webui_secret_key"
+IF NOT "%WEBUI_SECRET_KEY_FILE%" == "" (
+    SET "KEY_FILE=%WEBUI_SECRET_KEY_FILE%"
+)
+
 IF "%PORT%"=="" SET PORT=8080
 IF "%HOST%"=="" SET HOST=0.0.0.0
 SET "WEBUI_SECRET_KEY=%WEBUI_SECRET_KEY%"