# CSV Analyst
from pandas import read_csv
from dotenv import load_dotenv
import streamlit as st
from langchain_openai import OpenAI
from langchain_experimental.agents import create_csv_agent

load_dotenv()

st.set_page_config(page_title='CSV Analyst')
st.header('Query your CSV file')

file = st.file_uploader('Upload your CSV File', type='csv')

if file is not None:
    quest = st.text_input('Enter your Query', placeholder='Query')

    llm = OpenAI(temperature=0, max_tokens=100)
    agent = create_csv_agent(llm, file, verbose=True)

    if quest:
        with st.spinner('Processing'):
            response = agent.run(quest)
        st.write(response)