This Streamlit application creates an interactive Data Visualization Assistant that can understand Natural Language Queries and generate appropriate Visualizations using LLMs. The demand for AI-powered data visualization tools is surging as businesses seek faster, more intuitive ways to understand their data. We can tap into this growing market by building our own AI-powered visualization tools that integrate seamlessly with existing data workflows.
We'll build an AI Data Visualization Agent using Together AI's powerful language models and E2B's secure code execution environment. This agent will understand natural language queries about your data and automatically generate appropriate visualizations, making data exploration intuitive and efficient.
This tutorial can also be considered as a demo for the E2B Code Interpreter and Together AI, for anyone who's getting started with these libraries!
Features:
💬 Natural language query interface for data visualization
📊 Support for multiple visualization types (line, bar, scatter, pie, bubble charts)
🧹 Automatic data preprocessing and cleaning
🔐 Secure code execution in E2B's sandboxed environment
💻 Interactive Streamlit interface for easy data upload and visualization
⏳ Real-time visualization generation and display
🤖 Available Models:
→ Meta-Llama 3.1 405B
→ DeepSeek V3
→ Qwen 2.5 7B
→ Meta-Llama 3.3 70B
A Streamlit application that acts as your personal data visualization expert, powered by LLMs. Simply upload your dataset and ask questions in natural language - the AI agent will analyze your data, generate appropriate visualizations, and provide insights through a combination of charts, statistics, and explanations.
- Ask questions about your data in plain English
- Get instant visualizations and statistical analysis
- Receive explanations of findings and insights
- Interactive follow-up questioning
- Automatic choice of appropriate chart types
- Dynamic visualization generation
- Statistical visualization support
- Custom plot formatting and styling
- Meta-Llama 3.1 405B for complex analysis
- DeepSeek V3 for detailed insights
- Qwen 2.5 7B for quick analysis
- Meta-Llama 3.3 70B for advanced queries
Follow the steps below to set up and run the application:
- Before anything else, Please get a free Together AI API Key here: https://api.together.ai/signin
- Get a free E2B API Key here: https://e2b.dev/ ; https://e2b.dev/docs/legacy/getting-started/api-key
- Clone the Repository
git clone https://github.com/GURPREETKAURJETHRA/AI-Data-Visualization-Agent.git cd AI-Data-Visualization-Agent
- Install the dependencies
pip install -r requirements.txt
- Run the Streamlit app
streamlit run ai_data_visualisation_agent.py
Happy coding! 🚀✨
Distributed under the MIT License. See LICENSE
for more information.