file strcture changes

Author: her3ticAVI

scripts/cleanup.py

@@ -2,12 +2,22 @@
 import re
 
 # Paths
-base_dir = os.path.dirname(os.path.abspath(__file__))
-labs_dir = os.path.join(base_dir, 'labs')
-readme_path = os.path.join(base_dir, 'README.md')
+base_dir = os.path.dirname(os.path.abspath(__file__))           # /scripts
+repo_root = os.path.abspath(os.path.join(base_dir, ".."))       # repo root
+labs_dir = os.path.join(repo_root, 'labs')
+readme_path = os.path.join(repo_root, 'README.md')
+
+# 0. Safety checks
+if not os.path.exists(readme_path):
+    print(f"README not found at {readme_path}, skipping cleanup.")
+    exit(0)
+
+if not os.path.exists(labs_dir):
+    print(f"Labs directory not found at {labs_dir}, skipping cleanup.")
+    exit(0)
 
 # 1. Remove files named "tmp" or "tnp" with no extension
-for root, dirs, files in os.walk(base_dir):
+for root, dirs, files in os.walk(repo_root):
     for file in files:
         if file in ('tmp', 'tnp'):
             file_path = os.path.join(root, file)