AI-Powered Threat Modelling Extension

[jgadsden]

Describe what problem your feature request solves:
It would be good to build in a generative AI engine into Threat Dragon that suggests threats for a given data flow diagram model.

The benefits have been described in AI-Powered Threat Modeling with OWASP Threat Dragon – Part 2 (along with Part 1)

This TD AI Modeler tool is open source and @InfosecOTB has provided example threats generated for the threat models

It would be good to integrate this into Threat Dragon, possibly following ideas on the implementation from the discussion topic

Describe the solution you'd like:
initial integration with the desktop version

Additional context:

This is the over-arching issue with various tasks:

[jgadsden]

Thanks for taking this on @InfosecOTB , feel free to add sub-issues if needed
there is a new branch ai-extension created for this

[InfosecOTB]

I’ve just completed the code, which I believe is ready as a candidate for the final solution.
The working implementation is available in the ai-extension branch.
Full details on how it works are documented in the README.md file.
I’ve tested it successfully on Windows 11 and Ubuntu 24.04.
I don’t have access to macOS, so I’d greatly appreciate it if someone could test it there - and on any other supported platforms.
If you find any bugs, issues, or have suggestions, please let me know!

Some print screens from working program.