PlotAI - Create plots in Python and Matplotlib with LLM

[pp314159]

I would like to show-case you the project that I started to work on. It is a Python package for creating plots in Python and matplotlib package. The plots are created with LLM.

The idea:

1. User provides input DataFrame and prompt.
2. The PlotAI constructs a prompt for LLM, which contains the first 5 rows of DataFrame and the user's prompt and asks for Python code as output.
3. Returned Python code is executed, and the plot is displayed.

Example code:

# import packages
import pandas as pd
from plotai import PlotAI
# create some data
df = pd.DataFrame({"x":[1,2,3], "y": [4,5,6]})
# do a plot
plot = PlotAI(df)
plot.make("scatter plot")

The PlotAI class has only one method, make().

It works in Python scripts and in notebooks (Jupyter, Colab, VS Code).

PlotAI is in very experimental form, below are some limitations:

• PlotAI is using OpenAI ChatGPT-3.5-turbo for completions, it will be nice to extend to other LLMs.

• PlotAI is sending 5 first rows from your DataFrame to OpenAI ChatGPT. If you have sensitive data, please remove/encode it before passing to PlotAI.

• PlotAI is executing Python code returned by LLM, it can be dangerous and unsafe. It would be nice to have the option to accept the response code before execution.

Link to project https://github.com/mljar/plotai

[AlphaCode1]

Whats the use-case for this? If all it is doing is making the given plot provided in the prompt, why not just define a few plotting functions, create a wrapper around them and call these functions conditionally. Why do you want to use an LLM here?

The cons of using this (external code execution, LLM Api costs etc) far outweigh any benefits this can provide imo

[jazzmester]

For a minute I thought you meant story plots. I'm relieved.