import glob
import os
import shutil
from os import path

import constants
import train
from augmentation import Augmentator

# Augmentator().augment_annotations()
# train.train_dataset()
# train.resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')

model_dir = path.join(constants.models_dir, f'{constants.prefix}2025-05-18')

for file in glob.glob(path.join(model_dir, 'weights', 'epoch*')):
    os.remove(file)
shutil.copy(path.join(model_dir, 'weights', 'best.pt'), constants.CURRENT_PT_MODEL)

train.export_current_model()
print('success!')