Uses Google Gemini AI to generate PlantUML code for creating UML Class and Sequence diagrams from domain descriptions or free text input
UMLmcp
This project is a Python service that uses Google Gemini to generate PlantUML code for UML Class and Sequence diagrams. It exposes a gRPC interface and an MCP tool for generating UML diagrams.
Setup
Create a virtual environment:
python -m venv .venv source .venv/bin/activateInstall the dependencies:
pip install -r requirements.txtCreate a
key.txt
file in the root of the project and paste your Gemini API key in the first line.
Generating gRPC stubs
To generate the gRPC stubs, run the following command:
Running the servers
gRPC server
To run the gRPC server, run the following command:
MCP server
The MCP (Model-Context-Protocol) server exposes the generate_uml
tool, allowing other processes to generate UML diagrams. To run the MCP server, use the following command:
CLI usage
To use the CLI, run the following command:
Testing
To run the tests, run the following command:
This server cannot be installed
remote-capable server
The server can be hosted and run remotely because it primarily relies on remote services or has no dependency on the local environment.
Enables users to generate PlantUML code for UML Class and Sequence diagrams using Google Gemini AI. Supports both domain JSON input and free text descriptions to create structured UML diagrams.