Create ai.py