feat: 支持 Hugging Face Spaces 和容器持久化存储

- 自动检测环境：HF Pro 使用 /data，本地使用 ./data
- 统一所有数据文件路径：
- accounts.json: data/accounts.json
- settings.yaml: data/settings.yaml
- stats.json: data/stats.json
- images/: data/images/

Dockerfile

@@ -8,12 +8,14 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && apt-get autoremove -y \
     && rm -rf /var/lib/apt/lists/*
 COPY main.py .
-# 复制 uptime_tracker 模块
-COPY uptime_tracker.py .
 # 复制 core 模块
 COPY core ./core
 # 复制 util 目录
 COPY util ./util
+# 复制 templates 目录
+COPY templates ./templates
+# 复制 static 目录
+COPY static ./static
 # 创建数据目录
 RUN mkdir -p ./data/images
 # 声明数据卷（运行时需要 -v 挂载才能持久化）

core/account.py

@@ -18,8 +18,11 @@ if TYPE_CHECKING:
 
 logger = logging.getLogger(__name__)
 
-# 配置文件路径
-ACCOUNTS_FILE = "accounts.json"
+# 配置文件路径 - 自动检测环境
+if os.path.exists("/data"):
+    ACCOUNTS_FILE = "/data/accounts.json"  # HF Pro 持久化
+else:
+    ACCOUNTS_FILE = "data/accounts.json"  # 本地存储（统一到 data 目录）
 
 
 @dataclass

core/config.py

@@ -87,7 +87,13 @@ class AppConfig(BaseModel):
 class ConfigManager:
     """配置管理器（单例）"""
 
-    def __init__(self, yaml_path: str = "data/settings.yaml"):
+    def __init__(self, yaml_path: str = None):
+        # 自动检测环境并设置默认路径
+        if yaml_path is None:
+            if os.path.exists("/data"):
+                yaml_path = "/data/settings.yaml"  # HF Pro 持久化
+            else:
+                yaml_path = "data/settings.yaml"  # 本地存储
         self.yaml_path = Path(yaml_path)
         self._config: Optional[AppConfig] = None
         self.load()

main.py

@@ -1,4 +1,4 @@
-import json, time, os, asyncio, uuid, ssl, re, yaml
+import json, time, os, asyncio, uuid, ssl, re, yaml, shutil
 from datetime import datetime, timezone, timedelta
 from typing import List, Optional, Union, Dict, Any
 from pathlib import Path
@@ -15,6 +15,27 @@ from util.streaming_parser import parse_json_array_stream_async
 from collections import deque
 from threading import Lock
 
+# ---------- 数据目录配置 ----------
+# 自动检测环境：HF Spaces Pro 使用 /data，本地使用 ./data
+if os.path.exists("/data"):
+    DATA_DIR = "/data"  # HF Pro 持久化存储
+    logger_prefix = "[HF-PRO]"
+else:
+    DATA_DIR = "./data"  # 本地持久化存储
+    logger_prefix = "[LOCAL]"
+
+# 确保数据目录存在
+os.makedirs(DATA_DIR, exist_ok=True)
+
+# 统一的数据文件路径
+ACCOUNTS_FILE = os.path.join(DATA_DIR, "accounts.json")
+SETTINGS_FILE = os.path.join(DATA_DIR, "settings.yaml")
+STATS_FILE = os.path.join(DATA_DIR, "stats.json")
+IMAGE_DIR = os.path.join(DATA_DIR, "images")
+
+# 确保图片目录存在
+os.makedirs(IMAGE_DIR, exist_ok=True)
+
 # 导入认证模块
 from core.auth import verify_api_key
 from core.session_auth import is_logged_in, login_user, logout_user, require_login, generate_session_secret
@@ -59,7 +80,6 @@ log_buffer = deque(maxlen=3000)
 log_lock = Lock()
 
 # 统计数据持久化
-STATS_FILE = "data/stats.json"
 stats_lock = asyncio.Lock()  # 改为异步锁
 
 async def load_stats():
@@ -142,12 +162,6 @@ CHAT_URL = config.public_display.chat_url
 IMAGE_GENERATION_ENABLED = config.image_generation.enabled
 IMAGE_GENERATION_MODELS = config.image_generation.supported_models
 
-# ---------- 图片存储配置 ----------
-if os.path.exists("/data"):
-    IMAGE_DIR = "/data/images"  # HF Pro持久化存储
-else:
-    IMAGE_DIR = "./data/images"  # 本地持久化存储
-
 # ---------- 重试配置 ----------
 MAX_NEW_SESSION_TRIES = config.retry.max_new_session_tries
 MAX_REQUEST_RETRIES = config.retry.max_request_retries
@@ -315,6 +329,15 @@ async def startup_event():
     """应用启动时初始化后台任务"""
     global global_stats
 
+    # 文件迁移逻辑：将根目录的旧文件迁移到 data 目录
+    old_accounts = "accounts.json"
+    if os.path.exists(old_accounts) and not os.path.exists(ACCOUNTS_FILE):
+        try:
+            shutil.copy(old_accounts, ACCOUNTS_FILE)
+            logger.info(f"{logger_prefix} 已迁移 {old_accounts} -> {ACCOUNTS_FILE}")
+        except Exception as e:
+            logger.warning(f"{logger_prefix} 文件迁移失败: {e}")
+
     # 加载统计数据
     global_stats = await load_stats()
     logger.info(f"[SYSTEM] 统计数据已加载: {global_stats['total_requests']} 次请求, {global_stats['total_visitors']} 位访客")