Create ai.py

OLD ONE still working on the new one