Eu ia dizer que te salvaria nesse desafio, mas a stack é outra... pra quem quiser fazer um de IA, checa o FastroAI.