Update Dockerfile

Dockerfile

@@ -10,8 +10,13 @@ RUN apt-get update && \
 RUN git clone https://github.com/ErlichLiu/DeepClaude.git /app
 WORKDIR /app
 
-# 安装Python依赖（包括新增的 colorlog）
-RUN pip install --no-cache-dir fastapi uvicorn python-dotenv huggingface_hub colorlog
+# 安装必要的依赖
+RUN pip install --no-cache-dir fastapi uvicorn python-dotenv huggingface_hub
+
+# 自动检测可能需要的依赖（包括colorlog、tiktoken等）
+RUN grep -r "import" --include="*.py" /app | grep -v "from app" | sort | uniq | \
+    grep -oP "(?<=import )[a-zA-Z0-9_]+|(?<=from )[a-zA-Z0-9_]+" | sort | uniq | \
+    xargs -I{} pip install --no-cache-dir {} || true
 
 # 如果仓库中有requirements.txt，也安装其中的依赖
 RUN if [ -f requirements.txt ]; then \
@@ -24,11 +29,60 @@ RUN mkdir -p /app/data && chmod -R 777 /app/data
 # 创建.env文件
 RUN touch .env
 
-# 复制备份脚本（稍后单独创建）
-COPY sync_data.sh /app/sync_data.sh
-RUN chmod +x /app/sync_data.sh
-
+# 暴露Hugging Face默认端口
 EXPOSE 7860
 
+# 创建备份脚本 sync_data.sh
+RUN echo '#!/bin/bash\n\
+if [ -z "$HF_TOKEN" ] || [ -z "$DATASET_ID" ]; then\n\
+    echo "Starting without backup functionality - missing HF_TOKEN or DATASET_ID"\n\
+    exec uvicorn app.main:app --host 0.0.0.0 --port 7860\n\
+    exit 0\n\
+fi\n\
+export HUGGING_FACE_HUB_TOKEN=$HF_TOKEN\n\
+restore_latest() {\n\
+python3 -c "\n\
+from huggingface_hub import HfApi\n\
+import os\n\
+api = HfApi()\n\
+files = api.list_repo_files(repo_id='\''$DATASET_ID'\'', repo_type='\''dataset'\'')\n\
+backup_files = sorted([f for f in files if f.startswith('\''backup_'\'') and f.endswith('\''.tar.gz'\'')])\n\
+if backup_files:\n\
+ latest = backup_files[-1]\n\
+ api.hf_hub_download(repo_id='\''$DATASET_ID'\'', filename=latest, repo_type='\''dataset'\'', local_dir='\''/tmp'\'')\n\
+ os.system(f'\''tar -xzf /tmp/{latest} -C /app'\'')\n\
+ os.remove(f'\''/tmp/{latest}'\'')\n\
+ print(f'\''Restored from {latest}'\'')\n\
+else:\n\
+ print('\''No backup found'\'')\n\
+"\n\
+}\n\
+sync_data() {\n\
+while true; do\n\
+ cd /app\n\
+ timestamp=$(date +%Y%m%d_%H%M%S)\n\
+ backup_file="backup_${timestamp}.tar.gz"\n\
+ if [ -d "data" ] && [ "$(ls -A data 2>/dev/null)" ]; then\n\
+  tar -czf "/tmp/${backup_file}" data/\n\
+  python3 -c "\n\
+from huggingface_hub import HfApi\n\
+api = HfApi()\n\
+api.upload_file(path_or_fileobj='\''/tmp/${backup_file}'\'', path_in_repo='\''${backup_file}'\'', repo_id='\''${DATASET_ID}'\'', repo_type='\''dataset'\'')\n\
+print('\''Backup uploaded successfully'\'')\n\
+backup_files = sorted([f for f in api.list_repo_files('\''${DATASET_ID}'\'', repo_type='\''dataset'\'') if f.startswith('\''backup_'\'')])\n\
+for old_backup in backup_files[:-50]:\n\
+ api.delete_file(path_in_repo=old_backup, repo_id='\''${DATASET_ID}'\'', repo_type='\''dataset'\'')\n\
+ print(f'\''Deleted old backup: {old_backup}'\'')\n\
+"\n\
+  rm -f "/tmp/${backup_file}"\n\
+ else\n\
+  echo "No data to backup"\n\
+ fi\n\
+ sleep ${SYNC_INTERVAL:-7200}\n\
+done\n\
+}\n\
+(restore_latest; sync_data &) \n\
+exec uvicorn app.main:app --host 0.0.0.0 --port 7860\n' > /app/sync_data.sh && chmod +x /app/sync_data.sh
+
 # 使用备份脚本启动服务
 CMD ["/app/sync_data.sh"]