AI diagram generator

AI UML diagram generator.

In practice teams use about four of UML's fourteen diagram types, and this draws those four properly rather than all of them badly.

Download for macOS
v0.1.33 · .dmg · Apple Silicon & Intel
Signed & notarized by Apple · opens without a Gatekeeper warning
sha256 698955a0187bc039f4c74f5d05a9f10fbb27376a45788a0a241d1326b73873c7
Download for Windows instead
$curl -fsSL https://lucidtrain.com/install.sh | sh

The UML worth using

UML defines fourteen diagram types and working teams use a handful: sequence for interactions, component for structure, class for domain models, and state for anything with a lifecycle. The others are either specialised or have been quietly replaced by better-fitting conventions such as C4. Treating UML as a vocabulary to borrow from rather than a standard to comply with is what keeps it useful.

How Lucid Train draws one

Describe what you want, or point it at a repository. The agent emits a semantic graph of nodes and edges with no coordinates, and the ELK layout engine computes positions and orthogonal routing. That separation is the reason the output cannot come back with boxes overlapping or arrows crossing through cards, which is the characteristic failure when a language model is asked to place things on a canvas itself. It also means editing is editing the graph: change a component and the layout re-solves rather than drifting out of alignment.

State diagrams are the underused one

Anything with a status column has a state machine, and it is almost never written down: orders, subscriptions, deployments, tickets. Drawing the states and the permitted transitions usually reveals transitions the code allows that nobody intended, which makes this one of the few diagram types that regularly finds bugs.

Specification compliance is not the goal

This produces diagrams that read correctly to an engineer familiar with UML, not diagrams that would pass validation against the specification. If you need formal UML for a regulated deliverable or model-driven code generation, a dedicated modelling tool is the right choice and this is not it.

Local, offline, your models

It runs as a desktop application. Against a local Ollama model the whole thing works offline with no API key and no telemetry, which is the practical difference from every browser-based diagram tool: your architecture never leaves the machine.

UML types, and whether they earn their place

TypeUse it forIn practice
SequenceInteractions over timeGenuinely useful, widely used
ComponentStructure and interfacesUseful, often replaced by C4 containers
ClassDomain modelsUseful for the model, rarely for the code
State machineLifecycles and transitionsUnderused, frequently finds real bugs
ActivityProcess flowA flowchart with more notation
Use caseActors and goalsRare in engineering practice
DeploymentWhere things runUsually better as an infrastructure diagram

Try it

shell
$ Draw a UML state machine for an order: every state, every permitted transition, and what triggers each one.
FAQ

Questions

Is UML still worth using?

As a shared vocabulary, yes: sequence and state diagrams in particular are widely understood and hard to improve on. As a complete method with tooling and model-driven development, it has largely been abandoned for good reasons.

Can it generate a class diagram from code?

Yes, and the result is usually too detailed to be useful. A class diagram of the domain model is worth having; one of every class in a codebase is a worse version of what your IDE already shows.

Does it export PlantUML?

The canvas exports SVG, PNG and Excalidraw. Ask the agent for PlantUML source and it will write it, which is a different artifact from the drawn diagram.

Which UML diagram should I use for architecture?

Component, or step outside UML and use C4 containers, which most teams find clearer for the same job.

Related