Skip to content

msamylea/AI-Chart-Generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 

Repository files navigation

AI Chart Generation with Mermaid

Added support for Gemini, HuggingFace, and Ollama. Updated UI and made parsing changes. Added several additional types of mermaid charts (Gantt, Block, Sankey, Requirements Diagram, ZenUML)

To run, just clone the repository and run run.py which will start the local Flask app at http://127.0.0.1:5000

If you get strange syntax errors, upgrade Mermaid via npm/yarn/whichever you need, if they continue, it's likely just the LLM output - try updating your prompts. You can try to play with the parsing code, but it's already using a lot of regex to clean the raw output.

Updated UI and added ability to select LLM Provider, Model, enter API key, and set kwargs:

  • HuggingFace OpenAI is HF models using OpenAI API (typically used with HF Pro subscription)
  • HuggingFace Text is HF models via InferenceClient
  • Gemini is Google Gemini models
  • Ollama is self served models
  • OpenAI is.. OpenAI

image

image

image

image

image

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published