Create 4 in a row game in python console application

Hello, I have an existing code of client/server 4 in a row python console application. I need to refactor existing code and add some more functionallity that missing for example AI logic (when player want to plays against the computer). the code should run on Linux.