Clean codebase and add demographic enhancements

Major codebase cleanup and feature additions

Implement categorical demographics and improved two-tower model

initial generation