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/activate
Install the dependencies:
pip install -r requirements.txt
Create 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:

python -m grpc_tools.protoc -I proto --python_out=grpc_server/generated --grpc_python_out=grpc_server/generated proto/uml_service.proto

Running the servers

gRPC server

To run the gRPC server, run the following command:

python -m grpc_server.server

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:

python -m mcp_server.server