It sounds like you and I are thinking along the same line: https://github.com/joemccall86/cap5600-project2.

I am also taking AI this semester and was considering expanding upon something like this for my masters' capstone project. The application of machine learning to this type of situation is going to be very interesting to say the least.

Just wanted to say good luck!