| import streamlit as st | |
| from pandasai.llm.openai import OpenAI | |
| from dotenv import load_dotenv | |
| import os | |
| import pandas as pd | |
| from pandasai import PandasAI | |
| load_dotenv() | |
| openai_api_key = os.getenv("OPENAI_API_KEY") | |
| def chat_with_csv(df,prompt): | |
| llm = OpenAI(api_token=openai_api_key) | |
| pandas_ai = PandasAI(llm) | |
| result = pandas_ai.run(df, prompt=prompt) | |
| print(result) | |
| return result | |
| st.set_page_config(layout='wide') | |
| st.title("ChatCSV powered by LLM") | |
| input_csv = st.file_uploader("Upload your CSV file", type=['csv']) | |
| if input_csv is not None: | |
| col1, col2 = st.columns([1,1]) | |
| with col1: | |
| st.info("CSV Uploaded Successfully") | |
| data = pd.read_csv(input_csv) | |
| st.dataframe(data, use_container_width=True) | |
| with col2: | |
| st.info("Chat Below") | |
| input_text = st.text_area("Enter your query") | |
| if input_text is not None: | |
| if st.button("Chat with CSV"): | |
| st.info("Your Query: "+input_text) | |
| result = chat_with_csv(data, input_text) | |
| st.success(result) | |