How to detect colors using Python