AI-driven code generation