Notes

import os
import shutil

# Source and destination base
source_base = "TEM DATA"
destination_base = "For_Chetu"

# Project names
projects = ["Alex's Project", "Maiia's Project"]
subfolders = ["Checkpoints", "Datasets", "Notebooks"]

# Create For_Chetu and subfolders
os.makedirs(destination_base, exist_ok=True)

for project in projects:
source_unzipped = os.path.join(source_base, project, "unzipped")
dest_project = os.path.join(destination_base, project)

# Create project folder in For_Chetu
os.makedirs(dest_project, exist_ok=True)

# Create Checkpoints, Datasets, Notebooks
for folder in subfolders:
os.makedirs(os.path.join(dest_project, folder), exist_ok=True)

# Iterate over files/folders in unzipped
for item in os.listdir(source_unzipped):
item_path = os.path.join(source_unzipped, item)

# Decide destination based on keyword
dest_folder = None
lowercase_name = item.lower()
if "checkpoint" in lowercase_name:
dest_folder = "Checkpoints"
elif "dataset" in lowercase_name:
dest_folder = "Datasets"
elif "notebook" in lowercase_name:
dest_folder = "Notebooks"

if dest_folder:
shutil.copytree(item_path, os.path.join(dest_project, dest_folder, item)) if os.path.isdir(item_path) else shutil.copy2(item_path, os.path.join(dest_project, dest_folder))
else:
# Copy to root of Alex/Maiia Project
shutil.copytree(item_path, os.path.join(dest_project, item)) if os.path.isdir(item_path) else shutil.copy2(item_path, dest_project)

print("✅ Folder copy completed. Nothing was moved.")