Update backup.py

backup.py

@@ -21,55 +21,43 @@ def run_backup():
     os.makedirs(env["XDG_CACHE_HOME"], exist_ok=True)
     os.makedirs(env["TMPDIR"], exist_ok=True)
 
-    local_path = '/home/vscode/workspace'
-     
-    for line in iter(process.stdout.readline, ''):
-        print("[Backup]", line.strip())
-        
+    local_path = "/home/vscode/workspace"
+
+    # --- STEP 1: Delete old files ---
+    print("[Backup] Deleting old files...")
     cmd = [
-        "hf", "repo-files", "delete",
-        BACKUP_REPO,
-        "workspace/",
+        "hf", "repo-files", "delete", BACKUP_REPO, "workspace/",
         "--repo-type", "dataset",
     ]
-
     process = subprocess.Popen(
-        cmd,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        env=env,
-        text=True,
-        cwd="/tmp",
+        cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
+        env=env, text=True, cwd="/tmp"
     )
-
-    for line in iter(process.stdout.readline, ''):
+    for line in iter(process.stdout.readline, ""):
         print("[Backup]", line.strip())
-
     process.wait()
-    
-    # Upload with huggingface-cli
+
+    # --- STEP 2: Upload new backup ---
+    print("[Backup] Uploading workspace...")
     cmd = [
-        "hf", "upload",
-        BACKUP_REPO,
-        local_path,
-        "workspace/",
+        "hf", "upload", BACKUP_REPO, local_path, "workspace/",
         "--repo-type", "dataset",
     ]
-
     process = subprocess.Popen(
-        cmd,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        env=env,
-        text=True,
-        cwd="/tmp",
+        cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
+        env=env, text=True, cwd="/tmp"
     )
+    for line in iter(process.stdout.readline, ""):
+        print("[Backup]", line.strip())
+    process.wait()
+
+    print("[Backup] Completed at", datetime.datetime.utcnow().isoformat(), "UTC")
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     while True:
-        print('Backup started.')
+        print("[Backup] Backup started.")
         try:
             run_backup()
         except Exception as e:
             print("[Backup] Failed:", e)
-        time.sleep(45 * 60)  # 45 minutes
+        time.sleep(45 * 60)  # 45 minutes