How to add categorical color legend for scatter plot created by graph_objects

Great, thanks a lot!