Upload folder using huggingface_hub

persistence.py

@@ -0,0 +1,111 @@
+import os
+import sqlite3
+import json
+import shutil
+from pathlib import Path
+from datetime import datetime
+
+class PersistenceManager:
+    """
+    带云端同步功能的持久化管理器
+    支持本地 SQLite 存储，并可选择同步到 Hugging Face Dataset
+    """
+    def __init__(self, db_path="data.sqlite", dataset_repo=None, hf_token=None):
+        self.db_path = db_path
+        self.dataset_repo = dataset_repo or os.getenv("DATASET_REPO_ID")
+        self.hf_token = hf_token or os.getenv("HF_TOKEN")
+        
+        # 初始化数据库表
+        self._init_db()
+        
+        # 如果在 HF 环境且配置了仓库，则尝试恢复数据
+        if self.dataset_repo and self.hf_token:
+            self.restore_from_cloud()
+
+    def _init_db(self):
+        """初始化数据库结构"""
+        conn = sqlite3.connect(self.db_path)
+        cursor = conn.cursor()
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS settings (
+                key TEXT PRIMARY KEY,
+                value TEXT,
+                updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+            )
+        ''')
+        conn.commit()
+        conn.close()
+
+    def get_all_config(self, default_config):
+        """从数据库加载所有配置"""
+        conn = sqlite3.connect(self.db_path)
+        cursor = conn.cursor()
+        cursor.execute('SELECT key, value FROM settings')
+        rows = cursor.fetchall()
+        conn.close()
+        
+        config = default_config.copy()
+        for key, value in rows:
+            try:
+                # 尝试解析 JSON（以支持列表和字典）
+                config[key] = json.loads(value)
+            except:
+                config[key] = value
+        return config
+
+    def save_config(self, config_dict):
+        """保存配置到数据库并同步到云端"""
+        conn = sqlite3.connect(self.db_path)
+        cursor = conn.cursor()
+        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        
+        for key, value in config_dict.items():
+            val_str = json.dumps(value, ensure_ascii=False)
+            cursor.execute('''
+                INSERT OR REPLACE INTO settings (key, value, updated_at)
+                VALUES (?, ?, ?)
+            ''', (key, val_str, now))
+            
+        conn.commit()
+        conn.close()
+        
+        # 触发云端备份
+        if self.dataset_repo and self.hf_token:
+            self.save_to_cloud()
+
+    def restore_from_cloud(self):
+        """从 Hugging Face Dataset 下载数据库"""
+        try:
+            from huggingface_hub import hf_hub_download
+            print(f"[*] 正在从云端 Dataset [{self.dataset_repo}] 恢复数据库...")
+            
+            downloaded_path = hf_hub_download(
+                repo_id=self.dataset_repo,
+                repo_type="dataset",
+                filename=self.db_path,
+                token=self.hf_token,
+                force_download=True
+            )
+            
+            shutil.copy(downloaded_path, self.db_path)
+            print("✅ 数据库恢复成功")
+        except Exception as e:
+            print(f"⚠️ 云端恢复跳过或失败: {e}")
+
+    def save_to_cloud(self):
+        """将数据库上传到 Hugging Face Dataset"""
+        try:
+            from huggingface_hub import HfApi
+            api = HfApi(token=self.hf_token)
+            print(f"[*] 正在备份数据库到云端 [{self.dataset_repo}]...")
+            
+            api.upload_file(
+                path_or_fileobj=self.db_path,
+                path_in_repo=self.db_path,
+                repo_id=self.dataset_repo,
+                repo_type="dataset",
+                commit_message=f"Persist Gemini cookies at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+            )
+            print("✅ 云端备份成功")
+        except Exception as e:
+            print(f"❌ 云端备份失败: {e}")

requirements.txt

@@ -2,3 +2,4 @@ httpx>=0.25.0
 fastapi>=0.104.0
 uvicorn>=0.24.0
 openai>=1.0.0
+huggingface_hub

server.py

@@ -21,6 +21,15 @@ import httpx
 import hashlib
 import secrets
 import asyncio
+from persistence import PersistenceManager
+
+# ============ 持久化配置 ============
+# 如果要在 HF 上持久化，请设置 DATASET_REPO_ID 环境变量（例如: 'luoluoluo22/gemini-config'）
+pm = PersistenceManager(
+    db_path="data.sqlite", 
+    dataset_repo=os.getenv("DATASET_REPO_ID"),
+    hf_token=os.getenv("HF_TOKEN")
+)
 
 # ============ 配置 ============
 API_KEY = os.getenv("API_KEY", "sk-geminixxxxx")
@@ -139,7 +148,7 @@ DEFAULT_MODEL_IDS = {
     "thinking": "e051ce1aa80aa576",
 }
 
-# 配置存储
+# 配置存储 (内存中的镜像，初始为默认值)
 _config = {
     "SNLM0E": "",
     "SECURE_1PSID": "",
@@ -532,39 +541,14 @@ def parse_tool_calls(content: str) -> tuple:
 
 
 def load_config():
-    """
-    加载配置，优先级:
-    1. config_data.json (前端保存的配置)
-    2. config.py (本地开发配置，仅作为备用)
-    """
+    """从本地 SQLite 数据库加载配置"""
     global _config
-    loaded_from_json = False
-    
-    # 优先从 JSON 文件加载
-    if os.path.exists(CONFIG_FILE):
-        try:
-            with open(CONFIG_FILE, "r", encoding="utf-8") as f:
-                saved = json.load(f)
-                if saved.get("SNLM0E") and saved.get("SECURE_1PSID"):
-                    _config.update(saved)
-                    loaded_from_json = True
-        except:
-            pass
-    
-    # 如果 JSON 没有有效配置，尝试从 config.py 加载
-    if not loaded_from_json:
-        try:
-            import config
-            for key in _config:
-                if hasattr(config, key) and getattr(config, key):
-                    _config[key] = getattr(config, key)
-        except:
-            pass
+    _config = pm.get_all_config(_config)
 
 
 def save_config():
-    with open(CONFIG_FILE, "w", encoding="utf-8") as f:
-        json.dump(_config, f, indent=2, ensure_ascii=False)
+    """保存配置到 SQLite 并同步到云端"""
+    pm.save_config(_config)
 
 
 def get_client(auto_refresh: bool = True):