Changed PIL Image importing method to fix error

Forgot to add Optional import

Added callback & callback_steps so I can use in my GUI

initial commit