What's stopping you from writing your Rust like this?
Just write python inline in your Rust code https://crates.io/crates/inline-python
I'm looking for an app to draw diagram
Some Markdown formats (like Gitlab Flavored Markdown) support PlantUML.