Update docker-entrypoint.sh

docker-entrypoint.sh

@@ -6,29 +6,18 @@ dir_shell=/ql/shell
 
 
 
-echo -e "======================初始化并更新GitHub储存库内容========================\n"
-
-# 从环境变量获取GitHub令牌
-GITHUB_TOKEN=$GITHUB_TOKEN
-
-# 定义GitHub仓库地址
-REPO_URL="https://${GITHUB_TOKEN}@github.com/lang03383/hf_ql_his.git"
-
-# 检查是否已初始化Git仓库
-if [ ! -d "/ql/data/.git" ]; then
-    # 如果没有初始化Git仓库，则克隆仓库
-    git clone $REPO_URL /ql/data
-else
-    # 如果已经初始化了Git仓库，则拉取最新内容
-    cd /ql/data
-    git pull origin main
-fi
-
-echo -e "GitHub内容同步完成...\n"
+python3 -c "from datasets import load_dataset"
 
 echo -e "======================写入rclone配置========================\n"
 echo "$RCLONE_CONF" > ~/.config/rclone/rclone.conf
 
+echo -e "======================从 Datasets 下载数据========================\n"
+
+# 使用 Space Secrets 中的环境变量
+python3 -c "from datasets import load_dataset; load_dataset('$DATASET_NAME', use_auth_token='$ACCESS_TOKEN').save_to_disk('/ql/data')"
+
+echo -e "数据下载完成...\n"
+
 echo -e "======================1. 检测配置文件========================\n"
 import_config "$@"
 make_dir /etc/nginx/conf.d
@@ -61,19 +50,15 @@ if [[ $EnableExtraShell == true ]]; then
   echo -e "自定义脚本后台执行中...\n"
 fi
 
-echo -e "====================7. 启动实时同步到GitHub========================\n"
+echo -e "======================定期备份数据到 Datasets========================\n"
+
+# 设置备份频率 (每分钟备份一次)
+BACKUP_FREQUENCY="* * * * *"
 
-# 设置远程仓库地址
-git remote set-url origin $REPO_URL
+# 创建 cron 任务，使用 Space Secrets 中的环境变量
+(crontab -l 2>/dev/null; echo "$BACKUP_FREQUENCY python3 -c \"from datasets import load_dataset, DatasetDict; DatasetDict.from_disk('/ql/data').push_to_hub('$DATASET_NAME', use_auth_token='$ACCESS_TOKEN', private=True)\"") | crontab -
 
-# 启动实时同步部分
-inotifywait -m -r -e modify,attrib,close_write,move,create,delete /ql/data |
-while read -r directory events filename; do
-    cd /ql/data
-    git add .
-    git commit -m "Automated commit for $filename on $(date)"
-    git push origin main
-done &
+echo -e "数据备份任务已设置...\n"
 
 echo -e "##########写入登陆信息############"
 echo "{ \"username\": \"$ADMIN_USERNAME\", \"password\": \"$ADMIN_PASSWORD\" }" > /ql/data/config/auth.json