When using the Diagram as code (DAC),a nice to have would be expanded editing possibilities for the Ai-generated diagrams → Allow editing Ai-generated diagrams in the same way as Figures created from scratch. Edit the text/colors, nodes etc of the blocks directly on the diagram instead of through the code/Ai.

Editing the code for the block adjusts and moves the lines and the blocks in the diagram to strange orientations. Fortunately there is a "Reset Layout" button, which shows this is something you are likely aware of. However, the automatic resetting of the layout is also not always logical or neat. Making the diagram difficult to follow. Perhaps there is a way to preserve the layout when making edits and only changing the format of the specific block which is being edited by the AI or via the code. So that the layout can remain as required.

Also with DAC, to select multiple blocks to edit and adjust the layouts, you have to use shift+click to select multiple blocks? It would be useful to be able to click and drag to select multiple blocks created through DAC.