Upload 8 files

.env.example

@@ -13,6 +13,18 @@ ADMIN_KEY=your-admin-secret-key
 # 服务端口（可选，默认 7860）
 # PORT=7860
 
+# ============================================
+# 数据库配置（可选，用于无持久化存储的环境如 HF Spaces）
+# ============================================
+# 支持 PostgreSQL 数据库存储（账户/设置/统计）
+# 未设置时使用本地文件存储（原有行为）
+#
+# 示例（Neon PostgreSQL）：
+# DATABASE_URL=postgresql://user:password@ep-xxx.aws.neon.tech/dbname?sslmode=require
+#
+# 注意：使用数据库存储需要安装 asyncpg：pip install asyncpg
+# DATABASE_URL=
+
 # ============================================
 # 其他配置请在管理面板的"系统设置"中配置
 # 包括：API密钥、代理、图片生成、重试策略等

.gitignore

@@ -24,7 +24,7 @@ wheels/
 venv/
 env/
 ENV/
-
+.venv
 # IDE
 .vscode/
 .idea/

main.py

@@ -72,6 +72,9 @@ from core import uptime as uptime_tracker
 # 导入配置管理和模板系统
 from core.config import config_manager, config
 
+# 数据库存储支持
+from core import storage
+
 # ---------- 日志配置 ----------
 
 # 内存日志缓冲区 (保留最近 3000 条日志，重启后清空)
@@ -83,6 +86,13 @@ stats_lock = asyncio.Lock()  # 改为异步锁
 
 async def load_stats():
     """加载统计数据（异步）。"""
+    if storage.is_database_enabled():
+        try:
+            data = await asyncio.to_thread(storage.load_stats_sync)
+            if isinstance(data, dict):
+                return data
+        except Exception as e:
+            logger.error(f"[STATS] 数据库加载失败: {str(e)[:50]}")
     try:
         if os.path.exists(STATS_FILE):
             async with aiofiles.open(STATS_FILE, 'r', encoding='utf-8') as f:
@@ -104,6 +114,13 @@ async def load_stats():
 
 async def save_stats(stats):
     """保存统计数据（异步，避免阻塞事件循环）"""
+    if storage.is_database_enabled():
+        try:
+            saved = await asyncio.to_thread(storage.save_stats_sync, stats)
+            if saved:
+                return
+        except Exception as e:
+            logger.error(f"[STATS] 数据库保存失败: {str(e)[:50]}")
     try:
         async with aiofiles.open(STATS_FILE, 'w', encoding='utf-8') as f:
             await f.write(json.dumps(stats, ensure_ascii=False, indent=2))

requirements.txt

@@ -7,4 +7,8 @@ python-dotenv==1.0.1
 itsdangerous==2.1.2
 python-multipart==0.0.6
 pyyaml>=6.0
-jinja2>=3.1.0
+jinja2>=3.1.0
+
+# 可选：PostgreSQL 数据库支持（用于 HF Spaces 等无持久化存储的环境）
+# 如需使用，请取消下行注释并设置 DATABASE_URL 环境变量
+# asyncpg>=0.29.0