Update start.sh

start.sh

@@ -5,7 +5,8 @@ MUSIC_DIR=${MUSIC_DIR:-/music}
 DATASET_MUSIC_NAME=${DATASET_MUSIC_NAME:-"your-username/music-dataset"}
 MUSIC_TOKEN=${MUSIC_TOKEN:-""}
 BACKUP_DATASET_ID=${BACKUP_DATASET_ID:-"your-username/navidrome-backup"}
-BACKUP_INTERVAL=${BACKUP_INTERVAL:-3600}  # 默认3600秒备份一次
+BACKUP_INTERVAL=${BACKUP_INTERVAL:-3600}  # 默认1小时备份一次
+MUSIC_UPDATE_INTERVAL=${MUSIC_UPDATE_INTERVAL:-3600}  # 默认1小时更新一次音乐
 
 echo "[INFO] Starting Navidrome setup"
 
@@ -20,42 +21,20 @@ chmod -R 755 /.cache
 # 激活Python虚拟环境
 source /venv/bin/activate
 
-# 恢复备份
+# 恢复备份（如果存在）
 if [ -n "$BACKUP_DATASET_ID" ] && [ -n "$MUSIC_TOKEN" ]; then
     echo "[INFO] Attempting to restore from backup..."
     python /backup.py download "$MUSIC_TOKEN" "$BACKUP_DATASET_ID" "/data"
 fi
 
-# 下载数据集中的音乐文件到音乐目录
+# 启动音乐更新进程
 if [ -n "$DATASET_MUSIC_NAME" ] && [ -n "$MUSIC_TOKEN" ]; then
-    echo "[INFO] Downloading music files from Hugging Face Dataset..."
-    python -c "
-from huggingface_hub import snapshot_download
-import os
-import shutil
-# 设置环境变量
-os.environ['HF_TOKEN'] = '${MUSIC_TOKEN}'
-music_dir = '${MUSIC_DIR}'
-dataset_name = '${DATASET_MUSIC_NAME}'
-try:
-    # 确保缓存目录存在并有正确权限
-    os.makedirs('/.cache', exist_ok=True)
-    
-    # 下载数据集
-    print(f'Downloading dataset {dataset_name} to {music_dir}')
-    snapshot_download(
-        repo_id=dataset_name,
-        repo_type='dataset',
-        local_dir=music_dir,
-        token=os.environ['HF_TOKEN']
-    )
-    print('Download completed successfully')
-except Exception as e:
-    print(f'Error downloading dataset: {str(e)}')
-    # 创建测试文件确保目录不为空
-    with open(os.path.join(music_dir, 'test.txt'), 'w') as f:
-        f.write('This is a test file')
-"
+    echo "[INFO] Starting music update process..."
+    python /update_music.py "$DATASET_MUSIC_NAME" "$MUSIC_TOKEN" "$MUSIC_DIR" "$MUSIC_UPDATE_INTERVAL" "false" &
+    MUSIC_UPDATE_PID=$!
+    echo "[INFO] Music update process started with PID: $MUSIC_UPDATE_PID"
+else
+    echo "[WARNING] Music update disabled. Set DATASET_MUSIC_NAME and MUSIC_TOKEN to enable."
 fi
 
 # 检查音乐目录
@@ -83,6 +62,8 @@ backup_data() {
 if [ -n "$BACKUP_DATASET_ID" ] && [ -n "$MUSIC_TOKEN" ]; then
     echo "[INFO] Starting backup process with dataset: ${BACKUP_DATASET_ID}"
     backup_data &
+    BACKUP_PID=$!
+    echo "[INFO] Backup process started with PID: $BACKUP_PID"
 else
     echo "[WARNING] Backup disabled. Set BACKUP_DATASET_ID and MUSIC_TOKEN to enable."
 fi