AI-powered stage design