README.md

metadata

title: Csv Chatbot
emoji: 📊
colorFrom: yellow
colorTo: gray
sdk: gradio
sdk_version: 4.37.1
app_file: app.py
pinned: false

CSV Chatbot

This is a simple LLM agent that can do simple data analysis over csv data. It generates a python code using pandas library to do the analysis. Then that python code is executed and the output is returned to the user.

Setup

cp .env.example .env

# Edit .env file and add your OpenAI API key

pip install -r requirements.txt

Usage

python main.py

Example

Limitations

• It does not execute the code in a sandbox environment. So, it can execute any code that is written in the python code generated by the agent.
• Messages history is not taken into account. So, the agent does not have any context of the previous messages.
• More advanced libraries like langchain can be used to manage complex queries and control flows.