Diffusion Based Data Augmentation Python Code Runner Offline