Update sync_run.py

sync_run.py

@@ -4,83 +4,59 @@ import subprocess
 import threading
 from huggingface_hub import HfApi, snapshot_download
 
-# 配置
 TOKEN = os.getenv("HF_TOKEN")
 DATASET_ID = os.getenv("DATASET_ID")
 
-# 定义需要监控的目录和文件类型
-SYNC_FOLDERS = ["data", "output", "config"]
-SYNC_EXTENSIONS = [".json", ".db", ".yaml", ".yml", ".txt", ".env"]
+# 定义需要持久化的文件夹和文件
+# 该项目主要数据在 data，配置可能在根目录的 .json 或 .env
+PERSIST_FILES = ["data", "output", "config.json", "settings.json", ".env"]
 
 api = HfApi(token=TOKEN)
 
 def download_data():
-    """启动前：从 Dataset 下载所有备份"""
-    if not DATASET_ID:
-        print("[System] 警告: 未配置 DATASET_ID")
-        return
+    if not DATASET_ID: return
     try:
-        print(f"[System] 正在从 Dataset ({DATASET_ID}) 拉取持久化数据...")
-        # 下载到当前目录，允许覆盖，但排除代码文件防止版本冲突
+        print(f"[System] 正在从 Dataset 拉取持久化数据...")
+        # 仅下载数据文件，不下载代码文件 (.py, .html, .js)
         snapshot_download(
             repo_id=DATASET_ID,
             repo_type="dataset",
-            local_dir=".", 
+            local_dir=".",
             token=TOKEN,
-            ignore_patterns=["*.py", "Dockerfile", "requirements.txt", "logs/*", ".git/*"]
+            allow_patterns=["data/*", "output/*", "*.json", "*.env", "*.db"],
+            ignore_patterns=["*.py", "static/*", "templates/*", "logs/*"]
         )
-        print("[System] 数据拉取完成。")
+        print("[System] 拉取完成。")
     except Exception as e:
-        print(f"[System] 拉取失败 (首次运行或网络问题): {e}")
+        print(f"[System] 拉取失败: {e}")
 
 def upload_data():
-    """运行中：定时将改动上传至 Dataset"""
     while True:
-        # 每 2 分钟备份一次（缩短时间，防止改完密码还没备份就休眠了）
-        time.sleep(120) 
-        if not DATASET_ID:
-            continue
-            
+        time.sleep(120) # 每 2 分钟备份一次
+        if not DATASET_ID: continue
         try:
-            # 扫描当前目录下所有符合条件的文件进行上传
-            # 我们直接上传整个根目录，但通过 allow_patterns 过滤出我们要的文件
+            # 上传所有数据和配置文件
             api.upload_folder(
                 folder_path=".",
                 repo_id=DATASET_ID,
                 repo_type="dataset",
-                commit_message="Auto-backup data and configs",
-                allow_patterns=[
-                    "data/**", 
-                    "output/**", 
-                    "config/**",
-                    "*.json", 
-                    "*.db", 
-                    "*.yaml", 
-                    "*.yml", 
-                    ".env"
-                ],
-                ignore_patterns=["logs/**", "__pycache__/**"],
+                allow_patterns=["data/*", "output/*", "*.json", "*.env", "*.db"],
+                ignore_patterns=["logs/*", "__pycache__/*", "*.py", "static/*", "templates/*"],
                 run_as_future=True
             )
-            print(f"[Backup] {time.strftime('%H:%M:%S')} 备份同步任务已提交")
+            print(f"[Backup] {time.strftime('%H:%M:%S')} 同步已提交至 Dataset")
         except Exception as e:
             if "No files have been modified" not in str(e):
                 print(f"[Backup] 备份出错: {e}")
 
 if __name__ == "__main__":
-    # 确保基础目录存在
-    for folder in SYNC_FOLDERS:
-        os.makedirs(folder, exist_ok=True)
+    os.makedirs("data", exist_ok=True)
+    os.makedirs("output", exist_ok=True)
 
-    # 1. 启动前同步云端数据到本地
     download_data()
+    
+    threading.Thread(target=upload_data, daemon=True).start()
 
-    # 2. 启动后台备份线程
-    backup_thread = threading.Thread(target=upload_data, daemon=True)
-    backup_thread.start()
-
-    # 3. 启动主程序
-    # 注意：根据你的日志，程序默认端口是 8000，但 HF 要求 7860
-    # 我们强制它运行在 7860
-    print("[System] 正在启动 OpenAI/Codex CLI...")
+    print("[System] 正在启动 Web UI...")
+    # 强制端口 7860
     subprocess.run(["python", "webui.py", "--host", "0.0.0.0", "--port", "7860"])