Generate control-flow graph from source code using ollama and graphviz

gen-dot-llm.sh

#!/usr/bin/env sh

# Usage
# ./gen-dot-llm.sh filename.java Function

temp_jpg=$(mktemp)
full_prompt=$(mktemp)

prompt="generate a control-flow graph from the following code in dot language.
Consider only the function $2. Output only the raw dot code without any other words.
Do not format as markdown."

code="$(cat $1)"

echo "$prompt" >> $full_prompt
echo "$code" >> $full_prompt

ollama run qwen2.5-coder < $full_prompt | dot -Tjpg > /tmp/dot.jpg

echo "Saved graph in /tmp/dot.jpg"