justplotme is a library that uses LLMs to perform the AGI level task of creating plots.
pip install justplotmeOpen example.ipynb in JupyterLab / Notebook. You'll need an openai api key (only llm currently supported) to use the widget. Export it to an env variable called OPENAI_API_KEY. DO NOT actually put the key in the notebook, you might end up leaking it. In the notebook, you can use justplotme as follows:
import os
from justplotme import PlotMe
OPENAI_KEY = os.getenv("OPENAI_API_KEY")
# df is a pandas DataFrame
PlotMe(df, api_key=OPENAI_KEY)Create a virtual environment and and install justplotme in editable mode with the optional development dependencies:
python3 -m venv .venv
source .venv/bin/activate
pip install -e ".[dev]"Changes made in src/justplotme/static/ will be reflected
in the notebook.
- V0 that works on the browser
- Release on pypi
- Make logging work, less painful
- Make justplotme work in VSCode jupyter notebooks
- Support more LLMs
- Support multiline statements, not just expressions
- Add support to different plotting backends
- More data exploration modalities? Maps maybe?